mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2024-09-20 08:13:12 +02:00
51 lines
2.4 KiB
Python
51 lines
2.4 KiB
Python
import psutil
|
|
import shutil
|
|
import os
|
|
import argparse
|
|
import sys
|
|
|
|
# Validating arguments
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--maximum-backup-size-percent', type=int, dest='maximum_backup_size_percent',required=True, choices=range(0,100), help="The directory from which the data should be encrypted.")
|
|
parser.add_argument('--backups-folder-path',type=str,dest='backups_folder_path',required=True, help="The folder in which the backups are stored")
|
|
args = parser.parse_args()
|
|
|
|
def print_used_disc_space():
|
|
print("%d %% of disk %s are used" % (psutil.disk_usage(args.backups_folder_path).percent,args.backups_folder_path))
|
|
|
|
warning_counter=0
|
|
|
|
for host_backup_directory_name in os.listdir(args.backups_folder_path):
|
|
host_backup_directory_path = os.path.join(args.backups_folder_path, host_backup_directory_name)
|
|
for application_directory in os.listdir(host_backup_directory_path):
|
|
|
|
# The directory which contains all backup versions of the application
|
|
versions_directory = os.path.join(host_backup_directory_path, application_directory) + "/"
|
|
|
|
versions = os.listdir(versions_directory)
|
|
versions.sort(reverse=False)
|
|
versions_counter=len(versions)
|
|
|
|
print_used_disc_space()
|
|
for version in versions:
|
|
version_path=os.path.join(versions_directory, version)
|
|
version_status_pulling_path=os.path.join(versions_directory, version, ".pulling")
|
|
print("Checking directory %s ..." % (version_path))
|
|
if psutil.disk_usage(args.backups_folder_path).percent > args.maximum_backup_size_percent:
|
|
if versions_counter >= 1:
|
|
print("Deleting %s to free space" % (version_path))
|
|
shutil.rmtree(version_path)
|
|
versions_counter-=1
|
|
else:
|
|
print("Deletion not possible. There needs to be at least one backup version")
|
|
warning_counter+=1
|
|
elif os.path.exists(version_status_pulling_path):
|
|
last_version=versions[-1]
|
|
if last_version != version:
|
|
print("Deleting %s due to unfinished pull" % (version_path))
|
|
shutil.rmtree(version_path)
|
|
versions_counter-=1
|
|
print_used_disc_space()
|
|
print("Cleaning up finished.")
|
|
sys.exit(warning_counter)
|