Multimedika/Bot_Development

1from db.models import Bot, Session_Publisher2from db.models import Session as SessionModel3from db.query.base_query import BaseQuery4 5 6class BotQuery(BaseQuery):7    def __init__(self, user):8        super().__init__(user)9 10    def add_bot(self, db, session_id, bot, metadata_id, type="bot_one"):11        # Create the new session12        if type == "bot_one":13            new_bot_one = Session_Publisher(14                id=session_id,15                user_id=self.user.get("id"),16                bot_name=bot,17                metadata_id=metadata_id,18            )19 20        else:21            new_bot_one = SessionModel(22                id=session_id,23                user_id=self.user.get("id"),24                bot_id=bot,25                metadata_id=metadata_id,26            )27 28        return self.add(db, new_bot_one)29 30    def get_session_ids_bot(self, db, metadata_id, type="bot_one"):31        model = Session_Publisher32        columns = [33            Session_Publisher.id,34            Session_Publisher.bot_name,35            Session_Publisher.updated_at,36        ]37        filter_conditions = [38            Session_Publisher.user_id == self.user.get("id"),39            Session_Publisher.metadata_id == metadata_id,40        ]41 42        sessions = self.get_columns(43            db,44            model=model,45            columns=columns,46            filter_conditions=filter_conditions,47            multiple=True,48        )49 50        return sessions51 52    def update_session_bot(self, db, bot_id, name):53        update_data = {"name": name}54        self.update(db, Bot, bot_id, update_data)55 56    def delete_session_bot(self, db, bot_id):57        self.delete(db, Bot, bot_id)58 59    def get_bot(self, db, bot_id):60        return self.get(db, Bot, bot_id)61 62    # def get_all_bots(self, db):63    #     return self.get_all(db, Bot)64