added data_sql for saving data

created data_sql for user data
This commit is contained in:
levina 2021-11-14 07:15:13 +07:00 committed by GitHub
parent f502dd6a2b
commit af807b5bec
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -0,0 +1,37 @@
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 chatlists():
global CHAT_ID
try:
CHAT_ID = {int(x.chat_id) for x in SESSION.query(Chats).all()}
return CHAT_ID
finally:
SESSION.close()