| |
| from sqlalchemy import Column, String |
| from . import BASE, SESSION |
|
|
|
|
| class userbase(BASE): |
| __tablename__ = "UserBase" |
| chat_id = Column(String(14), primary_key=True) |
|
|
| def __init__(self, chat_id): |
| self.chat_id = chat_id |
|
|
|
|
| userbase.__table__.create(checkfirst=True) |
|
|
|
|
| def add_to_userbase(chat_id: int): |
| __user = userbase(str(chat_id)) |
| SESSION.add(__user) |
| SESSION.commit() |
|
|
|
|
| def full_userbase(): |
| users = SESSION.query(userbase).all() |
| SESSION.close() |
| return users |
|
|
|
|
| def present_in_userbase(chat_id): |
| try: |
| return SESSION.query(userbase).filter( |
| userbase.chat_id == str(chat_id)).one() |
| except BaseException: |
| return None |
| finally: |
| SESSION.close() |
|
|
