fix: added country name to gbfs populate script

MobilityData · Aug 19, 2024 · 609aa5f · 609aa5f
1 parent 2cc0d59
commit 609aa5f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/api/src/scripts/populate_db_gbfs.py b/api/src/scripts/populate_db_gbfs.py
@@ -2,6 +2,7 @@
 
 import pandas as pd
 import pytz
+import pycountry
 
 from database.database import generate_unique_id, configure_polymorphic_mappers
 from database_gen.sqlacodegen_models import Gbfsfeed, Location, Gbfsversion, Externalid
@@ -95,6 +96,7 @@ def populate_db(self):
             location = self.db.session.get(Location, location_id) or Location(
                 id=location_id,
                 country_code=country_code,
+                country=pycountry.countries.get(alpha_2=country_code).name if country_code else None,
                 municipality=municipality,
             )
             gbfs_feed.locations.clear()