86 lines
No EOL
3 KiB
Python
86 lines
No EOL
3 KiB
Python
#!/usr/bin/env python3
|
|
"""Dump automatique de la base MariaDB.
|
|
Execute chaque nuit via cron dans le conteneur Flask.
|
|
Garde les 7 derniers dumps (rotation).
|
|
"""
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
import gzip
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
|
|
DUMP_DIR = Path("/backups")
|
|
DUMP_DIR.mkdir(parents=True, exist_ok=True)
|
|
|
|
TIMESTAMP = datetime.now().strftime("%Y%m%d_%H%M%S")
|
|
DUMP_FILE = DUMP_DIR / f"gmao_db_{TIMESTAMP}.sql.gz"
|
|
|
|
DB_HOST = os.environ.get("DB_HOST", "mariadb")
|
|
DB_PORT = os.environ.get("DB_PORT", "3306")
|
|
DB_NAME = os.environ.get("MARIADB_DATABASE", "gmao_db")
|
|
DB_USER = os.environ.get("MARIADB_USER", "gmao")
|
|
DB_PASS = os.environ.get("MARIADB_PASSWORD", "gmao123")
|
|
ROOT_PASS = os.environ.get("MARIADB_ROOT_PASSWORD", "root123")
|
|
|
|
# Utiliser pymysql pour exporter le schema + donnees
|
|
try:
|
|
import pymysql
|
|
from pymysql.cursors import DictCursor
|
|
|
|
conn = pymysql.connect(
|
|
host=DB_HOST, port=int(DB_PORT),
|
|
user="root", password=ROOT_PASS,
|
|
database=DB_NAME, charset="utf8mb4"
|
|
)
|
|
cursor = conn.cursor()
|
|
|
|
# Obtenir la liste des tables
|
|
cursor.execute("SHOW TABLES")
|
|
tables = [row[0] for row in cursor.fetchall()]
|
|
|
|
with gzip.open(str(DUMP_FILE), "wt", encoding="utf-8") as f:
|
|
f.write(f"-- Dump {DB_NAME} - {datetime.now().isoformat()}\n")
|
|
f.write("SET FOREIGN_KEY_CHECKS=0;\n\n")
|
|
|
|
for table in tables:
|
|
# Schema
|
|
cursor.execute(f"SHOW CREATE TABLE `{table}`")
|
|
create_sql = cursor.fetchone()[1]
|
|
f.write(f"DROP TABLE IF EXISTS `{table}`;\n")
|
|
f.write(f"{create_sql};\n\n")
|
|
|
|
# Donnees
|
|
cursor.execute(f"SELECT * FROM `{table}`")
|
|
rows = cursor.fetchall()
|
|
if rows:
|
|
cols = [d[0] for d in cursor.description]
|
|
col_names = ", ".join(f"`{c}`" for c in cols)
|
|
for row in rows:
|
|
values = []
|
|
for val in row:
|
|
if val is None:
|
|
values.append("NULL")
|
|
elif isinstance(val, (int, float)):
|
|
values.append(str(val))
|
|
else:
|
|
escaped = str(val).replace("\\", "\\\\").replace("'", "\\'")
|
|
values.append(f"'{escaped}'")
|
|
f.write(f"INSERT INTO `{table}` ({col_names}) VALUES ({', '.join(values)});\n")
|
|
f.write("\n")
|
|
|
|
f.write("SET FOREIGN_KEY_CHECKS=1;\n")
|
|
|
|
conn.close()
|
|
|
|
# Rotation : garder les 7 derniers
|
|
dumps = sorted(DUMP_DIR.glob("gmao_db_*.sql.gz"), key=lambda p: p.stat().st_mtime, reverse=True)
|
|
for old in dumps[7:]:
|
|
old.unlink()
|
|
|
|
size = DUMP_FILE.stat().st_size / 1024
|
|
print(f"{datetime.now()}: Dump cree: {DUMP_FILE} ({size:.0f} KB, {len(tables)} tables)")
|
|
|
|
except Exception as e:
|
|
print(f"{datetime.now()}: ERREUR dump: {e}", file=sys.stderr)
|
|
sys.exit(1) |