kopia lustrzana https://github.com/bugout-dev/moonstream
Add customer connection.
rodzic
dcc5eeb5e2
commit
08fab2c16c
|
@ -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)}"
|
f"Initial function call crawl jobs count: {len(initial_function_call_jobs)}"
|
||||||
)
|
)
|
||||||
|
|
||||||
customer_connection = get_db_connection(args.customer_uuid)
|
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 == "":
|
if customer_connection == "":
|
||||||
raise ValueError("No connection string found for the customer")
|
raise ValueError("No connection string found for the customer")
|
||||||
|
@ -984,10 +987,12 @@ def main() -> None:
|
||||||
)
|
)
|
||||||
|
|
||||||
historical_crawl_parser_v3.add_argument(
|
historical_crawl_parser_v3.add_argument(
|
||||||
"--address",
|
"--addresses",
|
||||||
"-a",
|
"-a",
|
||||||
required=False,
|
required=False,
|
||||||
|
nargs="*",
|
||||||
type=str,
|
type=str,
|
||||||
|
help="Addresses to crawl",
|
||||||
)
|
)
|
||||||
|
|
||||||
historical_crawl_parser_v3.add_argument(
|
historical_crawl_parser_v3.add_argument(
|
||||||
|
@ -1076,6 +1081,13 @@ def main() -> None:
|
||||||
help="Customer UUID",
|
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(
|
historical_crawl_parser_v3.add_argument(
|
||||||
"--user-uuid",
|
"--user-uuid",
|
||||||
type=UUID,
|
type=UUID,
|
||||||
|
|
Ładowanie…
Reference in New Issue