Explicitly register asynpg dialect with GINO (#259)

Author: dmannarino

app/application.py

@@ -9,6 +9,17 @@
 from .gino import Gino, GinoEngine
 from .settings.globals import GLOBALS
 
+# Explicitly register the gino asyncpg dialect with SQLAlchemy
+# This ensures it's available before any database connections are attempted
+try:
+    from sqlalchemy.dialects import registry
+    import gino.dialects.asyncpg
+    registry.register("asyncpg", "gino.dialects.asyncpg", "AsyncpgDialect")
+    registry.register("postgresql.asyncpg", "gino.dialects.asyncpg", "AsyncpgDialect")
+except Exception:
+    # If registration fails, the normal import should still work
+    pass
+
 READ_ENGINE: Optional[GinoEngine] = None
 SessionLocal: Optional[Session] = None
 Base = declarative_base()
@@ -20,6 +31,7 @@
 
 db = Gino(
     app,
+    driver="asyncpg",
     host=GLOBALS.database_config.host,
     port=GLOBALS.database_config.port,
     user=GLOBALS.database_config.username,
@@ -41,8 +53,20 @@ def get_synchronous_db() -> Iterator[Session]:
         raise RuntimeError("No database url set.")
 
     if SessionLocal is None:
-        db_conn = GLOBALS.database_config.url
-        engine = create_engine(db_conn, pool_size=5, max_overflow=0)
+        # Create a synchronous database URL using psycopg2 instead of asyncpg
+        # asyncpg only works with async engines, not synchronous ones
+        from sqlalchemy.engine.url import URL
+
+        db_config = GLOBALS.database_config
+        sync_url = URL(
+            drivername="postgresql+psycopg2",
+            username=db_config.username,
+            password=str(db_config.password) if db_config.password else None,
+            host=db_config.host,
+            port=db_config.port,
+            database=db_config.database,
+        )
+        engine = create_engine(sync_url, pool_size=5, max_overflow=0)
         SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 
     synchronous_db: Optional[Session] = None

app/gino.py

@@ -17,6 +17,9 @@
 from starlette.exceptions import HTTPException
 from starlette.types import Receive, Scope, Send
 
+# Explicitly import the asyncpg dialect to ensure it's registered with SQLAlchemy
+import gino.dialects.asyncpg  # noqa: F401
+
 logger = logging.getLogger("gino.ext.starlette")
 
 

app/settings/globals.py

@@ -134,7 +134,7 @@ def set_database_config(cls, v, values, **kwarg) -> DatabaseURL:
         if db_reader_secret:
             secret = json.loads(db_reader_secret)
             v = DatabaseURL(
-                drivername="postgresql+asyncpg",
+                drivername="asyncpg",
                 username=secret["username"],
                 password=secret["password"],
                 host=secret["host"],
@@ -143,7 +143,7 @@ def set_database_config(cls, v, values, **kwarg) -> DatabaseURL:
             )
         else:
             v = DatabaseURL(
-                drivername="postgresql+asyncpg",
+                drivername="asyncpg",
                 username=input.get("reader_username"),
                 password=input.get("reader_password"),
                 host=input.get("reader_host"),

tests/run_tests.sh

@@ -2,6 +2,11 @@
 
 set -e
 
+# Clean Python cache to avoid stale bytecode issues
+echo "Cleaning Python cache..."
+find /app -type d -name __pycache__ -exec rm -rf {} + 2>/dev/null || true
+find /app -type f -name "*.pyc" -delete 2>/dev/null || true
+
 pushd /app/tests/terraform
 terraform init && terraform plan && terraform apply -auto-approve
 popd