From 398a982c8207f8b15d05cb0fb16b101eaaa95173 Mon Sep 17 00:00:00 2001 From: Dmitriy Kuzmenko Date: Sun, 3 Nov 2019 14:09:07 +0300 Subject: [PATCH] add fix currency --- .../commands/add_establishment_currency.py | 33 ++++++++++--------- 1 file changed, 17 insertions(+), 16 deletions(-) diff --git a/apps/establishment/management/commands/add_establishment_currency.py b/apps/establishment/management/commands/add_establishment_currency.py index bc6da057..6840eda8 100644 --- a/apps/establishment/management/commands/add_establishment_currency.py +++ b/apps/establishment/management/commands/add_establishment_currency.py @@ -23,21 +23,22 @@ class Command(BaseCommand): except Establishment.DoesNotExist: continue else: - country_list = resp.json() - if country_list: - currency_dict = country_list[0].get("currency") - if currency_dict: - name = currency_dict.get("name") - if name: - currency, created = Currency.objects.get_or_create( - slug=currency_dict.get("name").lower(), - ) - if created: - currency.name = {"en-GB": name}, - currency.sign = currency_dict.get("symbol"), - currency.save() - est.currency = currency - est.save() - count += 1 + if resp.status_code == requests.codes.ok: + country_list = resp.json() + if isinstance(country_list, list): + currency_dict = country_list[0].get("currency") + if currency_dict: + name = currency_dict.get("name") + if name: + currency, created = Currency.objects.get_or_create( + slug=currency_dict.get("name").lower(), + ) + if created: + currency.name = {"en-GB": name}, + currency.sign = currency_dict.get("symbol"), + currency.save() + est.currency = currency + est.save() + count += 1 self.stdout.write(self.style.WARNING(f'Created/updated {count} objects.'))