wip

learn_sql_model/api/hero.py

@@ -83,5 +83,5 @@ async def get_heros(
 ) -> Heros:
     "get all heros"
     statement = select(Hero)
-    heros = session.execute(statement).all()
+    heros = session.exec(statement).all()
     return Heros(__root__=heros)

learn_sql_model/config.py

@@ -27,7 +27,6 @@ class ApiClient(BaseModel):
     protocol: str = "https"
     url: str = f"{protocol}://{host}"
 
-
 class Database:
     def __init__(self, config: "Config" = None) -> None:
         if config is None:
@@ -41,19 +40,22 @@ class Database:
                 "transactions": None,
             }
             self.db_state = ContextVar("db_state", default=self.db_state_default.copy())
+            
+        self.db_conf = {}
+        if 'sqlite' in self.config.database_url:
+            self.db_conf = {
+                'connect_args': {"check_same_thread": False},
+                'pool_recycle': 3600,
+                'pool_pre_ping': True,
+            }
+        self._engine = create_engine(
+                self.config.database_url,
+                **self.db_conf
+            )
 
     @property
     def engine(self) -> "Engine":
-        try:
+        return self._engine
-            return self._engine
-        except AttributeError:
-            self._engine = create_engine(
-                self.config.database_url,
-                connect_args={"check_same_thread": False},
-                pool_recycle=3600,
-                pool_pre_ping=True,
-            )
-            return self._engine
 
     @property
     def session(self) -> "Session":
@@ -99,19 +101,14 @@ def get_config(overrides: dict = {}) -> Config:
 
 
 config = get_config()
-engine = create_engine(
+database = get_database()
-    config.database_url,
-    connect_args={"check_same_thread": False},
-    pool_recycle=3600,
-    pool_pre_ping=True,
-)
 
-SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=database.engine)
 
 
 def get_session() -> "Session":
-    with Session(engine) as session:
+    with Session(database.engine) as session:
-        yield SessionLocal()
+        yield session
 
 
 async def reset_db_state(config: Config = None) -> None:

learn_sql_model/game/player.py

@@ -16,7 +16,7 @@ class Player:
         self.hero = HeroCreate(**hero.dict()).post()
 
         self.game = game
-        self.others = Heros(heros=[])
+        self.others = [] #Heros(heros=[])
         self.width = 16
         self.height = 16
         self.white = (255, 255, 255)