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.'))