Refactored the Base class source.

Author: drmacsika

core/base.py

@@ -1,4 +1,7 @@
 from core.settings import settings
 
 Base = settings.Base
+
+
+from accounts.models import User
 from blog.models import Category, Post

core/database.py

@@ -1,23 +1,24 @@
-from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
-from sqlalchemy.orm import sessionmaker
+# from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
+# from sqlalchemy.orm import sessionmaker
 
-from core.settings import settings
+# from core.settings import settings
 
-engine = create_async_engine(settings.SQLALCHEMY_DATABASE_URI, echo = True)
-async_session = sessionmaker(engine, expire_on_commit=False, class_=AsyncSession)
+# engine = create_async_engine(settings.SQLALCHEMY_DATABASE_URI, echo = True)
+# async_session = sessionmaker(engine, expire_on_commit=False, class_=AsyncSession)
 
 
-# Postgres settings
-async def init_models():
-    async with engine.begin() as conn:
-        # await conn.run_sync(Base.metadata.drop_all)
-        await conn.run_sync(settings.Base.metadata.create_all)
+# # Postgres settings
+# async def init_models():
+#     async with engine.begin() as conn:
+#         # await conn.run_sync(Base.metadata.drop_all)
+#         await conn.run_sync(settings.Base.metadata.create_all)
 
 
-async def get_session() -> AsyncSession:
-    async_session = sessionmaker(
-        engine, class_=AsyncSession, expire_on_commit=False
-    )
-    async with async_session() as session:
-        yield session
+# async def get_session() -> AsyncSession:
+#     async_session = sessionmaker(
+#         engine, class_=AsyncSession, expire_on_commit=False
+#     )
+#     async with async_session() as session:
+#         yield session
+