This commit is contained in:
+19
-5
@@ -1,12 +1,13 @@
|
||||
from argparse import ArgumentDefaultsHelpFormatter, ArgumentParser
|
||||
from dataclasses import dataclass
|
||||
from api import get_logger, get_api_config
|
||||
from typing import List
|
||||
from api import MAPPING, EndPointNotAvailableException, Server, get_logger, get_api_config
|
||||
|
||||
|
||||
parser = ArgumentParser(formatter_class=ArgumentDefaultsHelpFormatter)
|
||||
parser.add_argument("--verbose", "-v", action="count", default=0)
|
||||
parser.add_argument("--config", "-c", default="kontor-api")
|
||||
parser.add_argument("--dry-run", "-m", action="store_true")
|
||||
parser.add_argument("--server", "-s")
|
||||
parser.add_argument("--cleanup", "-d", action="store_true")
|
||||
args = parser.parse_args()
|
||||
|
||||
@@ -15,7 +16,20 @@ if __name__== "__main__":
|
||||
logger = get_logger(args.verbose, "kontor")
|
||||
logger.info("kontor.sync started")
|
||||
apiConfig = get_api_config(logger, args.config)
|
||||
for server in apiConfig.server:
|
||||
data = server.request(logger, "media_file")
|
||||
logger.info(len(data))
|
||||
server_list: List[Server] = []
|
||||
if args.server:
|
||||
server = apiConfig.get_server(args.server)
|
||||
if server:
|
||||
server_list.append(server)
|
||||
else:
|
||||
server_list.extend(apiConfig.server)
|
||||
for server in server_list:
|
||||
for table, path in MAPPING.items():
|
||||
try:
|
||||
data = server.request(logger, table=table)
|
||||
logger.info("%s: %s", table, len(data))
|
||||
if len(data) == 1:
|
||||
logger.info("show data: %s", data)
|
||||
except EndPointNotAvailableException:
|
||||
logger.info("Endpoint not implemented")
|
||||
logger.info("kontor.sync finished")
|
||||
|
||||
Reference in New Issue
Block a user