def delete_all_by_registry_id(registry_id): """ Delete all OaiHarvesterSet used by a registry. Args: registry_id: The registry id. """ OaiHarvesterSet.delete_all_by_registry_id(registry_id)
def update_for_all_harvest_by_list_ids(list_oai_set_ids, harvest): """Update the harvest for all OaiHarvesterSet by a list of ids. Args: list_oai_set_ids: List of OaiHarvesterSet ids. harvest: Harvest (True/False) """ OaiHarvesterSet.update_for_all_harvest_by_list_ids(list_oai_set_ids, harvest)
def update_for_all_harvest_by_registry_id(registry_id, harvest): """ Update the harvest for all OaiHarvesterSet used by the registry. Args: registry_id: The registry id. harvest: Harvest (True/False). """ OaiHarvesterSet.update_for_all_harvest_by_registry_id( registry_id=registry_id, harvest=harvest)
def test_list_sets_as_object_return_object_and_ok_status(self, mock_set, mock_transform): # Arrange mock_set.return_value = [], status.HTTP_200_OK mock_transform.return_value = [OaiHarvesterSet(), OaiHarvesterSet()] # Act data, status_code = oai_verbs_api.list_sets_as_object("") # Assert self.assertTrue(all(isinstance(item, OaiHarvesterSet) for item in data)) self.assertEquals(status_code, status.HTTP_200_OK)
def get_all(): """ Get all OaiHarvesterSet. Returns: List of OaiHarvesterSet. """ return OaiHarvesterSet.get_all()
def get_by_id(oai_harvester_set_id): """ Get an OaiHarvesterSet by its id. Args: oai_harvester_set_id: The OaiHarvesterSet id. Returns: OaiHarvesterSet instance. """ return OaiHarvesterSet.get_by_id(oai_set_id=oai_harvester_set_id)
def _set_oai_record_fields(oai_record): """Set OaiRecord fields. Args: oai_record: Returns: OaiRecord with assigned fields. """ oai_record.identifier = "oai:test/id.0006" oai_record.last_modification_date = datetime.datetime.now() oai_record.deleted = False oai_record.harvester_sets = [OaiHarvesterSet(), OaiHarvesterSet()] oai_record.harvester_metadata_format = OaiHarvesterMetadataFormat() oai_record.registry = OaiRegistry() oai_record.xml_content = "<test><message>Hello</message></test>" return oai_record
def _create_oai_harvester_set(): """ Get an OaiHarvesterSet object. Returns: OaiHarvesterSet instance. """ oai_harvester_set = OaiHarvesterSet() _set_oai_harvester_set_fields(oai_harvester_set) return oai_harvester_set
def get_all_to_harvest_by_registry_id(registry_id, order_by_field=None): """ List all OaiHarvesterSet to harvest used by a registry Args: registry_id: The registry id. order_by_field: Order by field. Returns: List of OaiHarvesterSet. """ return OaiHarvesterSet.get_all_by_registry_id_and_harvest( registry_id=registry_id, harvest=True, order_by_field=order_by_field)
def get_all_by_list_registry_ids(list_registry_ids, order_by_field=None): """ Return a list of OaiHarvesterSet by a list of registry ids. Possibility to order_by the list Args: list_registry_ids: List of registry ids. order_by_field: Order by field. Returns: List of OaiHarvesterSet. """ return OaiHarvesterSet.get_all_by_list_registry_ids( list_registry_ids=list_registry_ids, order_by_field=order_by_field)
def get_all_by_registry_id(registry_id, order_by_field=None): """ Get all OaiHarvesterSet used by a registry. Args: registry_id: The registry id. order_by_field: Order by field. Returns: List of OaiHarvesterSet. """ return OaiHarvesterSet.get_all_by_registry_id( registry_id=registry_id, order_by_field=order_by_field)
def get_by_set_spec_and_registry_id(set_spec, registry_id): """ Get an OaiHarvesterSet by its set_spec and registry_id. Args: set_spec: The set spec. registry_id: The registry id. Returns: OaiHarvesterSet instance. """ return OaiHarvesterSet.get_by_set_spec_and_registry_id( set_spec=set_spec, registry_id=registry_id)
def transform_dict_set_to_oai_harvester_set(data): """ Transforms a dict to a list of OaiHarvesterSet object. Args: data: Data to transform. Returns: List of OaiHarvesterSet instances. """ return [ OaiHarvesterSet(set_name=obj['setName'], set_spec=obj['setSpec'], raw=raw_xml_to_dict(obj['raw'])) for obj in data ]