def query_userid_key(trace_id, user_id, organization_id=None, filter_expression=None, convert_response=None, is_cw_logger=False): if (is_cw_logger): logger = common_utils.begin_cw_logger(trace_id, __name__, inspect.currentframe()) else: logger = common_utils.begin_logger(trace_id, __name__, inspect.currentframe()) key_conditions = { 'UserID': { 'AttributeValueList': [user_id], 'ComparisonOperator': 'EQ' } } if (organization_id): key_conditions['OrganizationID'] = { 'AttributeValueList': [organization_id], 'ComparisonOperator': 'EQ' } result = DB_utils.query(trace_id, Tables.PM_AFFILIATION, key_conditions, filter_expression, is_cw_logger=is_cw_logger) if result and convert_response: result = common_utils.convert_list_response(trace_id, result, RESPONSE_AFFILIATION, is_cw_logger=is_cw_logger) return common_utils.response(result, logger)
def query_organization_id(trace_id, organization_id, filter_expression=None): pm_logger = common_utils.begin_logger(trace_id, __name__, inspect.currentframe()) key_conditions = { 'OrganizationID': { 'AttributeValueList': [organization_id], 'ComparisonOperator': 'EQ' } } result = DB_utils.query(trace_id, Tables.PM_LATEST_CHECK_RESULT, key_conditions, filter_expression) return common_utils.response(result, pm_logger)
def query(trace_id, security_check_webhook_id): pm_logger = common_utils.begin_logger(trace_id, __name__, inspect.currentframe()) key_conditions = { 'SecurityCheckWebhookID': { 'AttributeValueList': [security_check_webhook_id], 'ComparisonOperator': 'EQ' } } result = DB_utils.query(trace_id, Tables.PM_SECURITY_CHECK_WEBHOOK_CALL_HISTORY, key_conditions, None) return common_utils.response(result, pm_logger)
def get_projects(trace_id, project_id, filter_expression=None, convert_response=None): pm_logger = common_utils.begin_logger(trace_id, __name__, inspect.currentframe()) key_conditions = { 'ProjectID': { 'AttributeValueList': [project_id], 'ComparisonOperator': 'EQ' } } result = DB_utils.query(trace_id, Tables.PM_PROJECTS, key_conditions, filter_expression) if convert_response: result = common_utils.convert_list_response( trace_id, result, RESPONSE_PROJECTS) return common_utils.response(result, pm_logger)
def query_daily_check_key(trace_id, awsaccount, executed_date): pm_logger = common_utils.begin_logger(trace_id, __name__, inspect.currentframe()) key_conditions = { 'AWSAccount': { 'AttributeValueList': [awsaccount], 'ComparisonOperator': 'EQ' } } key_conditions['ExecutedDate'] = { 'AttributeValueList': [executed_date], 'ComparisonOperator': 'EQ' } result = DB_utils.query(trace_id, Tables.PM_DAILY_CHECK_EXECUTED_AWSACCOUNTS, key_conditions, None) return common_utils.response(result, pm_logger)
def query_key_by_check_history_id(trace_id, project_id, organization_id, check_history_id): pm_logger = common_utils.begin_logger(trace_id, __name__, inspect.currentframe()) filter = Attr('CheckHistoryID').eq(check_history_id) key_conditions = { 'OrganizationID': { 'AttributeValueList': [organization_id], 'ComparisonOperator': 'EQ' }, 'ProjectID': { 'AttributeValueList': [project_id], 'ComparisonOperator': 'EQ' } } result = DB_utils.query(trace_id, Tables.PM_LATEST_CHECK_RESULT, key_conditions, filter) return common_utils.response(result, pm_logger)
def get_check_history_by_status(trace_id, check_history_id, check_status, convert_response=None): pm_logger = common_utils.begin_logger(trace_id, __name__, inspect.currentframe()) key_conditions = { 'CheckHistoryID': { 'AttributeValueList': [check_history_id], 'ComparisonOperator': 'EQ' } } filter = Attr('CheckStatus').eq(check_status) result = DB_utils.query(trace_id, Tables.PM_CHECK_HISTORY, key_conditions, filter) if convert_response: result = common_utils.convert_response(trace_id, result, RESPONSE_HISTORY) return common_utils.response(result, pm_logger)
def query_awscoop(trace_id, coop_id, filter_expression=None, convert_response=None): pm_logger = common_utils.begin_logger(trace_id, __name__, inspect.currentframe()) key_conditions = { 'CoopID': { 'AttributeValueList': [coop_id], 'ComparisonOperator': 'EQ' } } result = DB_utils.query(trace_id, Tables.PM_AWSACCOUNTCOOPS, key_conditions, filter_expression) if convert_response: result = common_utils.convert_list_response(trace_id, result, RESPONSE_AWSCOOPS, RESPONSE_REQUIRED) return common_utils.response(result, pm_logger)
def query(user_id, organization_id, filter_expression=None, convert_response=None): pm_logger = common_utils.begin_logger(user_id, __name__, inspect.currentframe()) key_conditions = { 'UserID': { 'AttributeValueList': [user_id], 'ComparisonOperator': 'EQ' }, 'OrganizationID': { 'AttributeValueList': [organization_id], 'ComparisonOperator': 'EQ' } } result = DB_utils.query(user_id, Tables.PM_AFFILIATION, key_conditions, filter_expression) if result and convert_response: result = common_utils.convert_list_response(user_id, result, RESPONSE_AFFILIATION) return common_utils.response(result, pm_logger)