From 512915dd4587491dad19c13cde92daada5756d72 Mon Sep 17 00:00:00 2001 From: Guillermo Serrahima Date: Mon, 2 Feb 2026 02:43:14 +0100 Subject: Initial commit. Achieved minimum functionality --- velascobot2/utils.py | 48 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 48 insertions(+) create mode 100644 velascobot2/utils.py (limited to 'velascobot2/utils.py') diff --git a/velascobot2/utils.py b/velascobot2/utils.py new file mode 100644 index 0000000..e89e1d4 --- /dev/null +++ b/velascobot2/utils.py @@ -0,0 +1,48 @@ +from enum import StrEnum + + +class GraphicType(StrEnum): + STICKER = "STICKER" + ANIMATION = "ANIMATION" + VIDEO = "VIDEO" + + +class Response(StrEnum): + TEXT = "TEXT" + GRAPHIC = "GRAPHIC" + + +class ChatData: + def __init__( + self, + id: str, + chat_type: str, + name: str, + processed_messages: int = 0, + message_period: int = 10, + answer_chance: float = 0.5, + restricted: bool = False, + silenced: bool = False, + ) -> None: + self.id = id + self.type = chat_type + self.name = name + self.processed_messages = processed_messages + self.message_period = message_period + self.answer_chance = answer_chance + self.restricted = restricted + self.silenced = silenced + + @staticmethod + def from_sql_row(row: tuple[str, str, str, int, int, float, int, int]): + print("From SQL row:", repr(row)) + return ChatData( + row[0], + row[1], + row[2], + int(row[3]), + int(row[4]), + float(row[5]), + int(row[6]) == 1, + int(row[7]) == 1, + ) -- cgit v1.2.3