SocialNetwork

1
import sqlalchemy as sa
2
import sqlalchemy.orm as orm
3
from sqlalchemy.orm import Session
4
import sqlalchemy.ext.declarative as dec
5

6
SqlAlchemyBase = dec.declarative_base()
7

8
__factory = None
9

10
def global_init(db_file):
11
    global __factory
12

13
    if __factory:
14
        return
15

16
    if not db_file or not db_file.strip():
17
        raise Exception("Необходимо указать файл базы данных.")
18

19
    conn_str = f'sqlite:///{db_file.strip()}?check_same_thread=False'
20
    print(f"Подключение к базе данных по адресу {conn_str}")
21

22
    engine = sa.create_engine(conn_str, echo=False)
23
    __factory = orm.sessionmaker(bind=engine, autoflush=False)
24

25
    from . import __all_models
26

27
    SqlAlchemyBase.metadata.create_all(engine)
28

29

30
def create_session() -> Session:
31
    global __factory
32
    return __factory()