From 604e44d82e70298751bf23226cbfb094e3ec76d5 Mon Sep 17 00:00:00 2001 From: "(Stagiar) Ezri Mudde" Date: Tue, 21 Sep 2021 11:31:08 +0200 Subject: [PATCH] Fix usage of non-int ids --- fixture_magic/management/commands/custom_dump.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/fixture_magic/management/commands/custom_dump.py b/fixture_magic/management/commands/custom_dump.py index dbe2783..e9c683f 100644 --- a/fixture_magic/management/commands/custom_dump.py +++ b/fixture_magic/management/commands/custom_dump.py @@ -41,8 +41,20 @@ def handle(self, *args, **options): (app_label, model_name) = dump_settings['primary'].split('.') include_primary = dump_settings.get("include_primary", False) dump_me = loading.get_model(app_label, model_name) - objs = dump_me.objects.filter(pk__in=[int(i) for i in pks]) - for obj in objs.all(): + try: + parsers = int, long, str + except NameError: + parsers = int, str + for parser in parsers: + try: + objs = dump_me.objects.filter(pk__in=map(parser, pks)).all() + except ValueError: + pass + else: + break + else: + objs = [] + for obj in objs: # get the dependent objects and add to serialize list for dep in dump_settings['dependents']: try: