refactor: initial data optimization

prefeitura-rio · Jan 16, 2024 · 8c5df55 · 8c5df55
1 parent 0ff64c9
commit 8c5df55
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/api/scripts/database_initial_data.py b/api/scripts/database_initial_data.py
@@ -84,6 +84,7 @@ async def run():
     estados     = relacao_estado_municipio[['UF','Nome_UF']].drop_duplicates()
     municipios  = relacao_estado_municipio[['UF','Nome_Município','Código Município Completo']]
 
+    municipios_para_criar = []
     for _, estado in estados.iterrows():
         state, _ = await State.get_or_create(
             country=brasil,
@@ -92,11 +93,14 @@ async def run():
         )
 
         for _, municipio in municipios[municipios['UF'] == estado['UF']].iterrows():
-            await City.get_or_create(
-                state=state,
-                code=municipio['Código Município Completo'],
-                name=municipio['Nome_Município']
+            municipios_para_criar.append(
+                City(
+                    state=state,
+                    code=municipio['Código Município Completo'],
+                    name=municipio['Nome_Município']
+                )
             )
+    City.bulk_create(municipios_para_criar, ignore_conflicts=True)
     logger.info("States and Cities created successfully")
 
     await Tortoise.close_connections()