global session to local session

orm/base.py

@@ -2,7 +2,7 @@ from typing import TypeVar, Any, Dict, Generic, Callable
 
 from sqlalchemy import create_engine, Column, Integer
 from sqlalchemy.ext.declarative import declarative_base
-from sqlalchemy.orm import sessionmaker
+from sqlalchemy.orm import Session
 from sqlalchemy.sql.schema import Table
 
 from settings import DB_URL
@@ -10,48 +10,48 @@ from orm._retry import RetryingQuery
 
 # engine = create_engine(DB_URL, convert_unicode=True, echo=False)
 engine = create_engine(DB_URL, 
-                        convert_unicode=True, 
-                        echo=False,
-                        #pool_size=10,
-                        #max_overflow=2,
-                        #pool_recycle=300,
-                        pool_pre_ping=True,
-                        #pool_use_lifo=True
-                        )
-
-Session = sessionmaker(autocommit=False, autoflush=False, bind=engine, query_cls=RetryingQuery)
-#Session = sessionmaker(autocommit=False, autoflush=False, bind=engine)
-global_session = Session()
+						convert_unicode=True, 
+						echo=False,
+						#pool_size=10,
+						#max_overflow=2,
+						#pool_recycle=300,
+						pool_pre_ping=True,
+						#pool_use_lifo=True
+						future=True
+						)
 
 T = TypeVar("T")
 
 REGISTRY: Dict[str, type] = {}
 
+def local_session():
+	return Session(bind=engine, expire_on_commit=False)
+
 
 class Base(declarative_base()):
-    __table__: Table
-    __tablename__: str
-    __new__: Callable
-    __init__: Callable
+	__table__: Table
+	__tablename__: str
+	__new__: Callable
+	__init__: Callable
 
-    __abstract__: bool = True
-    __table_args__ = {"extend_existing": True}
-    id: int = Column(Integer, primary_key=True)
-    session = global_session
+	__abstract__: bool = True
+	__table_args__ = {"extend_existing": True}
+	id: int = Column(Integer, primary_key=True)
 
-    def __init_subclass__(cls, **kwargs):
-        REGISTRY[cls.__name__] = cls
+	def __init_subclass__(cls, **kwargs):
+		REGISTRY[cls.__name__] = cls
 
-    @classmethod
-    def create(cls: Generic[T], **kwargs) -> Generic[T]:
-        instance = cls(**kwargs)
-        return instance.save()
+	@classmethod
+	def create(cls: Generic[T], **kwargs) -> Generic[T]:
+		instance = cls(**kwargs)
+		return instance.save()
 
-    def save(self) -> Generic[T]:
-        self.session.add(self)
-        self.session.commit()
-        return self
+	def save(self) -> Generic[T]:
+		with local_session() as session:
+			session.add(self)
+			session.commit()
+		return self
 
-    def dict(self) -> Dict[str, Any]:
-        column_names = self.__table__.columns.keys()
-        return {c: getattr(self, c) for c in column_names}
+	def dict(self) -> Dict[str, Any]:
+		column_names = self.__table__.columns.keys()
+		return {c: getattr(self, c) for c in column_names}