From 5ca7407d99b07694fd5facf612f3d2c8562c000b Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Sat, 14 Sep 2019 18:29:47 +0200 Subject: Fix file imports - fix tests with redirs --- archaeological_finds/wizards.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'archaeological_finds/wizards.py') diff --git a/archaeological_finds/wizards.py b/archaeological_finds/wizards.py index b34bcdcbe..71bc6deed 100644 --- a/archaeological_finds/wizards.py +++ b/archaeological_finds/wizards.py @@ -194,8 +194,10 @@ class TreatmentWizard(TreatmentBase): if 'resulting_pk' in dct: dct['items'] = [] pks = dct.pop('resulting_pk') + if isinstance(pks, models.Find): + pks = [pks] if not isinstance(pks, (list, tuple)): - pks = pks.split(u',') + pks = str(pks).split(u',') for pk in pks: if isinstance(pk, models.Find): -- cgit v1.2.3