YunoHost/yunorunner
1
0
Fork 0
mirror of https://github.com/YunoHost/yunorunner.git synced 2024-09-03 20:05:52 +02:00
Code Issues Projects Releases Packages Wiki Activity

[enh] handle running job cancelation

Browse source
This commit is contained in:
Laurent Peuch 2019-01-22 03:27:43 +01:00
parent 23a12d61c0
commit 34873ae9f8
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
run.py
View file

@ -12,6 +12,7 @@ import itertools
from datetime import datetime, date from datetime import datetime, date
from collections import defaultdict from collections import defaultdict
from functools import wraps from functools import wraps
from concurrent.futures._base import CancelledError
import ujson import ujson
import aiohttp import aiohttp
@ -354,6 +355,13 @@ async def run_job(worker, job):
"data": model_to_dict(job), "data": model_to_dict(job),
}, ["jobs", f"job-{job.id}", f"app-jobs-{job.url_or_path}"]) }, ["jobs", f"job-{job.id}", f"app-jobs-{job.url_or_path}"])
except CancelledError:
job.log += "\n"
job.end_time = datetime.now()
job.state = "canceled"
job.save()
task_logger.info(f"Job '{job.name} #{job.id}' has been canceled")
except Exception as e: except Exception as e:
traceback.print_exc() traceback.print_exc()
task_logger.exception(f"ERROR in job '{job.name} #{job.id}'") task_logger.exception(f"ERROR in job '{job.name} #{job.id}'")