63 lines
2.0 KiB
Python
Executable File
63 lines
2.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
import logging
|
|
import os
|
|
import os.path
|
|
import sys
|
|
|
|
from dsc import SeafileClient, const
|
|
from dsc.misc import setup_uid, create_dir
|
|
|
|
_lg = logging.getLogger('dsc')
|
|
|
|
|
|
def main():
|
|
logging.basicConfig(format="%(asctime)s %(message)s", level=logging.INFO)
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--uid", default=os.getenv("SEAFILE_UID", default=1000), type=int)
|
|
parser.add_argument("--gid", default=os.getenv("SEAFILE_GID", default=1000), type=int)
|
|
parser.add_argument("--host", default=os.getenv("SERVER_HOST"))
|
|
parser.add_argument("--username", default=os.getenv("USERNAME"))
|
|
parser.add_argument("--password", default=os.getenv("PASSWORD"))
|
|
parser.add_argument("--libs", default=os.getenv("LIBRARY_ID"))
|
|
args = parser.parse_args()
|
|
|
|
setup_uid(args.uid, args.gid)
|
|
create_dir(const.DEFAULT_APP_DIR)
|
|
|
|
client = SeafileClient(args.host, args.username, args.password, const.DEFAULT_APP_DIR)
|
|
client.init_config()
|
|
client.start_daemon()
|
|
|
|
libs_to_sync = set()
|
|
for arg_lib in args.libs.split(sep=":"):
|
|
lib_id = client.get_library_id(arg_lib)
|
|
if lib_id:
|
|
libs_to_sync.add(lib_id)
|
|
else:
|
|
_lg.warning("Library %s is not found on server %s", arg_lib, args.host)
|
|
|
|
# don't start to sync libraries already in sync
|
|
libs_to_sync -= client.get_local_libraries()
|
|
|
|
# check for deprecated /data directory
|
|
if os.path.isdir(const.DEPRECATED_LIBS_DIR):
|
|
_lg.warning("*** DEPRECATED DIRECTORY FOUND ***")
|
|
_lg.warning("Deprecated directory %s is found, please mount your host directory with"
|
|
" libraries to %s instead", const.DEPRECATED_LIBS_DIR, const.DEFAULT_LIBS_DIR)
|
|
libs_dir = const.DEPRECATED_LIBS_DIR
|
|
else:
|
|
libs_dir = const.DEFAULT_LIBS_DIR
|
|
|
|
for lib_id in libs_to_sync:
|
|
client.sync_lib(lib_id, libs_dir)
|
|
client.watch_status()
|
|
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|