Add customer uri.

pull/1119/head
Andrey 2024-08-09 00:23:23 +03:00
rodzic aadbfe46ac
commit d7c8b7f9d7
1 zmienionych plików z 11 dodań i 1 usunięć

Wyświetl plik

@ -198,7 +198,10 @@ def handle_crawl_v3(args: argparse.Namespace) -> None:
index_engine = MoonstreamDBIndexesEngine()
logger.info(f"Blockchain type: {blockchain_type.value}")
customer_connection = get_db_connection(args.customer_uuid)
if args.customer_uri is not None:
customer_connection = args.customer_uri
else:
customer_connection = get_db_connection(args.customer_uuid)
customer_engine = MoonstreamCustomDBEngine(customer_connection)
@ -892,6 +895,13 @@ def main() -> None:
help="Customer UUID",
)
crawl_parser_v3.add_argument(
"--customer-uri",
type=str,
required=False,
help="Customer URI",
)
crawl_parser_v3.set_defaults(func=handle_crawl_v3)
historical_crawl_parser = subparsers.add_parser(