Use aiomyql driver to generate models

pyproject.toml

@@ -4,7 +4,7 @@ requires = [
     "setuptools_scm[toml]>=6.2",
     "sqlacodegen-v2",
     "sqlalchemy",
-    "pymysql",
+    "aiomysql",
 ]
 build-backend = "setuptools.build_meta"
 
@@ -25,7 +25,7 @@ dependencies = [
     "uvicorn",
     "starlette",
     "strawberry-graphql[asgi]",
-    "sqlalchemy",
+    "sqlalchemy[asyncio]",
     "aiomysql",
     "opentelemetry-api",
     "opentelemetry-sdk",

setup.py

@@ -1,14 +1,26 @@
+import asyncio
 import os
 
 import setuptools
 from sqlacodegen_v2.generators import DeclarativeGenerator
-from sqlalchemy import MetaData, create_engine
+from sqlalchemy import MetaData
+from sqlalchemy.ext.asyncio import create_async_engine
 
-if __name__ == "__main__":
-    engine = create_engine(os.environ["DATABASE_URL"])
+
+async def generate_models():
+    engine = create_async_engine(os.environ["DATABASE_URL"])
     metadata = MetaData()
-    metadata.reflect(engine, only=["EnergyScan"])
-    generator = DeclarativeGenerator(metadata, engine, set())
+    async with engine.begin() as connection:
+        await connection.run_sync(
+            lambda connection: metadata.reflect(connection, only=["EnergyScan"])
+        )
+        generator = await connection.run_sync(
+            lambda connection: DeclarativeGenerator(metadata, connection, set())
+        )
     with open("src/graph_energy_scan/models.py", "w") as models_file:
         models_file.write(generator.generate())
+
+
+if __name__ == "__main__":
+    asyncio.run(generate_models())
     setuptools.setup()