diff options
| author | Étienne Loks <etienne.loks@iggdrasil.net> | 2016-07-16 13:03:27 +0200 |
|---|---|---|
| committer | Étienne Loks <etienne.loks@iggdrasil.net> | 2016-07-16 13:03:27 +0200 |
| commit | 6bb7d4cbf47c61fd168998a0423a31814eaa7fe0 (patch) | |
| tree | f0b7c63834d4fe9125d043893dad8765975a78b1 /archaeological_finds/wizards.py | |
| parent | 3748baac6501db293424d1ae5d092e8672a68f7f (diff) | |
| parent | 242c13fdc9895abdcd1af79bce2a9b3f98f3f939 (diff) | |
| download | Ishtar-6bb7d4cbf47c61fd168998a0423a31814eaa7fe0.tar.bz2 Ishtar-6bb7d4cbf47c61fd168998a0423a31814eaa7fe0.zip | |
Merge branch 'master-WIP-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): |