im too lazy to write a comment, but believe me, i tried to write good code

Author: MothScientist

budget_graph/bot.py

@@ -20,6 +20,13 @@
 
 logger_database = setup_logger('logs/DatabaseLog.log', 'db_logger')
 
+feature_ids: dict[str, int] = {
+    'del_msg_after_transaction': 1,
+    'skip_input_date': 2,
+    'skip_input_category': 3,
+    'skip_input_description': 4,
+}
+
 
 @timeit
 def connect_db():
@@ -86,7 +93,7 @@ def close_db_flask_g(error):  # DO NOT REMOVE the parameter  # noqa
 
 def read_sql_file(request_name):
     filename: str = path.join(path.dirname(__file__), f'sql/{request_name}.sql')
-    logger_database.info(f'[DB_QUERY] request to *.sql file: {filename}')
+    logger_database.info(f'[REQUEST *.SQL] File: {filename}')
     with open(filename, 'r', encoding='utf-8') as sql_file:
         return sql_file.read()
 
@@ -97,6 +104,8 @@ class DatabaseQueries:
     The class is used to query the database.
     All details of the structure of functions are indicated in the comments to the first function
     """
+    __slots__ = ('__conn',)
+
     def __init__(self, connection):
         self.__conn = connection
 
@@ -363,12 +372,27 @@ def check_user_is_group_owner_by_telegram_id(self, telegram_id: int, group_id: i
                         read_sql_file('check_user_is_group_owner_by_telegram_id'),
                         {'owner': telegram_id, 'group_id': group_id}
                     )
-                    return bool(cur.fetchone()[0])
+                    return res[0] if (res := cur.fetchone()) else False
 
         except (DatabaseError, TypeError) as err:
-            logger_database.error(f"[DB_QUERY] {str(err)}, "
-                                  f"telegram_id: {logging_hash(telegram_id)}"
-                                  f"group id: {group_id}")
+            logger_database.error(f'[DB_QUERY] {str(err)}, '
+                                  f'telegram_id: {logging_hash(telegram_id)}'
+                                  f'group id: {group_id}')
+            return False
+
+    def check_user_is_premium_by_telegram_id(self, telegram_id: int) -> bool:
+        try:
+            with self.__conn as conn:
+                with conn.cursor() as cur:
+                    cur.execute(
+                        read_sql_file('check_user_is_premium_by_telegram_id'),
+                        {'telegram_id': telegram_id}
+                    )
+                    return res[0] if (res := cur.fetchone()) else False
+
+        except (DatabaseError, TypeError) as err:
+            logger_database.error(f'[DB_QUERY] {str(err)}, '
+                                  f'telegram_id: {logging_hash(telegram_id)}')
             return False
 
     def get_group_owner_username_by_group_id(self, group_id: int) -> str:
@@ -422,7 +446,7 @@ def get_user_timezone_by_telegram_id(self, telegram_id: int) -> int | None:
             with self.__conn as conn:
                 with conn.cursor() as cur:
                     cur.execute(read_sql_file('get_user_timezone_by_telegram_id'), {'telegram_id': telegram_id})
-                    return cur.fetchone()[0]
+                    return res[0] if (res := cur.fetchone()) else None
 
         except (DatabaseError, TypeError) as err:
             logger_database.error(f"[DB_QUERY] {str(err)}, "
@@ -435,7 +459,7 @@ def check_record_id_is_exist(self, group_id: int, transaction_id: int) -> bool:
                 with conn.cursor() as cur:
                     cur.execute(read_sql_file('check_record_id_is_exist'),
                                 {'group_id': group_id, 'transaction_id': transaction_id})
-                    return bool(cur.fetchone()[0])
+                    return bool(res[0]) if (res := cur.fetchone()) else False
 
         except (DatabaseError, TypeError) as err:
             logger_database.error(f"[DB_QUERY] {str(err)}, "
@@ -451,7 +475,7 @@ def check_username_is_exist(self, username: str) -> bool:
             with self.__conn as conn:
                 with conn.cursor() as cur:
                     cur.execute(read_sql_file('check_username_is_exist'), {'username': username})
-                    return bool(cur.fetchone()[0])
+                    return bool(res[0]) if (res := cur.fetchone()) else False
 
         except (DatabaseError, TypeError) as err:
             logger_database.error(f"[DB_QUERY] {str(err)}, "
@@ -463,7 +487,7 @@ def check_telegram_id_is_exist(self, telegram_id: int) -> bool:
             with self.__conn as conn:
                 with conn.cursor() as cur:
                     cur.execute(read_sql_file('check_telegram_id_is_exist'), {'telegram_id': telegram_id})
-                    return bool(cur.fetchone()[0])
+                    return bool(res[0]) if (res := cur.fetchone()) else False
 
         except (DatabaseError, TypeError) as err:
             logger_database.error(f"[DB_QUERY] {str(err)}, "
@@ -478,7 +502,7 @@ def check_token_is_unique(self, token: str) -> bool:
             with self.__conn as conn:
                 with conn.cursor() as cur:
                     cur.execute(read_sql_file('check_token_is_unique'), {'token': token})
-                    return not bool(cur.fetchone()[0])
+                    return not bool(res[0]) if (res := cur.fetchone()) else False
 
         except (DatabaseError, TypeError) as err:
             logger_database.error(f"[DB_QUERY] {str(err)}, "
@@ -496,42 +520,50 @@ def check_limit_users_in_group(self, group_id: int) -> bool:
             with self.__conn as conn:
                 with conn.cursor() as cur:
                     cur.execute(read_sql_file('check_limit_users_in_group'), {'group_id': group_id})
-                    return bool(cur.fetchone()[0])
+                    return bool(res[0]) if (res := cur.fetchone()) else False
 
         except (DatabaseError, TypeError) as err:
             logger_database.error(f"[DB_QUERY] {str(err)}, "
                                   f"group id: {group_id}")
             return False  # to prevent a user from being written to a non-existent group
 
-    def get_feature_status_del_msg_after_transaction(self, telegram_id: int) -> bool:
-        """
-        Checks the status whether the function is enabled
-        for the operation of the StorageMsgIdForDeleteAfterOperation class
-        """
+    def get_skip_operations_status(self, telegram_id: int) -> tuple:
+        try:
+            with self.__conn as conn:
+                with conn.cursor() as cur:
+                    cur.execute(read_sql_file('get_skip_operations_status'), {'telegram_id': telegram_id})
+                    return tuple(bool(status) for status in res[0]) if (res := cur.fetchone()) else tuple([False]*3)
+
+        except (DatabaseError, TypeError) as err:
+            logger_database.error(f"[DB_QUERY] {str(err)}, "
+                                  f"telegram_id: {telegram_id}")
+            return tuple([False]*3)  # default value - (False, False, False)
+
+    def get_feature_status(self, telegram_id: int, feature: str):
         try:
             with self.__conn as conn:
                 with conn.cursor() as cur:
                     cur.execute(
-                        read_sql_file('get_feature_status_del_msg_after_transaction'),
-                        {'telegram_id': telegram_id}
+                        read_sql_file('get_feature_status'),
+                        {'telegram_id': telegram_id, 'feature': feature_ids.get(feature)}
                     )
-                    return bool(cur.fetchone()[0])
+                    return bool(res[0]) if (res := cur.fetchone()) else False
 
         except (DatabaseError, TypeError) as err:
             logger_database.error(f"[DB_QUERY] {str(err)}, "
                                   f"telegram_id: {telegram_id}")
             return False  # default value - False
 
-    def change_feature_status_del_msg_after_transaction(self, telegram_id: int) -> bool:
+    def change_feature_status(self, telegram_id: int, feature: str):
         try:
             with self.__conn as conn:
                 with conn.cursor() as cur:
                     cur.execute(
-                        read_sql_file('change_feature_status_del_msg_after_transaction'),
-                        {'telegram_id': telegram_id}
+                        read_sql_file('change_feature_status'),
+                        {'telegram_id': telegram_id, 'feature': feature_ids.get(feature)}
                     )
                     conn.commit()
-            logger_database.info(f"[del_msg_transaction] successfully changed for {logging_hash(telegram_id)}")
+            logger_database.info(f"[{feature}] successfully changed for {logging_hash(telegram_id)}")
             return True
 
         except (DatabaseError, TypeError) as err:
@@ -817,7 +849,7 @@ def delete_group_with_users(self, group_id: int) -> bool:  # TODO REFERENCES IN
                                     DELETE FROM
                                         "budget_graph"."users"
                                     WHERE
-                                        "telegram_id" = ANY(ARRAY(SELECT "telegram_id" FROM telegram_id_users));
+                                        "telegram_id" IS NOT NULL AND "telegram_id" = ANY(ARRAY(SELECT "telegram_id" FROM telegram_id_users));
                                     
                                     -- groups
                                     DELETE FROM

budget_graph/db_manager.py

@@ -19,7 +19,7 @@ def __init__(self, feature_is_active=None):
 		self.msg_id_to_delete: list = []
 
 	def append(self, msg_id: int):
-		if self.feature_is_active:
+		if self.feature_is_active and msg_id not in self.msg_id_to_delete:
 			self.msg_id_to_delete.append(msg_id)
 
 	def delete_messages(self, bot, chat_id: int) -> None:
@@ -74,25 +74,63 @@ def get_language_buttons() -> tuple:
 
 
 @cache
-def get_diagram_buttons(user_language: str, user_is_owner: bool) -> tuple:
+def get_diagram_buttons(user_language: str, user_is_owner: bool, user_is_premium: bool) -> tuple:
+	return_buttons: list = []
 	buttons: tuple = (
 		[
 			InlineKeyboardButton(
 				receive_translation(user_language, 'my_budget_diagram'), callback_data='get_diagram_0'
-			)
+			)  # premium
 		],
 		[
 			InlineKeyboardButton(
 				receive_translation(user_language, 'budget_diagram_for_group'), callback_data='get_diagram_1'
-			)
+			)  # group_owner
 		],
 		[
 			InlineKeyboardButton(
 				receive_translation(user_language, 'specific_user_budget_diagram'), callback_data='get_diagram_2'
+			)  # group_owner + premium
+		],
+		[
+			InlineKeyboardButton(
+				receive_translation(user_language, 'get_diagram_info'), callback_data='get_diagram_info'
+			)  # for any user
+		]
+	)
+	# It is important that the buttons are in their original order.
+	if user_is_premium:
+		return_buttons.append(buttons[0])
+	if user_is_owner:
+		return_buttons.append(buttons[1])
+		if user_is_premium:
+			return_buttons.append(buttons[2])
+	return_buttons.append(buttons[-1])
+	return tuple(return_buttons)
+
+
+@cache
+def get_premium_buttons(user_language: str, premium_status: bool) -> tuple:
+	premium_buttons: tuple = (
+		[
+			InlineKeyboardButton(
+				receive_translation(user_language, 'get_premium'), callback_data='premium_get'
 			)
 		],
+		[
+			InlineKeyboardButton(
+				receive_translation(user_language, 'check_premium_status'), callback_data='premium_status'
+			)
+		],
+		[
+			InlineKeyboardButton(
+				receive_translation(user_language, 'about_premium'), callback_data='premium_info'
+			)
+		]
 	)
-	return buttons if user_is_owner else buttons[0]
+	if premium_status:
+		return tuple(premium_buttons[1:])
+	return tuple(premium_buttons[0:3:2])
 
 
 @cache
@@ -154,12 +192,15 @@ def get_category_translate(user_language: str) -> tuple:
 def get_bot_commands() -> list:
 	commands: list = [
 		BotCommand('start', 'Start'),
+		BotCommand('help', 'Help'),
+		BotCommand('premium', 'Premium'),
 		BotCommand('change_language', 'Change Language'),
-		BotCommand('del_msg_transaction', '[ON/OFF] Delete messages after successful transaction'),
+		BotCommand('del_msg_after_transaction', '[ON/OFF] Delete messages after successful transaction'),
+		BotCommand('skip_input_date', '[ON/OFF] Skip setting transaction date (current will be used)'),
+		BotCommand('skip_input_category', '[ON/OFF] Skip transaction category selection'),
+		BotCommand('skip_input_description', '[ON/OFF] Skip adding description to transaction'),
 		BotCommand('change_timezone', 'Change Time zone'),
 		BotCommand('get_my_id', 'Get my Telegram ID'),
-		BotCommand('premium', 'Premium'),
-		BotCommand('help', 'Help'),
 		BotCommand('project_github', 'GitHub')
 	]
 

budget_graph/helpers.py

@@ -55,7 +55,7 @@
     "jewelry": "Schmuck",
     "salary": "Gehalt",
     "other": "Andere",
-    "select_category": "Wählen Sie eine Kategorie",
+    "select_category": "Wählen Sie eine Kategorie:",
     "add_description": "Beschreibung hinzufügen (nicht mehr als 50 Zeichen)",
     "no_description": "Keine Beschreibung",
     "entry_add_success": "Eintrag erfolgreich hinzugefügt",
@@ -84,7 +84,7 @@
     "datetime": "Terminzeit",
     "description": "Beschreibung",
     "table_is_empty": "Ihre Datensatztabelle ist leer",
-    "csv_not_found_error": "Fehler. Versuchen Sie es später erneut oder melden Sie das Problem dem technischen Support.",
+    "error_connect_support": "Fehler. Versuchen Sie es später erneut oder melden Sie das Problem dem technischen Support.",
     "owner": "Eigentümer",
     "small_group_exception": "Es gibt nur 1 Mitglied in dieser Gruppe.",
     "username_new_owner": "Schreiben Sie unten den Namen des Benutzers (aus der Liste), den Sie als Eigentümer der Gruppe zuweisen möchten",
@@ -131,14 +131,27 @@
     "select_timezone": "Wählen Sie Ihre Zeitzone aus",
     "error_select_timezone": "Fehler beim Festlegen des Zeitzonenwerts",
     "timezone_changed": "Die Zeitzone für Ihr Konto wurde erfolgreich eingestellt",
-    "del_msg_transaction_on": "Funktion zum Löschen von Nachrichten nach der Transaktion aktiviert",
-    "del_msg_transaction_of": "Die Funktion zum Löschen von Nachrichten nach einer Transaktion ist deaktiviert",
-    "del_msg_transaction_err": "Beim Aktivieren/Deaktivieren der Funktionalität ist ein Fehler aufgetreten",
+    "del_msg_after_transaction_on": "Funktion zum Löschen von Nachrichten nach der Transaktion aktiviert",
+    "del_msg_after_transaction_off": "Die Funktion zum Löschen von Nachrichten nach einer Transaktion ist deaktiviert",
+    "feature_err": "Beim Aktivieren/Deaktivieren der Funktionalität ist ein Fehler aufgetreten",
     "get_diagram": "Diagramm erstellen",
     "my_budget_diagram": "Mein Diagramm",
     "budget_diagram_for_group": "Graph gruppieren",
     "specific_user_budget_diagram": "Diagramm nach bestimmtem Benutzer",
     "choose_diagram_type": "Wählen Sie einen verfügbaren Diagrammtyp aus:",
     "wait_diagram": "Wir erstellen bereits ein Diagramm. Bitte warten...",
-    "feature_uuid_filename": "Anwendungs-ID (Name der zukünftigen Datei)"
+    "feature_uuid_filename": "Anwendungs-ID (Name der zukünftigen Datei)",
+    "get_diagram_info": "Mehr zu Diagrammen",
+    "diagram_info": "Das Diagramm ist eine HTML-Datei, die Sie auf jedem Gerät öffnen können. Es enthält in Form eines Diagramms Informationen zu Ihren Ausgaben, den Ausgaben einer Gruppe oder bestimmter Benutzer (je nach ausgewählter Kategorie).\n\nNur der Gruppenbesitzer kann Informationen zu den Gruppenausgaben anfordern. Um Ihre Ausgaben anzuzeigen, müssen Sie über ein Premium-Abonnement verfügen. Wenn Sie die Ausgaben bestimmter Benutzer sehen möchten, muss der Eigentümer der Gruppe ebenfalls über ein Premium-Abonnement verfügen und nur er selbst kann ein solches Diagramm anfordern.\n\nSobald Sie die Datei erhalten haben, können Sie sie an jeden senden.\n\nWenn Sie den Zeitplan auswählen, der Sie interessiert, wird eine Anfrage gesendet. Die durchschnittliche Wartezeit zum Empfangen eines Diagramms beträgt ungefähr 1 Sekunde",
+    "get_premium": "Holen Sie sich den Premium-Status",
+    "check_premium_status": "Aktuelle Abonnementinformationen",
+    "about_premium": "Mehr zum Premium-Status",
+    "choose_menu": "Wählen Sie den Artikel, der Sie interessiert, aus dem Menü",
+    "premium_info": "Der Premium-Status bietet Zugriff auf erweiterte Funktionen wie exklusive Funktionen und vorrangigen technischen Support.\n\nNicht-Premium-Benutzer haben weiterhin Zugriff auf alle Kernfunktionen unseres Dienstes und können die Vorteile genauso nutzen wie andere Benutzer.\n\nDer Kauf eines Premium-Status durch Benutzer trägt zur Unterstützung und Entwicklung des Produkts bei",
+    "skip_input_date_on": "Funktion zum Überspringen der Eingabe des Transaktionsdatums aktiviert",
+    "skip_input_date_off": "Funktion zum Überspringen der Eingabe des Transaktionsdatums deaktiviert",
+    "skip_input_category_on": "Funktion zur Auswahl der Transaktionskategorie aktiviert",
+    "skip_input_category_off": "Die Funktion zur Auswahl der Transaktionskategorie ist deaktiviert",
+    "skip_input_description_on": "Die Funktion zum Hinzufügen einer Beschreibung für eine Transaktion ist aktiviert",
+    "skip_input_description_off": "Die Funktion zum Hinzufügen einer Beschreibung für eine Transaktion ist deaktiviert"
 }