update 1. GTP-6 Astra, оптимизация кода и фикс

This commit is contained in:
2026-09-26 21:19:09 +03:00
parent 70fce14772
commit 22e10b53b7
31 changed files with 2211 additions and 162 deletions

View File

@@ -8,6 +8,7 @@ from sqlalchemy import create_engine
from sqlalchemy.orm import Session, sessionmaker
from hockey_data.models import Base
from hockey_data.database import HockeyDatabase
class LocalTestDatabase:
@@ -29,14 +30,6 @@ class LocalTestDatabase:
def create_all(self) -> None:
Base.metadata.create_all(self.engine)
@contextmanager
def session(self) -> Iterator[Session]:
session = self.SessionFactory()
try:
yield session
session.commit()
except Exception:
session.rollback()
raise
finally:
session.close()
# Exercise the same transaction nesting as production, with a local engine.
session = HockeyDatabase.session
atomic_session = HockeyDatabase.atomic_session