not now
This commit is contained in:
parent
c9144f7b93
commit
1b93dd163a
@ -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()
|
@ -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()
|
Loading…
Reference in New Issue
Block a user