diff options
Diffstat (limited to 'create_metricity_db.py')
| -rw-r--r-- | create_metricity_db.py | 5 | 
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())  |