Xtao-Labs/video-stream
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

none

Browse Source
This commit is contained in:
levina 2021-11-14 06:53:08 +07:00 committed by GitHub
parent c6f7dc0457
commit 1356c93862
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
driver/database/__init__.py
View File

@ -1,29 +1,20 @@
import os import os
import re import re
from sqlalchemy import create_engine from sqlalchemy import create_engine
from config import DATABASE_URL as db
from sqlalchemy.ext.declarative import declarative_base from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import scoped_session, sessionmaker from sqlalchemy.orm import scoped_session, sessionmaker
from config import DATABASE_URL as db
if db and db.startswith("postgres://"): if db and db.startswith("postgres://"):
app = db.replace("postgres://", "postgresql://", 1) app = db.replace("postgres://", "postgresql://", 1)
BASE = declarative_base() BASE = declarative_base()
def start() -> scoped_session: def start() -> scoped_session:
engine = create_engine(app) engine = create_engine(app)
BASE.metadata.bind = engine BASE.metadata.bind = engine
BASE.metadata.create_all(engine) BASE.metadata.create_all(engine)
return scoped_session(sessionmaker(bind=engine, autoflush=False)) return scoped_session(sessionmaker(bind=engine, autoflush=False))
SESSION = start() SESSION = start()