61 lines
2.5 KiB
Python
61 lines
2.5 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
import os
|
||
import hashlib
|
||
import datetime
|
||
import subprocess
|
||
import pysftp
|
||
import telebot
|
||
from dotenv import load_dotenv
|
||
|
||
# Load environment variables from .env file
|
||
load_dotenv()
|
||
|
||
# Set up variables from environment variables
|
||
MARIADB_HOST = os.getenv("MARIADB_HOST", default="localhost")
|
||
MARIADB_USER = os.getenv("MARIADB_USER", default="root")
|
||
MARIADB_PASSWORD = os.getenv("MARIADB_PASSWORD", default="passwd")
|
||
MARIADB_DATABASE = os.getenv("MARIADB_DATABASE", default="DBName")
|
||
SQLITE_DATABASE = os.getenv("SQLITE_DATABASE", default="/Path/to/database")
|
||
SFTP_HOST = os.getenv("SFTP_HOST", default="localhost")
|
||
SFTP_USER = os.getenv("SFTP_USER", default="user")
|
||
SFTP_PASSWORD = os.getenv("SFTP_PASSWORD", default="passwd")
|
||
SFTP_PATH = os.getenv("SFTP_PATH", default="/Paht/to/sftp")
|
||
SFTP_PORT = os.getenv("SFTP_PORT", default="22")
|
||
TELEGRAM_BOT_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN", default="TOKEN")
|
||
TELEGRAM_CHAT_ID = os.getenv("TELEGRAM_CHAT_ID", default="ID_CHAT")
|
||
|
||
# Generate backups
|
||
start_time = datetime.datetime.now()
|
||
mariadb_backup_filename = f"mariadb_backup_{datetime.datetime.now().strftime('%Y%m%d')}.sql.gz"
|
||
subprocess.run(f"mariadb-dump -h {MARIADB_HOST} -u {MARIADB_USER} -p{MARIADB_PASSWORD} {MARIADB_DATABASE} | gzip > {mariadb_backup_filename}", shell=True, check=True)
|
||
sqlite_backup_filename = f"sqlite_backup_{datetime.datetime.now().strftime('%Y%m%d')}.sql.gz"
|
||
subprocess.run(f"sqlite3 {SQLITE_DATABASE} .dump | gzip > {sqlite_backup_filename}", shell=True, check=True)
|
||
|
||
# Calculate SHA512 hash of backups
|
||
mariadb_hash = hashlib.sha512(open(mariadb_backup_filename, "rb").read()).hexdigest()
|
||
sqlite_hash = hashlib.sha512(open(sqlite_backup_filename, "rb").read()).hexdigest()
|
||
|
||
# Transfer backups to SFTP server
|
||
with pysftp.Connection(host=SFTP_HOST, username=SFTP_USER, password=SFTP_PASSWORD, port=SFTP_PORT) as sftp:
|
||
sftp.put(mariadb_backup_filename, f"{SFTP_PATH}/{mariadb_backup_filename}")
|
||
sftp.put(sqlite_backup_filename, f"{SFTP_PATH}/{sqlite_backup_filename}")
|
||
|
||
# Delete local backup files
|
||
os.remove(mariadb_backup_filename)
|
||
os.remove(sqlite_backup_filename)
|
||
|
||
# Send message via Telegram
|
||
bot = telebot.TeleBot(TELEGRAM_BOT_TOKEN)
|
||
message = f"""
|
||
Respaldo completado!
|
||
|
||
* **MariaDB:** *{mariadb_backup_filename}* (hash: {mariadb_hash})
|
||
* **SQLite:** ️ *{sqlite_backup_filename}* (hash: {sqlite_hash})
|
||
|
||
⏱️ Tiempo de transferencia: {datetime.datetime.now() - start_time}
|
||
|
||
Disponible en SFTP: {SFTP_PATH}
|
||
"""
|
||
bot.send_message(TELEGRAM_CHAT_ID, message) |