diff options
Diffstat (limited to 'app/tasks')
| -rw-r--r-- | app/tasks/__init__.py | 6 | ||||
| -rw-r--r-- | app/tasks/importtasks.py | 10 |
2 files changed, 8 insertions, 8 deletions
diff --git a/app/tasks/__init__.py b/app/tasks/__init__.py index 40a824c..fe566b8 100644 --- a/app/tasks/__init__.py +++ b/app/tasks/__init__.py @@ -4,6 +4,12 @@ from celery import Celery from app import app from app.models import * +class TaskError(Exception): + def __init__(self, value): + self.value = value + def __str__(self): + return repr("TaskError: " + self.value) + class FlaskCelery(Celery): def __init__(self, *args, **kwargs): super(FlaskCelery, self).__init__(*args, **kwargs) diff --git a/app/tasks/importtasks.py b/app/tasks/importtasks.py index 7233d87..12b61dd 100644 --- a/app/tasks/importtasks.py +++ b/app/tasks/importtasks.py @@ -4,14 +4,7 @@ import urllib.request from urllib.parse import urlparse, quote_plus from app import app from app.models import * -from app.tasks import celery - -class TaskError(Exception): - def __init__(self, value): - self.value = value - def __str__(self): - return repr(self.value) - +from app.tasks import celery, TaskError class GithubURLMaker: def __init__(self, url): @@ -177,6 +170,7 @@ def getMeta(urlstr, author): @celery.task() def makeVCSRelease(id, branch): release = PackageRelease.query.get(id) + if release is None: raise TaskError("No such release!") |
