Compare commits

...

10 Commits

3 changed files with 28 additions and 13 deletions

View File

@ -5,6 +5,7 @@ import os
import re import re
import logging import logging
import logging.handlers import logging.handlers
import stat
import subprocess import subprocess
import sys import sys
@ -258,7 +259,7 @@ def weed_snapshots(fslist, snapshots, config, failed_snapshots):
continue continue
if source_fs not in snapshots: if source_fs not in snapshots:
continue continue
if not conf['weed_enable']: if not conf['weed_enable'] or conf['weed_enable'].lower() in ('false', 'no'):
continue continue
kwargs = {k: int(v) for k, v in conf.items() if k in [ kwargs = {k: int(v) for k, v in conf.items() if k in [
@ -288,7 +289,8 @@ def weed_snapshots(fslist, snapshots, config, failed_snapshots):
def main(): def main():
config = configparser.ConfigParser() config = configparser.ConfigParser()
config.read('/etc/zsnapper.ini') config.read(['/usr/local/etc/zsnapper.ini', '/etc/zsnapper.ini'])
ret = RET_CODES['SUCCESS'] ret = RET_CODES['SUCCESS']
log = logging.getLogger(LOGGER) log = logging.getLogger(LOGGER)
@ -367,9 +369,9 @@ def main():
try: try:
if remote in remote_snapshots: if remote in remote_snapshots:
remote_snapshots[remote] = zsnaplib.get_snapshots(zfs_cmd) remote_snapshots[remote] = zsnaplib.get_snapshots(zfs_cmd)
except zsnaplib.ZFSSnapshotError: except zsnaplib.ZFSSnapshotError as e:
del remote_snapshots[remote] del remote_snapshots[remote]
log.warning("Could not refresh snapshots on {}".format(remote)) log.warning("Could not refresh snapshots on {}: {}".format(remote, e))
snapshots = zsnaplib.get_snapshots(local_zfs_cmd) snapshots = zsnaplib.get_snapshots(local_zfs_cmd)
failed_send = send_snapshots(fslist, snapshots, config) failed_send = send_snapshots(fslist, snapshots, config)
@ -403,9 +405,22 @@ if __name__ == '__main__':
handler.setLevel(logging.WARNING) handler.setLevel(logging.WARNING)
log.addHandler(handler) log.addHandler(handler)
handler = logging.handlers.SysLogHandler(address='/dev/log') handler = None
formatter = logging.Formatter(fmt='zsnapper[%(process)s] %(message)s') for logsocket in ('/var/run/log', '/dev/log'):
handler.setFormatter(formatter) try:
handler.setLevel(logging.INFO) mode = os.stat(logsocket).st_mode
log.addHandler(handler) except FileNotFoundError:
continue
if stat.S_ISSOCK(mode):
handler = logging.handlers.SysLogHandler(address=logsocket)
formatter = logging.Formatter(fmt='zsnapper[%(process)s] %(message)s')
handler.setFormatter(formatter)
handler.setLevel(logging.INFO)
log.addHandler(handler)
break
if not handler:
log.warning('No syslog socket found, will not log to syslog')
sys.exit(main()) sys.exit(main())

View File

@ -6,7 +6,7 @@ try:
except ImportError: except ImportError:
from distutils import setup from distutils import setup
version = '0.3.1' version = '0.3.4'
setup( setup(
name='zsnapper', name='zsnapper',

View File

@ -37,7 +37,7 @@ def do_zfs_command(args, zfs_cmd, pipecmd=None):
# check if we try to create a snapshot that already exists. This can happen # check if we try to create a snapshot that already exists. This can happen
# if the script is run every minute and it takes more time than that to # if the script is run every minute and it takes more time than that to
# create all snapshots # create all snapshots
if ctrl_proc.returncode == 1 and re_match(re_err_ok, err): if ctrl_proc.returncode == 1 and re.match(re_err_ok, err.decode('utf-8')):
pass pass
elif ctrl_proc.returncode != 0: elif ctrl_proc.returncode != 0:
raise ZFSSnapshotError('Failed to execute {}: {}'.format(cmd, err)) raise ZFSSnapshotError('Failed to execute {}: {}'.format(cmd, err))
@ -83,7 +83,7 @@ def get_filesystems(zfs_cmd):
for row in out.splitlines(): for row in out.splitlines():
row = row.decode('UTF-8') row = row.decode('UTF-8')
ret.add(row.split()[0]) ret.add(row.split('\t')[0])
return ret return ret
@ -93,7 +93,7 @@ def get_snapshots(zfs_cmd):
snapshots = {} snapshots = {}
for row in out.splitlines(): for row in out.splitlines():
row = row.decode('UTF-8').split()[0] row = row.decode('UTF-8').split('\t')[0]
res = re_snapshot.match(row) res = re_snapshot.match(row)
if res: if res:
d = datetime.datetime.strptime(res.group(2), time_format) d = datetime.datetime.strptime(res.group(2), time_format)