Fix country_code svg

apps/location/transfer_data.py

@@ -1,11 +1,11 @@
-from django.db.models import Q, QuerySet
-
 from transfer.serializers.location import CountrySerializer, RegionSerializer, \
     CitySerializer, AddressSerializer, \
     Country
 from transfer.models import Cities, Locations
 from pprint import pprint
 
+from requests import get
+
 
 def transfer_countries():
     queryset = Cities.objects.raw("""SELECT cities.id, cities.country_code_2 
@@ -116,10 +116,14 @@ def transfer_addresses():
 
 def update_flags():
     queryset = Country.objects.only("id", "code", "svg_image").filter(old_id__isnull=False)
+    link_to_request = "https://s3.eu-central-1.amazonaws.com/gm-test.com/media"
 
     for query in queryset:
-        query.svg_image = f"/image/image/10-31-2019/{query.code}.svg"
-        query.save()
+        svg_link = f"/svg/country/10-31-2019/{query.code}.svg"
+        resp = get(f"{link_to_request}{svg_link}")
+        if resp.status_code == 200:
+            query.svg_image = svg_link
+            query.save()
 
 
 data_types = {