镜像自地址
https://github.com/tuna/tunasync.git
已同步 2025-12-06 22:46:47 +00:00
143 行
4.6 KiB
Python
143 行
4.6 KiB
Python
#!/usr/bin/env python2
|
|
# -*- coding:utf-8 -*-
|
|
import os
|
|
from datetime import datetime
|
|
from .mirror_provider import RsyncProvider, ShellProvider
|
|
from .btrfs_snapshot import BtrfsHook
|
|
from .loglimit import LogLimitHook
|
|
from .exec_pre_post import CmdExecHook
|
|
|
|
|
|
class MirrorConfig(object):
|
|
|
|
_valid_providers = set(("rsync", "debmirror", "shell", ))
|
|
|
|
def __init__(self, parent, options):
|
|
self._parent = parent
|
|
self._popt = self._parent._settings
|
|
self.options = dict(options.items()) # copy
|
|
self._validate()
|
|
|
|
def _validate(self):
|
|
provider = self.options.get("provider", None)
|
|
assert provider in self._valid_providers
|
|
|
|
if provider == "rsync":
|
|
assert "upstream" in self.options
|
|
|
|
elif provider == "shell":
|
|
assert "command" in self.options
|
|
|
|
local_dir_tmpl = self.options.get(
|
|
"local_dir", self._popt["global"]["local_dir"])
|
|
|
|
self.options["local_dir"] = local_dir_tmpl.format(
|
|
mirror_root=self._popt["global"]["mirror_root"],
|
|
mirror_name=self.name,
|
|
)
|
|
|
|
if "interval" not in self.options:
|
|
self.options["interval"] = self._popt["global"]["interval"]
|
|
|
|
assert isinstance(self.options["interval"], int)
|
|
|
|
log_dir = self.options.get(
|
|
"log_dir", self._popt["global"]["log_dir"])
|
|
if "log_file" not in self.options:
|
|
self.options["log_file"] = os.path.join(
|
|
log_dir, self.name, self.name + "_{date}.log")
|
|
|
|
self.log_dir = os.path.dirname(self.log_file)
|
|
|
|
if "use_btrfs" not in self.options:
|
|
self.options["use_btrfs"] = self._parent.use_btrfs
|
|
assert self.options["use_btrfs"] in (True, False)
|
|
|
|
def __getattr__(self, key):
|
|
if key in self.__dict__:
|
|
return self.__dict__[key]
|
|
else:
|
|
return self.__dict__["options"].get(key, None)
|
|
|
|
def to_provider(self, hooks=[], no_delay=False):
|
|
if self.provider == "rsync":
|
|
provider = RsyncProvider(
|
|
name=self.name,
|
|
upstream_url=self.upstream,
|
|
local_dir=self.local_dir,
|
|
log_dir=self.log_dir,
|
|
useIPv6=self.use_ipv6,
|
|
password=self.password,
|
|
exclude_file=self.exclude_file,
|
|
log_file=self.log_file,
|
|
interval=self.interval,
|
|
hooks=hooks,
|
|
)
|
|
elif self.options["provider"] == "shell":
|
|
provider = ShellProvider(
|
|
name=self.name,
|
|
command=self.command,
|
|
upstream_url=self.upstream,
|
|
local_dir=self.local_dir,
|
|
log_dir=self.log_dir,
|
|
log_file=self.log_file,
|
|
log_stdout=self.options.get("log_stdout", True),
|
|
interval=self.interval,
|
|
hooks=hooks
|
|
)
|
|
|
|
if not no_delay:
|
|
sm = self._parent.status_manager
|
|
last_update = sm.get_info(self.name, 'last_update')
|
|
if last_update not in (None, '-'):
|
|
last_update = datetime.strptime(last_update,
|
|
'%Y-%m-%d %H:%M:%S')
|
|
delay = int(last_update.strftime("%s")) \
|
|
+ self.interval * 60 - int(datetime.now().strftime("%s"))
|
|
if delay < 0:
|
|
delay = 0
|
|
provider.set_delay(delay)
|
|
|
|
return provider
|
|
|
|
def compare(self, other):
|
|
assert self.name == other.name
|
|
|
|
for key, val in self.options.iteritems():
|
|
if other.options.get(key, None) != val:
|
|
return False
|
|
|
|
return True
|
|
|
|
def hooks(self):
|
|
hooks = []
|
|
parent = self._parent
|
|
if self.options["use_btrfs"]:
|
|
working_dir = parent.btrfs_working_dir_tmpl.format(
|
|
mirror_root=parent.mirror_root,
|
|
mirror_name=self.name
|
|
)
|
|
service_dir = parent.btrfs_service_dir_tmpl.format(
|
|
mirror_root=parent.mirror_root,
|
|
mirror_name=self.name
|
|
)
|
|
gc_dir = parent.btrfs_gc_dir_tmpl.format(
|
|
mirror_root=parent.mirror_root,
|
|
mirror_name=self.name
|
|
)
|
|
hooks.append(BtrfsHook(service_dir, working_dir, gc_dir))
|
|
|
|
hooks.append(LogLimitHook())
|
|
|
|
if self.exec_pre_sync:
|
|
hooks.append(
|
|
CmdExecHook(self.exec_pre_sync, CmdExecHook.PRE_SYNC))
|
|
|
|
if self.exec_post_sync:
|
|
hooks.append(
|
|
CmdExecHook(self.exec_post_sync, CmdExecHook.POST_SYNC))
|
|
|
|
return hooks
|
|
|
|
# vim: ts=4 sw=4 sts=4 expandtab
|