This commit is contained in:
levina 2021-11-14 19:10:34 +07:00 committed by GitHub
parent c9144f7b93
commit 1b93dd163a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 0 additions and 55 deletions

View File

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

View File

@ -1,34 +0,0 @@
import threading
from driver.database import BASE, SESSION
from sqlalchemy import Column, String, UnicodeText
class Chats(BASE):
__tablename__ = "chats"
chat_id = Column(String(14), primary_key=True)
chat_name = Column(UnicodeText)
def __init__(self, chat_id, chat_name=None):
self.chat_id = chat_id
self.chat_name = chat_name
Chats.__table__.create(checkfirst=True)
CHATS_LOCK = threading.RLock()
CHATS_DATA = set()
def del_chat(chat_id):
with CHATS_LOCK:
chat = SESSION.query(Chats).get(str(chat_id))
if chat:
SESSION.delete(chat)
SESSION.commit()
def chatdata():
global CHAT_ID
try:
CHAT_ID = {int(x.chat_id) for x in SESSION.query(Chats).all()}
return CHAT_ID
finally:
SESSION.close()