start.py

#!/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("-s", "--server")
    parser.add_argument("-u", "--username")
    parser.add_argument("-p", "--password")
    parser.add_argument("-l", "--libraries")
    parser.add_argument("--uid", type=int)
    parser.add_argument("--gid", type=int)
    parser.add_argument("--upload-limit", type=int, default=0)
    parser.add_argument("--download-limit", type=int, default=0)
    parser.add_argument("--disable-verify-certificate", action="store_true")
    parser.add_argument("--delete-confirm-threshold", type=int, default=500)

    parser.set_defaults(
        server=os.getenv("SERVER_HOST"),
        username=os.getenv("USERNAME"),
        password=os.getenv("PASSWORD"),
        libraries=os.getenv("LIBRARY_ID"),
        uid=os.getenv("SEAFILE_UID", default=1000),
        gid=os.getenv("SEAFILE_GID", default=1000),
        upload_limit=os.getenv("UPLOAD_LIMIT"),
        download_limit=os.getenv("DOWNLOAD_LIMIT"),
        disable_verify_certificate=os.getenv("DISABLE_VERIFY_CERTIFICATE") in ("true", "1", "True"),
        delete_confirm_threshold=os.getenv("DELETE_CONFIRM_THRESHOLD"),
    )
    args = parser.parse_args()
    if not args.server:
        parser.error("Seafile server is not specified")
    if not args.username:
        parser.error("username is not specified")
    if not args.password:
        parser.error("password is not specified")
    if not args.libraries:
        parser.error("library is not specified")

    setup_uid(args.uid, args.gid)
    create_dir(const.DEFAULT_APP_DIR)

    client = SeafileClient(args.server, args.username, args.password, const.DEFAULT_APP_DIR)
    client.init_config()
    client.start_daemon()
    client.configure(args, check_for_daemon=False)

    libs_to_sync = set()
    for arg_lib in args.libraries.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.server)

    # 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()

    client.stop_daemon()
    return 0


if __name__ == "__main__":
    sys.exit(main())