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 | 05a15e46173ff8d2f4bb22e5c65b86bdfa2ed36e (patch) | |
| tree | d0f6a4c72463231f218eb40f5a9aa611cc150117 /archaeological_finds/wizards.py | |
| parent | 5e26d52a6268f2aa04d82108431463b184d45cb7 (diff) | |
| download | Ishtar-05a15e46173ff8d2f4bb22e5c65b86bdfa2ed36e.tar.bz2 Ishtar-05a15e46173ff8d2f4bb22e5c65b86bdfa2ed36e.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): |