replacing asyncio.run with run_sync

Author: dantownsend

piccolo/apps/migrations/commands/backwards.py

@@ -1,7 +1,6 @@
-import asyncio
-
 from piccolo.apps.migrations.auto import MigrationManager
 from piccolo.apps.migrations.tables import Migration
+from piccolo.utils.sync import run_sync
 from .base import BaseMigrationManager
 
 
@@ -69,12 +68,10 @@ def run(self):
             for migration_id in reversed_migration_ids:
                 print(f"Reversing {migration_id}")
                 migration_module = migration_modules[migration_id]
-                response = asyncio.run(
-                    migration_module.forwards()
-                )  # type: ignore
+                response = run_sync(migration_module.forwards())
 
                 if isinstance(response, MigrationManager):
-                    asyncio.run(response.run_backwards())
+                    run_sync(response.run_backwards())
 
                 Migration.delete().where(
                     Migration.name == migration_id

piccolo/apps/migrations/commands/base.py

@@ -1,5 +1,4 @@
 from __future__ import annotations
-import asyncio
 import importlib
 import os
 import sys
@@ -13,6 +12,7 @@
 from piccolo.apps.migrations.auto.diffable_table import DiffableTable
 from piccolo.apps.migrations.auto.schema_snapshot import SchemaSnapshot
 from piccolo.apps.migrations.tables import Migration
+from piccolo.utils.sync import run_sync
 
 
 class BaseMigrationManager(Finder):
@@ -85,7 +85,7 @@ def get_migration_managers(
         ] = self.get_migration_modules(migrations_folder)
 
         for _, migration_module in migration_modules.items():
-            response = asyncio.run(migration_module.forwards())
+            response = run_sync(migration_module.forwards())
             if isinstance(response, MigrationManager):
                 if max_migration_id:
                     if response.migration_id == max_migration_id:

piccolo/apps/migrations/commands/forwards.py

@@ -1,5 +1,4 @@
 from __future__ import annotations
-import asyncio
 import sys
 import typing as t
 
@@ -8,6 +7,7 @@
 from piccolo.apps.migrations.tables import Migration
 from piccolo.apps.migrations.auto import MigrationManager
 from piccolo.conf.apps import MigrationModule
+from piccolo.utils.sync import run_sync
 
 
 class ForwardsMigrationManager(BaseMigrationManager):
@@ -60,10 +60,10 @@ def run_migrations(self, app_config: AppConfig) -> None:
                 print(f"Faked {_id}")
             else:
                 migration_module = migration_modules[_id]
-                response = asyncio.run(migration_module.forwards())
+                response = run_sync(migration_module.forwards())
 
                 if isinstance(response, MigrationManager):
-                    asyncio.run(response.run())
+                    run_sync(response.run())
 
                 print(f"Ran {_id}")