Fix retention period of child files

Cosmetic changes
This commit is contained in:
Сергей Морозов 2022-05-31 08:48:22 +03:00
parent d38e1613c3
commit 1709fd5974
1 changed files with 73 additions and 74 deletions

View File

@ -9,6 +9,79 @@ default_daily = 7
default_weekly = 4 default_weekly = 4
default_monthly = 3 default_monthly = 3
class backupFile():
def __init__(self, daily, weekly, monthly, path = None, dateformat = "%Y%m%d") -> 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)
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)
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)
# Argument parser # Argument parser
parser = argparse.ArgumentParser(description="Cleanup old backups") parser = argparse.ArgumentParser(description="Cleanup old backups")
# path argument # path argument
@ -63,80 +136,6 @@ monthly = args.monthly
force = args.force force = args.force
directory = args.path[0] 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 # File processing
files = backupFile(daily = daily, weekly = weekly, monthly = monthly) files = backupFile(daily = daily, weekly = weekly, monthly = monthly)
# Generate file list with full paths # Generate file list with full paths