class CKAN():
def __init__(self):
self.ua = (
'waarismijnstemlokaal/1.0 (+https://waarismijnstemlokaal.nl/)')
self.ckanapi = RemoteCKAN(app.config['CKAN_URL'],
apikey=app.config['CKAN_API_KEY'],
user_agent=self.ua).action
self.elections = app.config['CKAN_CURRENT_ELECTIONS']
self.resources_metadata = self._get_resources_metadata()
def create_datastore(self, resource_id, fields):
self.ckanapi.datastore_create(resource_id=resource_id,
force=True,
fields=fields,
primary_key=['UUID'])
def resource_show(self, resource_id):
return self.ckanapi.resource_show(id=resource_id)
def delete_datastore(self, resource_id):
self.ckanapi.datastore_delete(resource_id=resource_id, force=True)
def _get_resources_metadata(self):
resources_metadata = {}
for election_key, election_value in self.elections.items():
resources_metadata[election_key] = {}
resources_metadata[election_key]['publish_resource'] = (
self.ckanapi.resource_show(
id=election_value['publish_resource']))
resources_metadata[election_key]['draft_resource'] = (
self.ckanapi.resource_show(
id=election_value['draft_resource']))
return resources_metadata
def get_records(self, resource_id):
return self.ckanapi.datastore_search(resource_id=resource_id,
limit=15000)
def filter_records(self, resource_id, datastore_filters={}):
return self.ckanapi.datastore_search(resource_id=resource_id,
filters=datastore_filters,
limit=15000)
def save_records(self, resource_id, records):
self.ckanapi.datastore_upsert(resource_id=resource_id,
force=True,
records=records,
method='upsert')
def delete_records(self, resource_id, filters=None):
self.ckanapi.datastore_delete(resource_id=resource_id,
force=True,
filters=filters)
# First delete all records in the publish_resource for the current
# gemeente, then upsert all draft_records of the current gemeente
# to the publish_resource
def publish(self, verkiezing, gemeente_code, draft_records):
election = self.elections[verkiezing]
self.delete_records(election['publish_resource'],
{'CBS gemeentecode': gemeente_code})
self.save_records(election['publish_resource'], draft_records)
