diff options
| author | Étienne Loks <etienne.loks@iggdrasil.net> | 2016-06-24 20:45:12 +0200 |
|---|---|---|
| committer | Étienne Loks <etienne.loks@iggdrasil.net> | 2016-06-24 20:45:12 +0200 |
| commit | 2f50c5f1ba28d9c2b660e550a437f61dadf6d37b (patch) | |
| tree | d0f6a4c72463231f218eb40f5a9aa611cc150117 /archaeological_finds/wizards.py | |
| parent | 29bace24d91984f2f6d5884a8094e2ddeb5d74d9 (diff) | |
| download | Ishtar-2f50c5f1ba28d9c2b660e550a437f61dadf6d37b.tar.bz2 Ishtar-2f50c5f1ba28d9c2b660e550a437f61dadf6d37b.zip | |
Reactivate treatments
Diffstat (limited to 'archaeological_finds/wizards.py')
| -rw-r--r-- | archaeological_finds/wizards.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/archaeological_finds/wizards.py b/archaeological_finds/wizards.py index fabe1039a..8a6b92440 100644 --- a/archaeological_finds/wizards.py +++ b/archaeological_finds/wizards.py @@ -85,6 +85,14 @@ class FindDeletionWizard(DeletionWizard): class TreatmentWizard(Wizard): model = models.Treatment + basket_step = 'basetreatment-treatment_creation' + + def get_form_kwargs(self, step): + kwargs = super(TreatmentWizard, self).get_form_kwargs(step) + if self.basket_step not in step: + return kwargs + kwargs['user'] = self.request.user + return kwargs class FindSourceWizard(SourceWizard): |