from internal.collection import get_collections, get_collection, get_guessed_labels, get_count_all from internal.pre_request_filters import make_clientid_param, make_initiate_file_upload_params, async_source, \ make_s3_query_params, make_s3csv_query_params, make_rds_params, make_user_id_param, make_email_param, \ decode_alliance_workspace_id_params, appsync_field_param from internal.post_request_filters import sign_upload_url, convert_chart_data_to_graphql, convert_filters_to_graphql, \ convert_analytics_groups_to_graphql, convert_filter_groups_to_graphql, \ convert_collections_to_graphql, convert_pelias_response from internal.queries import http_query, s3_query, s3csv_query, rds_ep_query, channel_id_query, exec_endpoint, \ test_error_query, decide_query from internal.helpers import get_schema_id, rds_query, make_iso_date_now, get_user_id from management.pre_request_filters import make_allianceid_param def make_internal_endpoints(DP_URL): from airflow.progress import get_enrichment_progress_endpoint return { "dataprocessor/initiate_file_upload": {"url": DP_URL + "initiate_file_upload", "method": "POST", "pre_filters": [make_user_id_param, make_clientid_param, make_initiate_file_upload_params], "rights_check": ['max_unique_fan_count'], "query": http_query, "post_filters": [sign_upload_url] }, "dataprocessor/process_file": {"url": DP_URL + "async_notify", "method": "POST", "pre_filters": [async_source, decode_alliance_workspace_id_params, make_user_id_param, make_clientid_param], "query": http_query, "post_filters": [convert_collections_to_graphql], }, "dataprocessor/delete_collection": {"url": DP_URL + "async_notify", "method": "POST", "pre_filters": [async_source, decode_alliance_workspace_id_params, make_clientid_param, make_user_id_param], "query": http_query, "post_filters": [convert_collections_to_graphql], }, "dataprocessor/rename_collection": {"url": DP_URL + "rename_collection", "method": "POST", "pre_filters": [decode_alliance_workspace_id_params, make_clientid_param, make_user_id_param, make_allianceid_param], "query": http_query, }, "dataprocessor/get_analytics": {"url": DP_URL + "get_analytics", "method": "POST", "pre_filters": [make_clientid_param], "query": http_query, "post_filters": [convert_chart_data_to_graphql], }, "dataprocessor/get_grouped_analytics": {"url": DP_URL + "get_grouped_analytics", "method": "POST", "pre_filters": [decode_alliance_workspace_id_params, make_clientid_param, make_allianceid_param], "query": http_query, "post_filters": [convert_analytics_groups_to_graphql], }, "dataprocessor/get_filters": {"url": DP_URL + "get_filters", "method": "POST", "pre_filters": [decode_alliance_workspace_id_params, make_allianceid_param, make_clientid_param, make_user_id_param], "query": http_query, "post_filters": [convert_filters_to_graphql], }, "dataprocessor/get_grouped_filters": {"url": DP_URL + "get_grouped_filters", "method": "POST", "pre_filters": [decode_alliance_workspace_id_params, make_allianceid_param, make_clientid_param, make_user_id_param], "query": http_query, "post_filters": [convert_filter_groups_to_graphql], }, "dataprocessor/get_saved_filters": {"url": DP_URL + "get_saved_filters", "method": "POST", "pre_filters": [decode_alliance_workspace_id_params, make_allianceid_param, make_clientid_param, make_user_id_param], "query": http_query, }, "dataprocessor/apply_filters": {"url": DP_URL + "apply_filters", "method": "POST", "pre_filters": [decode_alliance_workspace_id_params, make_allianceid_param, make_clientid_param, make_user_id_param], "query": http_query, "post_filters": [convert_collections_to_graphql], }, "dataprocessor/save_segment": {"url": DP_URL + "async_notify", "method": "POST", "pre_filters": [async_source, decode_alliance_workspace_id_params, make_user_id_param, make_clientid_param], "query": http_query, "post_filters": [convert_collections_to_graphql], }, "dataprocessor/drop_current_schema": {"url": DP_URL + "drop_current_schema", "method": "POST", "pre_filters": [make_clientid_param, make_user_id_param, make_email_param], "query": http_query, }, "dataprocessor/export_csv": {"url": DP_URL + "export_csv", "method": "POST", "pre_filters": [decode_alliance_workspace_id_params, make_allianceid_param, make_clientid_param, make_user_id_param, make_email_param, make_initiate_file_upload_params], "query": http_query, }, "internal/get_collection": {"pre_filters": [decode_alliance_workspace_id_params, make_allianceid_param, make_clientid_param], "query": get_collection, "post_filters": [convert_collections_to_graphql], }, "internal/get_collections": {"pre_filters": [decode_alliance_workspace_id_params, make_allianceid_param, make_clientid_param], "query": get_collections, "post_filters": [convert_collections_to_graphql], }, "internal/get_guessed_labels": {"pre_filters": [decode_alliance_workspace_id_params, make_clientid_param], "query": get_guessed_labels, }, "internal/get_count_all": {"pre_filters": [decode_alliance_workspace_id_params, make_clientid_param], "query": get_count_all, }, "internal/get_delete_consequences": {"url": DP_URL + "delete_consequences", "pre_filters": [appsync_field_param, decode_alliance_workspace_id_params, make_user_id_param, make_allianceid_param, make_clientid_param], "query": http_query, }, "internal/enrichment/status": { "pre_filters": [ decode_alliance_workspace_id_params, make_allianceid_param, make_clientid_param ], "query": get_enrichment_progress_endpoint }, "pelias/autocomplete": { "url": "http://pelias.fansifter.cloud/v1/autocomplete", "method": "GET", "query": http_query, "post_filters": [convert_pelias_response] }, "S3": {"pre_filters": [make_s3_query_params], "query": s3_query }, "S3CSVRows": {"pre_filters": [decode_alliance_workspace_id_params, make_clientid_param, make_s3csv_query_params], "query": decide_query }, "RDS": {"pre_filters": [make_rds_params], "query": rds_ep_query }, "getChannelId": {"pre_filters": [decode_alliance_workspace_id_params, make_clientid_param], "query": channel_id_query}, "testError": {"query": test_error_query}, }