Backup/functions.py at english_version · lgadrolean/Backup · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
import os
import shutil
import zipfile
from datetime import datetime

def log_message(log_path, message):
    with open(log_path, 'a', encoding='utf-8') as log_file:
        log_file.write(f"{datetime.now().strftime('%Y-%m-%d %H:%M:%S')} - {message}\n")

def copytree(src, dst, log_path):
    os.makedirs(dst, exist_ok=True)
    for item in os.listdir(src):
        s = os.path.join(src, item)
        d = os.path.join(dst, item)
        if any(ignored in s.split(os.sep) for ignored in ['.git', '__pycache__']):
            log_message(log_path, f"Ignorando a pasta: {s}")
            continue
        try:
            if os.path.isdir(s):
                copytree(s, d, log_path)
            else:
                shutil.copy2(s, d)
        except Exception as e:
            log_message(log_path, f"Erro ao copiar {s} para {d}: {e}")

def create_backup(source_path, hot_backup_path, daily_backup_path, backupname, log_path, onedrive_path, keep_count_cloud):
    timestamp = datetime.now().strftime('%Y%m%d_%H%M')
    backup_name = f"{backupname}_{timestamp}.zip"
    backup_path = os.path.join(hot_backup_path, backup_name)

    log_message(log_path, "###########################################")
    log_message(log_path, "Iniciando o processo de backup.")

    temp_path = os.path.join(hot_backup_path, "temp_backup")
    copytree(source_path, temp_path, log_path)
    log_message(log_path, f"Pasta copiada de {source_path} para {temp_path}.")

    with zipfile.ZipFile(backup_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
        for root, dirs, files in os.walk(temp_path):
            dirs[:] = [d for d in dirs if '.git' not in os.path.join(root, d)]
            files = [f for f in files if '.git' not in os.path.join(root, f)]
            for file in files:
                try:
                    zipf.write(os.path.join(root, file), os.path.relpath(os.path.join(root, file), temp_path))
                except Exception as e:
                    log_message(log_path, f"Erro ao comprimir {file}: {e}")
    log_message(log_path, f"Pasta comprimida em {backup_path}.")
    shutil.rmtree(temp_path)
    log_message(log_path, f"Pasta temporária {temp_path} removida.")

    daily_backup_name = f"{backupname}_{datetime.now().strftime('%Y%m%d')}.zip"
    daily_backup_path_full = os.path.join(daily_backup_path, daily_backup_name)
    if not os.path.exists(daily_backup_path_full):
        shutil.copy2(backup_path, daily_backup_path_full)
        log_message(log_path, f"Backup diário movido para {daily_backup_path_full}.")

    onedrive_backup_path_full = os.path.join(onedrive_path, daily_backup_name)
    if not os.path.exists(onedrive_backup_path_full):
        shutil.copy2(backup_path, onedrive_backup_path_full)
        log_message(log_path, f"Backup diário movido para o OneDrive: {onedrive_backup_path_full}.")

    maintain_backups(onedrive_path, backupname, log_path, keep_count_cloud, "cloud")

    # Log existing backup statistics
    log_backup_statistics(daily_backup_path, hot_backup_path, onedrive_path, backupname, log_path)

def maintain_backups(backup_path, backupname, log_path, keep_count, tipo="local"):
    log_message(log_path, f"Verificando backups antigos para remoção em {backup_path}.")
    backups = sorted([f for f in os.listdir(backup_path) if f.startswith(backupname) and f.endswith(".zip")], key=lambda x: os.path.getmtime(os.path.join(backup_path, x)), reverse=True)
    for backup in backups[keep_count:]:
        os.remove(os.path.join(backup_path, backup))
        log_message(log_path, f"Backup {tipo} antigo removido: {backup}")

def log_backup_statistics(daily_backup_path, hot_backup_path, onedrive_path, backupname, log_path):
    daily_backups = [f for f in os.listdir(daily_backup_path) if f.startswith(backupname) and f.endswith(".zip")]
    hot_backups = [f for f in os.listdir(hot_backup_path) if f.startswith(backupname) and f.endswith(".zip")]
    cloud_backups = [f for f in os.listdir(onedrive_path) if f.startswith(backupname) and f.endswith(".zip")]

    log_message(log_path, f"Número de backups diários: {len(daily_backups)}")
    log_message(log_path, f"Número de backups quentes: {len(hot_backups)}")
    log_message(log_path, f"Número de backups na nuvem: {len(cloud_backups)}")