from graphql_api.schema.file import FileInterface, CreateFile from graphql_api.schema.table import Table, TableType from graphql_api.schema.custom.rupture_generation_task import RuptureGenerationTask from graphql_api.schema.custom.common import KeyValuePair, KeyValuePairInput, KeyValueListPair, KeyValueListPairInput from importlib import import_module from graphql_api.data_s3 import get_data_manager from datetime import datetime as dt from graphql_api.config import STACK_NAME, CW_METRICS_RESOLUTION from graphql_api.cloudwatch import ServerlessMetricWriter db_metrics = ServerlessMetricWriter(lambda_name=STACK_NAME, metric_name="MethodDuration", resolution=CW_METRICS_RESOLUTION) def resolve_node(root, info, id_field, dm_type): """ Optimisation function, looks at the query and avoids a fetch if we only want to resolve the id field. """ t0 = dt.utcnow() assert dm_type in ["table", "thing"] node_id = getattr(root, id_field) if not node_id: return
from graphql_api.data_s3 import get_data_manager from .custom.general_task import GeneralTask, CreateGeneralTask, UpdateGeneralTask from .task_task_relation import CreateTaskTaskRelation from .table import CreateTable, Table from .custom.automation_task import AutomationTask, CreateAutomationTask, UpdateAutomationTask #from .custom.inversion_solution import from graphql_api.schema.custom.inversion_solution import InversionSolution, CreateInversionSolution, AppendInversionSolutionTables, LabelledTableRelationInput from graphql_api.cloudwatch import ServerlessMetricWriter from graphql_api.config import IS_OFFLINE, ES_REGION, ES_ENDPOINT, ES_INDEX, STACK_NAME db_metrics = ServerlessMetricWriter(lambda_name=STACK_NAME, metric_name="MethodDuration", resolution=1) credentials = boto3.Session().get_credentials() if not IS_OFFLINE else None awsauth = AWS4Auth(credentials.access_key, credentials.secret_key, ES_REGION, 'es', session_token=credentials.token) if not IS_OFFLINE else None s3_client_args = dict( aws_access_key_id='S3RVER', aws_secret_access_key='S3RVER', endpoint_url='http://localhost:4569') if IS_OFFLINE else {} search_manager = SearchManager(endpoint=ES_ENDPOINT,