[refactoring] Return default values if provided in getters. Also return dates as strings

This commit is contained in:
Adrià Mercader 2011-04-12 16:13:38 +01:00
parent a234d6ff6a
commit aaf7885f43
1 changed files with 18 additions and 8 deletions

View File

@ -36,7 +36,8 @@ def _get_source_status(source):
.order_by(HarvestJob.created.desc()).limit(1).first() .order_by(HarvestJob.created.desc()).limit(1).first()
if last_job: if last_job:
out['last_harvest_request'] = last_job.gather_finished #TODO: Should we encode the dates as strings?
out['last_harvest_request'] = str(last_job.gather_finished)
#Get HarvestObjects from last job whit links to packages #Get HarvestObjects from last job whit links to packages
@ -145,7 +146,10 @@ def _object_as_dict(obj):
def get_harvest_source(id,default=Exception,attr=None): def get_harvest_source(id,default=Exception,attr=None):
source = HarvestSource.get(id,default=default,attr=attr) source = HarvestSource.get(id,default=default,attr=attr)
if source:
return _source_as_dict(source) return _source_as_dict(source)
else:
return default
def get_harvest_sources(**kwds): def get_harvest_sources(**kwds):
sources = HarvestSource.filter(**kwds).all() sources = HarvestSource.filter(**kwds).all()
@ -187,9 +191,12 @@ def remove_harvest_source(source_id):
return True return True
def get_harvest_job(id,attr=None): def get_harvest_job(id,default=Exception,attr=None):
job = HarvestJob.get(id,attr) job = HarvestJob.get(id,default=default,attr=attr)
if job:
return _job_as_dict(job) return _job_as_dict(job)
else:
return default
def get_harvest_jobs(**kwds): def get_harvest_jobs(**kwds):
jobs = HarvestJob.filter(**kwds).all() jobs = HarvestJob.filter(**kwds).all()
@ -230,9 +237,12 @@ def run_harvest_jobs():
publisher.close() publisher.close()
return jobs return jobs
def get_harvest_object(id,attr=None): def get_harvest_object(id,default=Exception,attr=None):
obj = HarvestObject.get(id,attr) obj = HarvestObject.get(id,default=default,attr=attr)
if obj:
return _object_as_dict(obj) return _object_as_dict(obj)
else:
return default
def get_harvest_objects(**kwds): def get_harvest_objects(**kwds):
objects = HarvestObject.filter(**kwds).all() objects = HarvestObject.filter(**kwds).all()