| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162 |
- #!/usr/bin/env python2
- # -*- coding:utf-8 -*-
- import sh
- import os
- from datetime import datetime
- from .hook import JobHook
- class BtrfsVolumeError(Exception):
- pass
- class BtrfsHook(JobHook):
- def __init__(self, service_dir, working_dir, gc_dir):
- self.service_dir = service_dir
- self.working_dir = working_dir
- self.gc_dir = gc_dir
- def before_job(self, ctx={}, *args, **kwargs):
- self._create_working_snapshot()
- ctx['current_dir'] = self.working_dir
- def after_job(self, status=None, ctx={}, *args, **kwargs):
- if status == "success":
- self._commit_changes()
- ctx['current_dir'] = self.service_dir
- def _ensure_subvolume(self):
- # print(self.service_dir)
- try:
- ret = sh.btrfs("subvolume", "show", self.service_dir)
- except Exception, e:
- print(e)
- raise BtrfsVolumeError("Invalid subvolume")
- if ret.stderr != '':
- raise BtrfsVolumeError("Invalid subvolume")
- def _create_working_snapshot(self):
- self._ensure_subvolume()
- if os.path.exists(self.working_dir):
- print("Warning: working dir existed, are you sure no rsync job is running?")
- else:
- # print("btrfs subvolume snapshot {} {}".format(self.service_dir, self.working_dir))
- sh.btrfs("subvolume", "snapshot", self.service_dir, self.working_dir)
- def _commit_changes(self):
- self._ensure_subvolume()
- self._ensure_subvolume()
- gc_dir = self.gc_dir.format(timestamp=datetime.now().strftime("%s"))
- out = sh.mv(self.service_dir, gc_dir)
- assert out.exit_code == 0 and out.stderr == ""
- out = sh.mv(self.working_dir, self.service_dir)
- assert out.exit_code == 0 and out.stderr == ""
- # print("btrfs subvolume delete {}".format(self.tmp_dir))
- # sh.sleep(3)
- # out = sh.btrfs("subvolume", "delete", self.tmp_dir)
- # assert out.exit_code == 0 and out.stderr == ""
- # vim: ts=4 sw=4 sts=4 expandtab
|
