dds_glossary/database.py: improved init_engine

sentier-dev · Jun 26, 2024 · 9ed5691 · 9ed5691
1 parent a52cb50
commit 9ed5691
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/dds_glossary/database.py b/dds_glossary/database.py
@@ -33,12 +33,13 @@ def init_engine(
     database_url = database_url or get_settings().DATABASE_URL.get_secret_value()
     engine = create_engine(database_url)
 
-    if not database_exists(engine.url):
-        create_database(engine.url)
-    elif drop_database_flag:
+    if database_exists(engine.url):
+        if not drop_database_flag:
+            return engine
         drop_database(engine.url)
-        create_database(engine.url)
+        engine.dispose()
 
+    create_database(engine.url)
     Base.metadata.create_all(engine)
     return engine