diff --git a/bin/test-bulk-loader b/bin/test-bulk-loader index 64e76e1..ac11264 100755 --- a/bin/test-bulk-loader +++ b/bin/test-bulk-loader @@ -1,43 +1,40 @@ #!/usr/bin/env python3 import logging import sys import pygit2 from swh.core import config from swh.loader.git.git import BulkLoader DEFAULT_CONFIG = { 'db': ('str', 'dbname=softwareheritage-dev'), 'storage_base': ('str', '/tmp/swh-loader-git/test'), 'repo_path': ('str', None), 'origin': ('int', -1), 'authority': ('int', 1), 'validity': ('str', '2015-01-01 00:00:00+00'), 'create_origin': ('bool', True), 'send_contents': ('bool', True), 'send_directories': ('bool', True), 'send_revisions': ('bool', True), 'send_releases': ('bool', True), 'send_occurrences': ('bool', True), 'content_packet_size': ('int', 100000), 'directory_packet_size': ('int', 25000), 'revision_packet_size': ('int', 100000), 'release_packet_size': ('int', 100000), 'occurrence_packet_size': ('int', 100000), } logging.basicConfig(level=logging.INFO, format='%(asctime)s %(message)s') -my_config = {key: value for key, (type, value) in DEFAULT_CONFIG.items()} - -repo_path = pygit2.discover_repository(sys.argv[1]) -my_config['repo_path'] = repo_path +my_config = config.read(sys.argv[1], DEFAULT_CONFIG) loader = BulkLoader(my_config) loader.process()