def get_default_profile(database): """Returns the current profile by name. """ collection = get_profile_collection(database) arguments = [] arguments.append(mongo_utils.make_single_field_argument('default', True)) arguments.append( mongo_utils.make_single_field_argument('remove_date', None)) argument = general_utils.merge_list_of_dicts(arguments) cursor = mongo_utils.mongo_find_records(collection, argument=argument) profile_list = mongo_utils.unload_cursor(cursor) try: return profile_list[0] except IndexError: return None
def get_current_translators(database): """Returns all the translators that haven't been removed. """ collection = get_translator_collection(database) argument = mongo_utils.make_single_field_argument("remove_date", None) cursor = mongo_utils.mongo_find_records(collection, argument=argument) return mongo_utils.unload_cursor(cursor)
def get_current_records(database, structure): """Returns all the records that haven't been removed for a given structure. """ collection = get_record_collection(database, structure) argument = mongo_utils.make_single_field_argument("remove_date", None) cursor = mongo_utils.mongo_find_records(collection, argument=argument) return mongo_utils.unload_cursor(cursor)
def get_translator_by_id(database, translator_id): """Returns the current translator by id. """ collection = get_translator_collection(database) arguments = [] arguments.append( mongo_utils.make_single_field_argument('_id', translator_id)) arguments.append( mongo_utils.make_single_field_argument('remove_date', None)) argument = general_utils.merge_list_of_dicts(arguments) cursor = mongo_utils.mongo_find_records(collection, argument=argument) translator_list = mongo_utils.unload_cursor(cursor) try: return translator_list[0] except IndexError: return None
def update_function(project, function, changes): """Updates the given parameter set by adding changes for the given changed parameters to the current record. """ collection = get_language_collection(project) if type(function) in [dict, OrderedDict]: function_id = function['_id'] else: function_id = function argument = mongo_utils.make_single_field_argument('_id', function_id) updates = [] for change in changes: if '.' in change: nested_changes = change.split('.') nested_value_string = 'workflow' for nested_change in nested_changes: nested_value_string += '["' "{0}" '"]'.format(nested_change) updates.append( mongo_utils.make_update_argument(change, eval(nested_value_string))) else: updates.append( mongo_utils.make_update_argument(change, function[change])) update = mongo_utils.merge_update_args(updates) cursor = mongo_utils.mongo_update_one(collection, argument, update) if cursor.matched_count == 1: return get_function_by_id(project, function_id) return None
def update_evaluation(project, evaluation, changes): """Updates the given evaluation by adding changes for the given changed parameters to the current record. """ collection = get_evaluation_collection(project) evaluation_id = evaluation['_id'] argument = mongo_utils.make_single_field_argument('_id', evaluation_id) updates = [] for change in changes: if '.' in change: nested_changes = change.split('.') nested_value_string = 'evaluation' for nested_change in nested_changes: nested_value_string += '["' "{0}" '"]'.format(nested_change) updates.append( mongo_utils.make_update_argument(change, eval(nested_value_string))) else: updates.append( mongo_utils.make_update_argument(change, evaluation[change])) update = mongo_utils.merge_update_args(updates) cursor = mongo_utils.mongo_update_one(collection, argument, update) if cursor.matched_count == 1: return get_evaluation_by_id(project, evaluation_id) return None
def remove_project_record_by_name(project_name): """Removes a single record from the project database with the specified project name. """ collection = get_project_collection() argument = mongo_utils.make_single_field_argument("name", project_name) removal = mongo_utils.mongo_remove_one(collection, argument) return removal
def update_template(database, template, changes): """Updates the given template by adding changes for the given changed parameters to the current record. """ collection = get_template_collection(database) template_id = mongo_utils.ensure_objectid(template['_id']) argument = mongo_utils.make_single_field_argument('_id', template_id) updates = [] for change in changes: if '.' in change: nested_changes = change.split('.') nested_value_string = 'template' for nested_change in nested_changes: nested_value_string += '["' "{0}" '"]'.format(nested_change) updates.append( mongo_utils.make_update_argument(change, eval(nested_value_string))) else: updates.append( mongo_utils.make_update_argument(change, template[change])) update = mongo_utils.merge_update_args(updates) cursor = mongo_utils.mongo_update_one(collection, argument, update) if cursor.matched_count == 1: return get_template_by_id(database, template_id) return None
def get_current_templates_by_structure_name(database, structure_name): """Returns all templates in storage for the given structure name. """ collection = get_template_collection(database) arguments = [] arguments.append( mongo_utils.make_single_field_argument('structure', structure_name)) arguments.append( mongo_utils.make_single_field_argument('remove_date', None)) argument = general_utils.merge_list_of_dicts(arguments) cursor = mongo_utils.mongo_find_records(collection, argument=argument) template_list = mongo_utils.unload_cursor(cursor) try: return template_list except IndexError: return None
def get_current_structures(database, args=None): """Returns all the structures that haven't been removed. """ collection = get_structure_collection(database) arguments = [] if args: arguments = [ mongo_utils.make_single_field_argument(arg['key'], arg['value'], arg_type=arg['operation']) for arg in args ] arguments.append( mongo_utils.make_single_field_argument("remove_date", None)) argument = general_utils.merge_list_of_dicts(arguments) cursor = mongo_utils.mongo_find_records(collection, argument=argument) return mongo_utils.unload_cursor(cursor)
def get_enabled_functions(project, current_only=True): """Returns the enabled functions for the specified project. Optionally can return all records (including older) if current_only is set to False. """ collection = get_function_collection(project) arguments = [] if current_only: arguments.append( mongo_utils.make_single_field_argument('remove_date', None)) arguments.append( mongo_utils.make_single_field_argument('status', 'enabled')) argument = general_utils.merge_list_of_dicts(arguments) cursor = mongo_utils.mongo_find_records(collection, argument=argument, named_tuple=False) function_list = mongo_utils.unload_cursor(cursor) return function_list
def get_functions_for_component(project, component, current=True): """Returns functions for the given component for the given project. Optional parameter allows user to return current or all functions. """ collection = get_function_collection(project) arguments = [] if current: arguments.append( mongo_utils.make_single_field_argument('remove_date', None)) arguments.append( mongo_utils.make_single_field_argument('component', component)) argument = general_utils.merge_list_of_dicts(arguments) cursor = mongo_utils.mongo_find_records(collection, argument=argument, named_tuple=False) function_list = mongo_utils.unload_cursor(cursor) return function_list
def update_function_vcs_status_by_id(project, function_id, status): """Updates the vcs status of the given function with the specified status. """ collection = get_function_collection(project) argument = mongo_utils.make_single_field_argument('_id', function_id) update = mongo_utils.make_update_argument('vcs.status', status) cursor = mongo_utils.mongo_update_one(collection, argument, update) return cursor
def get_workflow_evaluations_by_name(project, workflow, name): """Returns all evaluations in the given workflow with the given name. """ collection = get_evaluation_collection(project) arguments = [] arguments.append(mongo_utils.make_single_field_argument('name', name)) arguments.append( mongo_utils.make_single_field_argument('workflow', workflow)) argument = general_utils.merge_list_of_dicts(arguments) cursor = mongo_utils.mongo_find_records(collection, argument=argument, named_tuple=False) evaluation_list = mongo_utils.unload_cursor(cursor) try: return evaluation_list except IndexError: return None
def get_all_current_groups(database): """Returns all groups that are current (no removal date) in the current database. """ collection = get_group_collection(database) argument = mongo_utils.make_single_field_argument('remove_date', None) cursor = mongo_utils.mongo_find_records(collection, argument=argument) return mongo_utils.unload_cursor(cursor)
def get_disabled_components(database, current_only=True): """Returns the disabled components for the specified database. Optionally can return all records (including older) if current_only is set to False. """ collection = get_component_collection(database) arguments = [] if current_only: arguments.append( mongo_utils.make_single_field_argument('remove_date', None)) arguments.append( mongo_utils.make_single_field_argument('status', 'disabled')) argument = general_utils.merge_list_of_dicts(arguments) cursor = mongo_utils.mongo_find_records(collection, argument=argument, named_tuple=False) component_list = mongo_utils.unload_cursor(cursor) return component_list
def update_component_vcs_status_by_id(database, component_id, status): """Updates the vcs status of the given component with the specified status. """ collection = get_component_collection(database) argument = mongo_utils.make_single_field_argument('_id', component_id) update = mongo_utils.make_update_argument('vcs.status', status) cursor = mongo_utils.mongo_update_one(collection, argument, update) return cursor
def get_structure_by_id(database, structure_id): """Returns the current profile by name. """ structure_id = mongo_utils.ensure_objectid(structure_id) collection = get_structure_collection(database) arguments = [] arguments.append( mongo_utils.make_single_field_argument('_id', structure_id)) arguments.append( mongo_utils.make_single_field_argument('remove_date', None)) argument = general_utils.merge_list_of_dicts(arguments) cursor = mongo_utils.mongo_find_records(collection, argument=argument) structure_list = mongo_utils.unload_cursor(cursor) try: return structure_list[0] except IndexError: return None
def update_evaluation_status_by_id(project, evaluation_id, status): """Updates the status of the evaluation with the given id to the given status. """ collection = get_evaluation_collection(project) argument = mongo_utils.make_single_field_argument('_id', evaluation_id) update = mongo_utils.make_update_argument('status', status) cursor = mongo_utils.mongo_update_one(collection, argument, update) return cursor
def get_current_evaluation_by_name(project, name): """Returns the current evaluation by name. """ collection = get_evaluation_collection(project) arguments = [] arguments.append(mongo_utils.make_single_field_argument('name', name)) arguments.append( mongo_utils.make_single_field_argument('remove_date', None)) argument = general_utils.merge_list_of_dicts(arguments) cursor = mongo_utils.mongo_find_records(collection, argument=argument, named_tuple=False) evaluation_list = mongo_utils.unload_cursor(cursor) try: return evaluation_list[0] except IndexError: return None
def get_components_by_name(database, name): """ Returns all the components by name, if it exists, otherwise returns False """ collection = get_component_collection(database) argument = mongo_utils.make_single_field_argument('name', name) cursor = mongo_utils.mongo_find_records(collection, argument=argument, named_tuple=False) return mongo_utils.unload_cursor(cursor)
def get_evaluations_by_name(project, name): """ Returns all the evaluations by name, if it exists, otherwise returns False """ collection = get_evaluation_collection(project) argument = mongo_utils.make_single_field_argument('name', name) cursor = mongo_utils.mongo_find_records(collection, argument=argument, named_tuple=False) return mongo_utils.unload_cursor(cursor)
def remove_component_by_id(database, component_id, timestamp): """Updates the component with the given id to have a removal datetime of the given datetime and sets the component status to removed. """ update_component_status_by_id(database, component_id, 'removed') collection = get_component_collection(database) argument = mongo_utils.make_single_field_argument('_id', component_id) update = mongo_utils.make_update_argument('remove_date', timestamp) cursor = mongo_utils.mongo_update_one(collection, argument, update) return cursor
def get_removed_evaluations(project): """Returns evaluations that have been removed from the specified project. Optionally can return all records (including older) if current_only is set to False. """ collection = get_evaluation_collection(project) argument = mongo_utils.make_single_field_argument('status', 'removed') cursor = mongo_utils.mongo_find_records(collection, argument=argument, named_tuple=False) return mongo_utils.unload_cursor(cursor)
def get_all_current_evaluations(project): """Returns all evaluations that are current (no removal date) in the current project. """ collection = get_evaluation_collection(project) argument = mongo_utils.make_single_field_argument('remove_date', None) cursor = mongo_utils.mongo_find_records(collection, argument=argument, named_tuple=False) return mongo_utils.unload_cursor(cursor)
def remove_evaluation_by_id(project, evaluation_id, timestamp): """Updates the evaluation with the given id to have a removal datetime of the given datetime and sets the evaluation status to removed. """ update_evaluation_status_by_id(project, evaluation_id, 'removed') collection = get_evaluation_collection(project) argument = mongo_utils.make_single_field_argument('_id', evaluation_id) update = mongo_utils.make_update_argument('remove_date', timestamp) cursor = mongo_utils.mongo_update_one(collection, argument, update) return cursor
def get_group_by_id(database, group_id): """Returns the group by the given id. """ collection = get_group_collection(database) argument = mongo_utils.make_single_field_argument('_id', group_id) cursor = mongo_utils.mongo_find_records(collection, argument=argument) group_list = mongo_utils.unload_cursor(cursor) try: return group_list[0] except IndexError: return None
def get_record_by_id(database, collection, record_id): """Returns the record for the given id. """ collection = get_record_collection(database, collection) argument = mongo_utils.make_single_field_argument('_id', record_id) cursor = mongo_utils.mongo_find_records(collection, argument=argument) record_list = mongo_utils.unload_cursor(cursor) try: return record_list[0] except IndexError: return None
def remove_template(database, template): """Removes the given template from the template collection in the given database. """ collection = get_template_collection(database) template_id = mongo_utils.ensure_objectid(template['_id']) arguments = [] arguments.append(mongo_utils.make_single_field_argument( '_id', template_id)) argument = general_utils.merge_list_of_dicts(arguments) result = mongo_utils.mongo_remove_one(collection, argument) return result
def update_component_functions_by_id(database, component_id, functions): """Updates the status of the component with the given id to the given status. """ collection = get_component_collection(database) argument = mongo_utils.make_single_field_argument('_id', component_id) update = mongo_utils.make_update_argument('functions', functions) cursor = mongo_utils.mongo_update_one(collection, argument, update) updated_component = None if cursor.modified_count == 1: updated_component = get_component_by_id(database, component_id) return updated_component