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
]
