diff --git a/bin/swh-loader-git b/bin/swh-loader-git index 89d16e3..246bbbb 100755 --- a/bin/swh-loader-git +++ b/bin/swh-loader-git @@ -1,29 +1,38 @@ #!/usr/bin/env python3 import logging import sys +from swh.core.logger import PostgresHandler from swh.loader.git import BulkLoader ADDITIONAL_CONFIG = { 'repo_path': ('str', None), 'origin_url': ('str', 'file:///dev/null'), 'authority': ('int', 1), 'validity': ('str', '2015-01-01 00:00:00+00'), } -logging.basicConfig(level=logging.DEBUG, - format='%(asctime)s %(name)s %(levelname)s %(message)s') +my_config = BulkLoader.parse_config_file(config_filename=sys.argv[1], + additional_configs=[ADDITIONAL_CONFIG]) + +logging.basicConfig( + level=logging.DEBUG, + format='%(asctime)s %(name)s %(levelname)s %(message)s', + handlers=[ + logging.StreamHandler(), + PostgresHandler(my_config['log_db']), + ], +) requests_log = logging.getLogger("requests") requests_log.setLevel(logging.CRITICAL) -my_config = BulkLoader.parse_config_file(config_filename=sys.argv[1], - additional_configs=[ADDITIONAL_CONFIG]) + loader = BulkLoader(my_config) loader.process(my_config['repo_path'], my_config['origin_url'], my_config['authority'], my_config['validity'])