Резервное копирование

bot_modules/backup.py

@@ -0,0 +1,115 @@
+# -*- coding: utf8 -*-
+# Общественное достояние 2023, Алексей Безбородов (Alexei Bezborodov) <AlexeiBv+mirocod_platform_bot@narod.ru> 
+
+# Бэкапы пользователя
+
+from bot_sys import bot_bd, log, config, keyboard, user_access
+from bot_modules import start, access, groups
+from aiogram import Bot, types
+
+import sqlite3
+
+from aiogram.dispatcher import Dispatcher
+
+bot = Bot(token=config.GetTelegramBotApiToken(), parse_mode=types.ParseMode.HTML)
+
+# ---------------------------------------------------------
+# БД
+module_name = 'backup'
+
+init_bd_cmds = [
+f"INSERT OR IGNORE INTO module_access (modName, modAccess) VALUES ('{module_name}', 'other=+');"
+]
+
+# ---------------------------------------------------------
+# Сообщения
+
+backup_message = '''
+<b>Здесь вы можете выполнить специальные операции по сервисному обслуживанию</b>
+'''
+
+backup_bd_message = '''
+<b>📦 Резервная копия базы данных</b>
+🕰 <code>@time</code>
+'''
+
+backup_log_message = '''
+<b>📦 Резервная копия логов</b>
+🕰 <code>@time</code>
+'''
+
+error_backup_message = '''
+<b>📦 Ошибка резервного копирования</b>
+'''
+
+backup_button_name = "📰 Резервные копии и логи"
+backup_bd_button_name = "📰 Резервные копия базы"
+backup_log_button_name = "📰 Логи"
+
+# ---------------------------------------------------------
+# Работа с кнопками
+
+def GetBackupKeyboardButtons(a_UserGroups):
+    cur_buttons = [
+        keyboard.ButtonWithAccess(backup_bd_button_name, user_access.AccessMode.EDIT, GetAccess()), 
+        keyboard.ButtonWithAccess(backup_log_button_name, user_access.AccessMode.EDIT, GetAccess())
+    ]
+    mods = [start]
+    return keyboard.MakeKeyboard(keyboard.GetButtons(mods) + cur_buttons, a_UserGroups)
+
+# ---------------------------------------------------------
+# Обработка сообщений
+
+async def BackupOpen(a_Message):
+    user_id = str(a_Message.from_user.id)
+    user_groups= groups.GetUserGroupData(user_id)
+    if not user_access.CheckAccessString(GetAccess(), user_groups, user_access.AccessMode.VIEW):
+        return await bot.send_message(user_id, access.access_denied_message, reply_markup = GetBackupKeyboardButtons(user_groups))
+
+    await bot.send_message(user_id, backup_message, reply_markup = GetBackupKeyboardButtons(user_groups))
+
+def BackupFileTemplate(a_Path, a_CaptionMessage, a_AccessFunc, a_ButtonFunc, a_ErrorMessage):
+    async def BackupFile(a_Message):
+        user_id = str(a_Message.from_user.id)
+        user_groups= groups.GetUserGroupData(user_id)
+        if not user_access.CheckAccessString(a_AccessFunc(), user_groups, user_access.AccessMode.EDIT):
+            return await bot.send_message(user_id, access.access_denied_message, reply_markup = a_ButtonFunc(user_groups))
+
+        document = await GetFile(a_Path)
+        if document is None:
+            return await bot.send_message(user_id, error_backup_message, reply_markup = a_ButtonFunc(user_groups))
+
+        await bot.send_document(user_id, document, caption = a_CaptionMessage.replace('@time', log.GetTime()), reply_markup = a_ButtonFunc(user_groups))
+    return BackupFile
+
+# ---------------------------------------------------------
+# Работа с базой данных пользователей
+
+async def GetFile(a_Path):
+    try:
+        document = open(a_Path, 'rb')
+        log.Success(f'Загружен файл {a_Path}')
+        return document
+    except Exception as e:
+        log.Error(f'Не удалось загрузить файл {a_Path}. Ошибка {str(e)}')
+        return None
+
+# ---------------------------------------------------------
+# API
+
+# Инициализация БД
+def GetInitBDCommands():
+    return init_bd_cmds
+
+def GetAccess():
+    return access.GetAccessForModule(module_name)
+
+# Доступные кнопки
+def GetModuleButtons():
+    return [keyboard.ButtonWithAccess(backup_button_name, user_access.AccessMode.VIEW, GetAccess())]
+
+# Обработка кнопок
+def RegisterHandlers(dp : Dispatcher):
+    dp.register_message_handler(BackupOpen, text = backup_button_name)
+    dp.register_message_handler(BackupFileTemplate(bot_bd.GetBDFileName(), backup_bd_message, GetAccess, GetBackupKeyboardButtons, error_backup_message), text = backup_bd_button_name)
+    dp.register_message_handler(BackupFileTemplate(log.g_log_file_name, backup_log_message, GetAccess, GetBackupKeyboardButtons, error_backup_message), text = backup_log_button_name)

bot_modules/start.py

@@ -4,7 +4,7 @@
 # Стартовое меню
 
 from bot_sys import log, config, keyboard, user_access
-from bot_modules import profile, projects, groups, access
+from bot_modules import profile, projects, groups, access, backup
 
 from aiogram.dispatcher import Dispatcher
 
@@ -25,13 +25,13 @@ start_message = '''
 <b>Приятного пользования!</b>
 '''
 
-start_menu_button_name = "☰ Главное меню"
+start_menu_button_name = "≣ Главное меню"
 
 # ---------------------------------------------------------
 # Работа с кнопками
 
 def GetStartKeyboardButtons(a_UserGroups):
-    mods = [profile, projects, groups, access]
+    mods = [profile, projects, groups, access, backup]
     return keyboard.MakeKeyboardForMods(mods, a_UserGroups)
 
 # ---------------------------------------------------------

main.py

@@ -10,13 +10,13 @@ from aiogram.dispatcher import Dispatcher
 from aiogram.contrib.fsm_storage.memory import MemoryStorage
 import sqlite3
 from bot_sys import config, log, bot_bd, user_access
-from bot_modules import profile, start, projects, groups, access
+from bot_modules import profile, start, projects, groups, access, backup
 
 storage = MemoryStorage()
 bot = Bot(token=config.GetTelegramBotApiToken(), parse_mode = types.ParseMode.HTML)
 dp = Dispatcher(bot, storage = storage)
 
-mods = [access, profile, start, projects, groups]
+mods = [access, profile, start, projects, groups, backup]
 
 init_bd_cmd = []
 for m in mods: