Add customer connection.

multiple-addresses
Andrey 2024-07-18 14:55:11 +03:00
rodzic dcc5eeb5e2
commit 08fab2c16c
1 zmienionych plików z 14 dodań i 2 usunięć

Wyświetl plik

@ -548,7 +548,10 @@ def handle_historical_crawl_v3(args: argparse.Namespace) -> None:
f"Initial function call crawl jobs count: {len(initial_function_call_jobs)}"
)
if args.customer_db_uri is None:
customer_connection = get_db_connection(args.customer_uuid)
else:
customer_connection = args.customer_db_uri
if customer_connection == "":
raise ValueError("No connection string found for the customer")
@ -984,10 +987,12 @@ def main() -> None:
)
historical_crawl_parser_v3.add_argument(
"--address",
"--addresses",
"-a",
required=False,
nargs="*",
type=str,
help="Addresses to crawl",
)
historical_crawl_parser_v3.add_argument(
@ -1076,6 +1081,13 @@ def main() -> None:
help="Customer UUID",
)
historical_crawl_parser_v3.add_argument(
"--customer-db-uri",
type=str,
required=False,
help="Customer database URI",
)
historical_crawl_parser_v3.add_argument(
"--user-uuid",
type=UUID,