def backup(self):
if not os.path.exists(self.host_dir):
- print("New host",host,".")
- os.makedir(self.host_dir)
+ print("New host",self.name,".")
+ os.makedirs(self.host_dir)
if not os.path.exists(self.subvol_dir):
try:
check_call(shlex.split("btrfs subvol create %s"% self.subvol_dir))
if self.keep == -1:
print("No keep specified for %s, keeping all"%self.name)
return
-
+ if not os.path.exists(self.host_dir):
+ print("New host, no pruning needed")
+ return
snaps = sorted([snap for snap in os.listdir(self.host_dir) if not snap == "latest" ], reverse=True)
while len(snaps) > self.keep:
snap = snaps.pop()
pass
class BackupRunner():
- def __init__(self, config_dir, dest_dir):
+ def __init__(self, config_dir):
self.config_dir = config_dir
- self.dest_dir = dest_dir
if not os.path.exists(self.config_dir):
print("No config found", self.config_dir)
sys-exit(-1)
- def run(self):
- self.hosts = os.listdir(self.config_dir)
+ def run(self, hostlist=None):
+ self.hosts = hostlist or os.listdir(self.config_dir)
for host in self.hosts:
if host == 'default.cfg':
continue
try:
+ configfile = os.path.join(self.config_dir, host)
+
+ if not os.path.exists(configfile):
+ # Trigger logging in the except clause
+ raise BaseException()
+
config = configparser.ConfigParser(strict=False)
config.read_file(open(os.path.join(self.config_dir, 'default.cfg'),'r'))
- config.read(os.path.join(self.config_dir, host))
+ config.read(configfile)
except BaseException as ex:
print("Config error for %s. Skipping host."%host)
continue
if os.geteuid() != 0:
print("You need to be root. Otherwise all permissions will be lost.")
sys.exit(-1)
- br = BackupRunner("/etc/butterbackup", "/mnt/data2")
- br.run()
+ br = BackupRunner("/etc/butterbackup")
+
+ hostlist = sys.argv[1:]
+ br.run(hostlist=hostlist)
+
sys.exit(0)
+