sqlbackups/respaldosql.py
2024-02-13 16:05:15 -05:00

61 lines
2.5 KiB
Python
Raw Blame History

This file contains invisible Unicode characters

This file contains invisible Unicode characters that are indistinguishable to humans but may be processed differently by a computer. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

#!/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)