mirror of
https://github.com/MuRuLOSE/limoka.git
synced 2026-06-18 15:14:18 +02:00
Commited backup
This commit is contained in:
323
hikariatama/ftg/backuper.py
Normal file
323
hikariatama/ftg/backuper.py
Normal file
@@ -0,0 +1,323 @@
|
||||
# █ █ ▀ █▄▀ ▄▀█ █▀█ ▀
|
||||
# █▀█ █ █ █ █▀█ █▀▄ █
|
||||
# © Copyright 2022
|
||||
# https://t.me/hikariatama
|
||||
#
|
||||
# 🔒 Licensed under the GNU AGPLv3
|
||||
# 🌐 https://www.gnu.org/licenses/agpl-3.0.html
|
||||
|
||||
# meta pic: https://static.dan.tatar/backuper_icon.png
|
||||
# meta banner: https://mods.hikariatama.ru/badges/backuper.jpg
|
||||
# meta developer: @hikarimods
|
||||
# scope: hikka_only
|
||||
# scope: hikka_min 1.2.10
|
||||
|
||||
import datetime
|
||||
import io
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
import zipfile
|
||||
|
||||
from telethon.tl.types import Message
|
||||
|
||||
from .. import loader, utils
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
DATA_DIR = (
|
||||
os.path.normpath(os.path.join(utils.get_base_dir(), ".."))
|
||||
if "OKTETO" not in os.environ and "DOCKER" not in os.environ
|
||||
else "/data"
|
||||
)
|
||||
|
||||
LOADED_MODULES_DIR = os.path.join(DATA_DIR, "loaded_modules")
|
||||
|
||||
|
||||
@loader.tds
|
||||
class BackuperMod(loader.Module):
|
||||
"""Create the backup of all modules or the whole database"""
|
||||
|
||||
strings = {
|
||||
"name": "Backuper",
|
||||
"backup_caption": (
|
||||
"<emoji document_id=5469718869536940860>👆</emoji> <b>This is your database"
|
||||
" backup. Do not give it to anyone, it contains personal info.</b>"
|
||||
),
|
||||
"reply_to_file": (
|
||||
"<emoji document_id=5312526098750252863>🚫</emoji> <b>Reply to .json or"
|
||||
" .zip"
|
||||
" file</b>"
|
||||
),
|
||||
"db_restored": (
|
||||
"<emoji document_id=5774134533590880843>🔄</emoji> <b>Database updated,"
|
||||
" restarting...</b>"
|
||||
),
|
||||
"modules_backup": (
|
||||
"<emoji document_id=6334332637041134172>🗃</emoji> <b>Backup mods ({})</b>"
|
||||
),
|
||||
"mods_restored": (
|
||||
"<emoji document_id=5314250708508220914>✅</emoji> <b>Mods restored,"
|
||||
" restarting</b>"
|
||||
),
|
||||
}
|
||||
|
||||
strings_ru = {
|
||||
"backup_caption": (
|
||||
"<emoji document_id=5469718869536940860>👆</emoji> <b>Это - бекап базы"
|
||||
" данных. Никому его не передавай, он содержит личную информацию.</b>"
|
||||
),
|
||||
"reply_to_file": (
|
||||
"<emoji document_id=5312526098750252863>🚫</emoji> <b>Ответь на .json или"
|
||||
" .zip файл</b>"
|
||||
),
|
||||
"db_restored": (
|
||||
"<emoji document_id=5774134533590880843>🔄</emoji> <b>База обновлена,"
|
||||
" перезагружаюсь...</b>"
|
||||
),
|
||||
"modules_backup": (
|
||||
"<emoji document_id=6334332637041134172>🗃</emoji> <b>Бекап модулей ({})</b>"
|
||||
),
|
||||
"mods_restored": (
|
||||
"<emoji document_id=5314250708508220914>✅</emoji> <b>Модули восстановлены,"
|
||||
" перезагружаюсь</b>"
|
||||
),
|
||||
"_cmd_doc_backupdb": "Создать бекап базы данных [будет отправлен в Избранное]",
|
||||
"_cmd_doc_restoredb": "Восстановить базу данных из файла",
|
||||
"_cmd_doc_backupmods": "Создать бекап модулей",
|
||||
"_cmd_doc_restoremods": "<reply to file> - Восстановить модули из файла",
|
||||
"_cls_doc": "Создает резервные копии",
|
||||
}
|
||||
|
||||
strings_de = {
|
||||
"backup_caption": (
|
||||
"<emoji document_id=5469718869536940860>👆</emoji> <b>Dies ist ein"
|
||||
" Datenbank-Backup. Gib es niemandem, es enthält persönliche"
|
||||
" Informationen.</b>"
|
||||
),
|
||||
"reply_to_file": (
|
||||
"<emoji document_id=5312526098750252863>🚫</emoji> <b>Antworte auf .json"
|
||||
" oder .zip Datei</b>"
|
||||
),
|
||||
"db_restored": (
|
||||
"<emoji document_id=5774134533590880843>🔄</emoji> <b>Datenbank"
|
||||
" aktualisiert, starte neu...</b>"
|
||||
),
|
||||
"modules_backup": (
|
||||
"<emoji document_id=6334332637041134172>🗃</emoji> <b>Backup-Module ({})</b>"
|
||||
),
|
||||
"mods_restored": (
|
||||
"<emoji document_id=5314250708508220914>✅</emoji> <b>Module"
|
||||
" wiederhergestellt, starte neu</b>"
|
||||
),
|
||||
"_cmd_doc_backupdb": (
|
||||
"Datenbank-Backup erstellen [wird in den Favoriten gesendet]"
|
||||
),
|
||||
"_cmd_doc_restoredb": "Datenbank aus Datei wiederherstellen",
|
||||
"_cmd_doc_backupmods": "Backup-Module erstellen",
|
||||
"_cmd_doc_restoremods": "<reply to file> - Module aus Datei wiederherstellen",
|
||||
"_cls_doc": "Erstellt Sicherungskopien",
|
||||
}
|
||||
|
||||
strings_hi = {
|
||||
"backup_caption": (
|
||||
"<emoji document_id=5469718869536940860>👆</emoji> <b>यह आपका डेटाबेस बैकअप"
|
||||
" है। किसी को भी न दें, यह व्यक्तिगत जानकारी सामग्री में है।</b>"
|
||||
),
|
||||
"reply_to_file": (
|
||||
"<emoji document_id=5312526098750252863>🚫</emoji> <b>.json या .zip फ़ाइल पर"
|
||||
" जवाब दें</b>"
|
||||
),
|
||||
"db_restored": (
|
||||
"<emoji document_id=5774134533590880843>🔄</emoji> <b>डेटाबेस अपडेट कर रहा"
|
||||
" है, पुनः आरंभ कर रहा है...</b>"
|
||||
),
|
||||
"modules_backup": (
|
||||
"<emoji document_id=6334332637041134172>🗃</emoji> <b>मॉड्स बैकअप ({})</b>"
|
||||
),
|
||||
"mods_restored": (
|
||||
"<emoji document_id=5314250708508220914>✅</emoji> <b>मॉड्स पुनः स्थापित कर"
|
||||
" रहे हैं, पुनः आरंभ कर रहे हैं</b>"
|
||||
),
|
||||
"_cmd_doc_backupdb": "डेटाबेस बैकअप बनाएं [फ़ेवरिट्स में भेजा जाएगा]",
|
||||
"_cmd_doc_restoredb": "फ़ाइल से डेटाबेस पुनः स्थापित करें",
|
||||
"_cmd_doc_backupmods": "मॉड्स बैकअप बनाएं",
|
||||
"_cmd_doc_restoremods": "<reply to file> - फ़ाइल से मॉड्स पुनः स्थापित करें",
|
||||
"_cls_doc": "बैकअप बनाता है",
|
||||
}
|
||||
|
||||
strings_uz = {
|
||||
"backup_caption": (
|
||||
"<emoji document_id=5469718869536940860>👆</emoji> <b>Bu sizning"
|
||||
" ma'lumotlar"
|
||||
" bazangizning e'loni. Kimga ko'rsatmasangiz, shu shaxsiy ma'lumotlarni o'z"
|
||||
" ichiga oladi.</b>"
|
||||
),
|
||||
"reply_to_file": (
|
||||
"<emoji document_id=5312526098750252863>🚫</emoji> <b>.json yoki .zip"
|
||||
" faylga"
|
||||
" javob bering</b>"
|
||||
),
|
||||
"db_restored": (
|
||||
"<emoji document_id=5774134533590880843>🔄</emoji> <b>Ma'lumotlar bazasi"
|
||||
" yangilandi, qayta ishga tushirilmoqda...</b>"
|
||||
),
|
||||
"modules_backup": (
|
||||
"<emoji document_id=6334332637041134172>🗃</emoji> <b>Modullar e'loni"
|
||||
" ({})</b>"
|
||||
),
|
||||
"mods_restored": (
|
||||
"<emoji document_id=5314250708508220914>✅</emoji> <b>Modullar qayta"
|
||||
" tiklandi, qayta ishga tushirilmoqda</b>"
|
||||
),
|
||||
"_cmd_doc_backupdb": (
|
||||
"Ma'lumotlar bazasini e'lon qiling [Favoritlarga jo'natiladi]"
|
||||
),
|
||||
"_cmd_doc_restoredb": "Fayldan ma'lumotlar bazasini tiklash",
|
||||
"_cmd_doc_backupmods": "Modullarni e'lon qiling",
|
||||
"_cmd_doc_restoremods": "<reply to file> - Fayldan modullarni tiklash",
|
||||
"_cls_doc": "E'lon qiladi",
|
||||
}
|
||||
|
||||
strings_tr = {
|
||||
"backup_caption": (
|
||||
"<emoji document_id=5469718869536940860>👆</emoji> <b>Bu veritabanı"
|
||||
" yedeğinizdir. Kimseye verin, kişisel bilgiler içerir.</b>"
|
||||
),
|
||||
"reply_to_file": (
|
||||
"<emoji document_id=5312526098750252863>🚫</emoji> <b>.json veya .zip"
|
||||
" dosyasına yanıt verin</b>"
|
||||
),
|
||||
"db_restored": (
|
||||
"<emoji document_id=5774134533590880843>🔄</emoji> <b>Veritabanı"
|
||||
" güncellendi, yeniden başlatılıyor...</b>"
|
||||
),
|
||||
"modules_backup": (
|
||||
"<emoji document_id=6334332637041134172>🗃</emoji> <b>Modüller yedeği"
|
||||
" ({})</b>"
|
||||
),
|
||||
"mods_restored": (
|
||||
"<emoji document_id=5314250708508220914>✅</emoji> <b>Modüller geri"
|
||||
" yüklendi, yeniden başlatılıyor</b>"
|
||||
),
|
||||
"_cmd_doc_backupdb": "Veritabanı yedeği oluştur [favorilere gönderilecek]",
|
||||
"_cmd_doc_restoredb": "Dosyadan veritabanını geri yükle",
|
||||
"_cmd_doc_backupmods": "Modüller yedeği oluştur",
|
||||
"_cmd_doc_restoremods": "<dosyaya yanıt ver> - Modülleri dosyadan geri yükle",
|
||||
"_cls_doc": "Yedek oluşturur",
|
||||
}
|
||||
|
||||
async def backupdbcmd(self, message: Message):
|
||||
"""Create database backup [will be sent in pm]"""
|
||||
txt = io.BytesIO(json.dumps(self._db).encode("utf-8"))
|
||||
txt.name = (
|
||||
f"db-backup-{getattr(datetime, 'datetime', datetime).now().strftime('%d-%m-%Y-%H-%M')}.json"
|
||||
)
|
||||
await self._client.send_file("me", txt, caption=self.strings("backup_caption"))
|
||||
await message.delete()
|
||||
|
||||
async def restoredbcmd(self, message: Message):
|
||||
"""Restore database from file"""
|
||||
reply = await message.get_reply_message()
|
||||
if not reply or not reply.media:
|
||||
await utils.answer(
|
||||
message,
|
||||
self.strings("reply_to_file"),
|
||||
)
|
||||
return
|
||||
|
||||
file = await self._client.download_file(reply.media, bytes)
|
||||
decoded_text = json.loads(file.decode("utf-8"))
|
||||
|
||||
if not self._db.process_db_autofix(decoded_text):
|
||||
raise RuntimeError("Attempted to restore broken database")
|
||||
|
||||
self._db.clear()
|
||||
self._db.update(**decoded_text)
|
||||
self._db.save()
|
||||
await utils.answer(message, self.strings("db_restored"))
|
||||
await self.allmodules.commands["restart"](
|
||||
await message.respond(f"{self.get_prefix()}restart --force")
|
||||
)
|
||||
|
||||
async def backupmodscmd(self, message: Message):
|
||||
"""Create backup of mods"""
|
||||
mods_quantity = len(self.lookup("Loader").get("loaded_modules", {}))
|
||||
|
||||
result = io.BytesIO()
|
||||
result.name = "mods.zip"
|
||||
|
||||
db_mods = json.dumps(self.lookup("Loader").get("loaded_modules", {})).encode()
|
||||
|
||||
with zipfile.ZipFile(result, "w", zipfile.ZIP_DEFLATED) as zipf:
|
||||
if "DYNO" not in os.environ:
|
||||
for root, _, files in os.walk(LOADED_MODULES_DIR):
|
||||
for file in files:
|
||||
with open(os.path.join(root, file), "rb") as f:
|
||||
zipf.writestr(file, f.read())
|
||||
mods_quantity += 1
|
||||
|
||||
zipf.writestr("db_mods.json", db_mods)
|
||||
|
||||
archive = io.BytesIO(result.getvalue())
|
||||
archive.name = (
|
||||
f"mods-{getattr(datetime, 'datetime', datetime).now().strftime('%d-%m-%Y-%H-%M')}.zip"
|
||||
)
|
||||
|
||||
await self._client.send_file(
|
||||
utils.get_chat_id(message),
|
||||
archive,
|
||||
caption=self.strings("modules_backup").format(mods_quantity),
|
||||
)
|
||||
await message.delete()
|
||||
|
||||
async def restoremodscmd(self, message: Message):
|
||||
"""<reply to file> - Restore mods from backup"""
|
||||
reply = await message.get_reply_message()
|
||||
if not reply or not reply.media:
|
||||
await utils.answer(message, self.strings("reply_to_file"))
|
||||
return
|
||||
|
||||
file = await self._client.download_file(reply.media, bytes)
|
||||
try:
|
||||
decoded_text = json.loads(file.decode("utf-8"))
|
||||
except Exception:
|
||||
try:
|
||||
file = io.BytesIO(file)
|
||||
file.name = "mods.zip"
|
||||
|
||||
with zipfile.ZipFile(file) as zf:
|
||||
for name in zf.namelist():
|
||||
with zf.open(name, "r") as module:
|
||||
if name == "db_mods.json":
|
||||
db_mods = json.loads(module.read().decode())
|
||||
if isinstance(db_mods, dict) and all(
|
||||
isinstance(key, str) and isinstance(value, str)
|
||||
for key, value in db_mods.items()
|
||||
):
|
||||
self.lookup("Loader").set("loaded_modules", db_mods)
|
||||
|
||||
continue
|
||||
|
||||
if "DYNO" not in os.environ:
|
||||
with open(
|
||||
os.path.join(LOADED_MODULES_DIR, name), "wb"
|
||||
) as f:
|
||||
f.write(module.read())
|
||||
except Exception:
|
||||
logger.exception("Can't restore mods")
|
||||
await utils.answer(message, self.strings("reply_to_file"))
|
||||
return
|
||||
else:
|
||||
if not isinstance(decoded_text, dict) or not all(
|
||||
isinstance(key, str) and isinstance(value, str)
|
||||
for key, value in decoded_text.items()
|
||||
):
|
||||
raise RuntimeError("Invalid backup")
|
||||
|
||||
self.lookup("Loader").set("loaded_modules", decoded_text)
|
||||
|
||||
await utils.answer(message, self.strings("mods_restored"))
|
||||
await self.allmodules.commands["restart"](
|
||||
await message.respond(f"{self.get_prefix()}restart --force")
|
||||
)
|
||||
Reference in New Issue
Block a user