def add_chat(chat_id) -> bool:
exists = sql.select(where="chats", condition={"chat_id": chat_id})
if not exists:
logging.info(chat_id)
sql.insert(table="chats", chat_id=chat_id)
return False
else:
return True
def add_winner_answer(chat_id, category, answer, message_id, poll_id,
vote_count):
sql.insert(table="winner_answers",
chat_id=chat_id,
message_id=message_id,
answer=answer,
poll_id=poll_id,
category=category,
vote_count=vote_count)
def add_winner_question(chat_id, category, question, message_id, poll_id):
if sql.select(where="winner_questions", condition={"category": category}):
sql.update(table="winner_questions",
chat_id=chat_id,
message_id=message_id,
question=question,
poll_id=poll_id,
condition=dict(category=category))
else:
sql.insert(table="winner_questions",
chat_id=chat_id,
message_id=message_id,
question=question,
poll_id=poll_id,
category=category)
def add_answer(chat_id, answer, message_id, poll_id):
sql.insert(table="answers",
chat_id=chat_id,
answer=answer,
message_id=message_id,
poll_id=poll_id)
def add_question(chat_id, question, message_id, poll_id):
sql.insert(table="questions",
chat_id=chat_id,
question=question,
message_id=message_id,
poll_id=poll_id)
def save_no_phase(messages: List[Message]):
for message in messages:
sql.insert(table="no_phase_message",
chat_id=message.chat.id,
message_id=message.message_id)
def save_sent(id_question, chat_id, message_id):
sql.insert(table="sent_messages",
id_question=id_question,
chat_id=chat_id,
message_id=message_id)