Zakhar Izmaylov
a2c71a49f2
- Consolidated and simplified code formatting by removing unnecessary line breaks and improving readability in various functions. - Updated the handling of backup file sending in backup.py for better clarity. - Streamlined error handling and logging messages in several handlers to enhance consistency. - Adjusted the Makefile to exclude specific files during Ruff checks and formatting. - Made minor adjustments to function signatures and parameter handling for improved clarity and consistency. This commit enhances code maintainability and readability without altering functionality.
102 lines
3.1 KiB
Python
102 lines
3.1 KiB
Python
import os
|
|
import subprocess
|
|
from datetime import datetime
|
|
|
|
from aiogram.types import BufferedInputFile
|
|
|
|
from config import ADMIN_ID, BACK_DIR, DB_NAME, DB_PASSWORD, DB_USER, PG_HOST, PG_PORT
|
|
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 = PG_HOST
|
|
PORT = PG_PORT
|
|
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,
|
|
"-p",
|
|
PORT,
|
|
"-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: 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()
|