bot_starters.py

from sqlalchemy import Column, String, UnicodeText

from . import BASE, SESSION


class Bot_Starters(BASE):
    __tablename__ = "bot_starters"
    user_id = Column(String(14), primary_key=True)
    first_name = Column(UnicodeText)
    date = Column(UnicodeText)
    username = Column(UnicodeText)

    def __init__(self, user_id, first_name, date, username):
        self.user_id = str(user_id)
        self.first_name = first_name
        self.date = date
        self.username = username


Bot_Starters.__table__.create(checkfirst=True)


def add_starter_to_db(
    user_id,
    first_name,
    date,
    username,
):
    to_check = get_starter_details(user_id)
    if not to_check:
        user = Bot_Starters(str(user_id), first_name, date, username)
        SESSION.add(user)
        SESSION.commit()
        return True
    rem = SESSION.query(Bot_Starters).get(str(user_id))
    SESSION.delete(rem)
    SESSION.commit()
    user = Bot_Starters(str(user_id), first_name, date, username)
    SESSION.add(user)
    SESSION.commit()
    return True


def del_starter_from_db(user_id):
    to_check = get_starter_details(user_id)
    if not to_check:
        return False
    rem = SESSION.query(Bot_Starters).get(str(user_id))
    SESSION.delete(rem)
    SESSION.commit()
    return True


def get_starter_details(user_id):
    try:
        if _result := SESSION.query(Bot_Starters).get(str(user_id)):
            return _result
        return None
    finally:
        SESSION.close()


def get_all_starters():
    try:
        return SESSION.query(Bot_Starters).all()
    except BaseException:
        return None
    finally:
        SESSION.close()