aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorGravatar Joe Banks <[email protected]>2024-04-07 22:48:35 +0100
committerGravatar Joe Banks <[email protected]>2024-04-07 22:48:35 +0100
commit72b2a26b9dfa2e061a5eff8aeb5d495e76348c04 (patch)
treed9991e4420d420595904d4885ad7ff28529ad9ae
parentBump Poetry Python version to 3.12 (diff)
Fix no active event loop in create_metricity_db.py
-rw-r--r--create_metricity_db.py5
1 files changed, 5 insertions, 0 deletions
diff --git a/create_metricity_db.py b/create_metricity_db.py
index af93c9f..4bf0183 100644
--- a/create_metricity_db.py
+++ b/create_metricity_db.py
@@ -1,4 +1,5 @@
"""Ensures the metricity db exists before running migrations."""
+
import asyncio
from urllib.parse import SplitResult, urlsplit
@@ -18,6 +19,7 @@ def parse_db_url(db_url: str) -> SplitResult:
raise ValueError("The given db_url is not a valid PostgreSQL database URL.")
return db_url_parts
+
async def create_db() -> None:
"""Create the Metricity database if it does not exist."""
parts = parse_db_url(build_db_uri())
@@ -45,5 +47,8 @@ async def create_db() -> None:
await sys_conn.close()
+
if __name__ == "__main__":
+ loop = asyncio.new_event_loop()
+ asyncio.set_event_loop(loop)
asyncio.get_event_loop().run_until_complete(create_db())