[Plugin] Schedule: Allow retraction of service

This commit is contained in:
Ferdinand Thiessen 2021-02-07 16:33:48 +01:00
parent ed361a7361
commit e7efa53071
1 changed files with 9 additions and 4 deletions

View File

@ -169,9 +169,14 @@ def delete_job(job: Job):
def assign_to_job(job: Job, user, value):
service = Service.query.get((job.id, user._id))
if service:
service.value = value
if value < 0:
if not service:
raise BadRequest
db.session.delete(service)
else:
service = Service(user_=user, value=value, job_=job)
db.session.add(service)
if service:
service.value = value
else:
service = Service(user_=user, value=value, job_=job)
db.session.add(service)
db.session.commit()