Add clean-backups.py script
This commit is contained in:
parent
51b2fa2f1a
commit
f39ddea823
|
@ -0,0 +1,146 @@
|
||||||
|
#!/usr/bin/env python3
|
||||||
|
from calendar import week
|
||||||
|
from datetime import date, timedelta
|
||||||
|
import os
|
||||||
|
import argparse
|
||||||
|
|
||||||
|
# Default retention parameters
|
||||||
|
default_daily = 7
|
||||||
|
default_weekly = 4
|
||||||
|
default_monthly = 3
|
||||||
|
|
||||||
|
# Argument parser
|
||||||
|
parser = argparse.ArgumentParser(description="Cleanup old backups")
|
||||||
|
# path argument
|
||||||
|
parser.add_argument(
|
||||||
|
"path",
|
||||||
|
metavar = "PATH",
|
||||||
|
type = str,
|
||||||
|
nargs = 1,
|
||||||
|
help = "directory path"
|
||||||
|
)
|
||||||
|
# daily argument
|
||||||
|
parser.add_argument(
|
||||||
|
"-d", "--daily",
|
||||||
|
type = int,
|
||||||
|
default = default_daily,
|
||||||
|
metavar = "N",
|
||||||
|
help = "keep N daily backups, default: {default_daily}".format(
|
||||||
|
default_daily = default_daily
|
||||||
|
)
|
||||||
|
)
|
||||||
|
# weekly argument
|
||||||
|
parser.add_argument(
|
||||||
|
"-w", "--weekly",
|
||||||
|
type = int,
|
||||||
|
default = default_weekly,
|
||||||
|
metavar = "N",
|
||||||
|
help = "keep N weekly backups, default: {default_weekly}".format(
|
||||||
|
default_weekly = default_weekly
|
||||||
|
)
|
||||||
|
)
|
||||||
|
# monthly argument
|
||||||
|
parser.add_argument(
|
||||||
|
"-m", "--monthly",
|
||||||
|
type = int,
|
||||||
|
default = default_monthly,
|
||||||
|
metavar = "N",
|
||||||
|
help = "keep N monthly backups, default: {default_monthly}".format(
|
||||||
|
default_monthly = default_monthly
|
||||||
|
)
|
||||||
|
)
|
||||||
|
# force removal
|
||||||
|
parser.add_argument(
|
||||||
|
"-f", "--force",
|
||||||
|
action="store_true",
|
||||||
|
help = "suppress remove confirmation"
|
||||||
|
)
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
daily = args.daily
|
||||||
|
weekly = args.weekly
|
||||||
|
monthly = args.monthly
|
||||||
|
force = args.force
|
||||||
|
directory = args.path[0]
|
||||||
|
|
||||||
|
class backupFile():
|
||||||
|
def __init__(self, daily, weekly, monthly, path = None, dateformat = "%Y%m%d", dates = None,) -> None:
|
||||||
|
self.path = path
|
||||||
|
self.daily = daily
|
||||||
|
self.weekly = weekly
|
||||||
|
self.monthly = monthly
|
||||||
|
self.dateformat = dateformat
|
||||||
|
curr_date = date.today() # Maybe this will be used to specify date as starting point...
|
||||||
|
if self.path == None:
|
||||||
|
self.filename = None
|
||||||
|
else:
|
||||||
|
self.filename = os.path.basename(path)
|
||||||
|
if dates == None:
|
||||||
|
dates = []
|
||||||
|
|
||||||
|
# Daily
|
||||||
|
for i in range(0, daily):
|
||||||
|
day = curr_date - timedelta(days = i)
|
||||||
|
dates.append(day)
|
||||||
|
# Weekly
|
||||||
|
monday = curr_date - timedelta(days=date.weekday(curr_date))
|
||||||
|
for i in range(0,weekly):
|
||||||
|
day = monday - timedelta( days = (i * 7))
|
||||||
|
if day not in dates:
|
||||||
|
dates.append(day)
|
||||||
|
# Monthly
|
||||||
|
day = curr_date.replace(day=1)
|
||||||
|
for i in range(0,monthly):
|
||||||
|
if day not in dates:
|
||||||
|
dates.append(day)
|
||||||
|
day = (day - timedelta(days=1)).replace(day=1)
|
||||||
|
self.dates = dates
|
||||||
|
|
||||||
|
def newFile(self, path, daily = None, weekly = None, monthly = None, dateformat = None):
|
||||||
|
if daily == None: daily = self.daily
|
||||||
|
if weekly == None: weekly = self.weekly
|
||||||
|
if monthly == None: monthly = self.monthly
|
||||||
|
if dateformat == None: dateformat = self.dateformat
|
||||||
|
newfile = backupFile(daily, weekly, monthly, path, dates = self.dates)
|
||||||
|
return(newfile)
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
val = "<{path}>".format(
|
||||||
|
path = self.path,
|
||||||
|
)
|
||||||
|
return(val)
|
||||||
|
|
||||||
|
def needRemove(self):
|
||||||
|
if self.filename == None:
|
||||||
|
need_remove = False
|
||||||
|
else:
|
||||||
|
need_remove = True
|
||||||
|
for date in self.dates:
|
||||||
|
if date.strftime(self.dateformat) in self.filename:
|
||||||
|
need_remove = False
|
||||||
|
break
|
||||||
|
return(need_remove)
|
||||||
|
|
||||||
|
def remove(self, force = False):
|
||||||
|
print("Removing {file}...".format(file = self))
|
||||||
|
# Check force option
|
||||||
|
if force:
|
||||||
|
os.unlink(self.path)
|
||||||
|
else:
|
||||||
|
# Remove interactively
|
||||||
|
print("Are you sure? (y/n) ", end="")
|
||||||
|
answer = input()
|
||||||
|
if answer == "y":
|
||||||
|
os.unlink(self.path)
|
||||||
|
|
||||||
|
def removeIfNeeded(self, force = False):
|
||||||
|
if self.needRemove():
|
||||||
|
self.remove(force = force)
|
||||||
|
|
||||||
|
# File processing
|
||||||
|
files = backupFile(daily = daily, weekly = weekly, monthly = monthly)
|
||||||
|
# Generate file list with full paths
|
||||||
|
paths = [os.path.join(directory, f) for f in os.listdir(directory) if os.path.isfile(os.path.join(directory, f))]
|
||||||
|
for path in paths:
|
||||||
|
f = files.newFile(path)
|
||||||
|
f.removeIfNeeded(force = force)
|
Loading…
Reference in New Issue