Vladless
100 lines
3.0 KiB
Python
100 lines
3.0 KiB
Python
from datetime import datetime
|
|
import os
|
|
import subprocess
|
|
from typing import Union
|
|
|
|
from aiogram.types import BufferedInputFile
|
|
|
|
from config import ADMIN_ID, BACK_DIR, DB_NAME, DB_PASSWORD, DB_USER
|
|
from logger import logger
|
|
|
|
|
|
async def backup_database():
|
|
from bot import bot
|
|
|
|
try:
|
|
if backup_file_path := _create_database_backup():
|
|
await _send_backup_to_admin(bot, backup_file_path)
|
|
_cleanup_old_backups()
|
|
except Exception as e:
|
|
logger.error(f"Ошибка при создании или отправке бэкапа: {e}")
|
|
|
|
|
|
def _create_database_backup():
|
|
USER = DB_USER
|
|
HOST = "localhost"
|
|
BACKUP_DIR = BACK_DIR
|
|
DATE = datetime.now().strftime("%Y-%m-%d-%H%M%S")
|
|
BACKUP_FILE = f"{BACKUP_DIR}/{DB_NAME}-backup-{DATE}.sql"
|
|
|
|
os.environ["PGPASSWORD"] = DB_PASSWORD
|
|
|
|
try:
|
|
subprocess.run(
|
|
[
|
|
"pg_dump",
|
|
"-U",
|
|
USER,
|
|
"-h",
|
|
HOST,
|
|
"-F",
|
|
"c",
|
|
"-f",
|
|
BACKUP_FILE,
|
|
DB_NAME,
|
|
],
|
|
check=True,
|
|
)
|
|
logger.info(f"Бэкап базы данных создан: {BACKUP_FILE}")
|
|
return BACKUP_FILE
|
|
except subprocess.CalledProcessError as e:
|
|
logger.error(f"Ошибка при создании бэкапа базы данных: {e}")
|
|
return None
|
|
finally:
|
|
del os.environ["PGPASSWORD"]
|
|
|
|
|
|
async def _send_backup_to_admin(bot, backup_file_path):
|
|
try:
|
|
with open(backup_file_path, "rb") as backup_file:
|
|
backup_input_file = BufferedInputFile(backup_file.read(), filename=os.path.basename(backup_file_path))
|
|
admin_ids: Union[int, list[int]] = ADMIN_ID
|
|
if isinstance(admin_ids, list):
|
|
for id in admin_ids:
|
|
await bot.send_document(id, backup_input_file)
|
|
logger.info(f"Бэкап базы данных отправлен админу: {id}")
|
|
else:
|
|
await bot.send_document(admin_ids, backup_input_file)
|
|
logger.info(f"Бэкап базы данных отправлен админу: {ADMIN_ID}")
|
|
except Exception as e:
|
|
logger.error(f"Ошибка при отправке бэкапа в Telegram: {e}")
|
|
|
|
|
|
def _cleanup_old_backups():
|
|
try:
|
|
subprocess.run(
|
|
[
|
|
"find",
|
|
BACK_DIR,
|
|
"-type",
|
|
"f",
|
|
"-name",
|
|
"*.sql",
|
|
"-mtime",
|
|
"+3",
|
|
"-exec",
|
|
"rm",
|
|
"{}",
|
|
";",
|
|
],
|
|
check=True,
|
|
)
|
|
logger.info("Старые бэкапы удалены.")
|
|
except subprocess.CalledProcessError as e:
|
|
logger.error(f"Ошибка при удалении старых бэкапов: {e}")
|
|
|
|
|
|
async def create_backup_and_send_to_admins(xui):
|
|
await xui.login()
|
|
await xui.database.export()
|