diff options
Diffstat (limited to 'pytask/taskapp/utils.py')
-rw-r--r-- | pytask/taskapp/utils.py | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/pytask/taskapp/utils.py b/pytask/taskapp/utils.py index d9905ad..264c32b 100644 --- a/pytask/taskapp/utils.py +++ b/pytask/taskapp/utils.py @@ -1,4 +1,4 @@ -from pytask.taskapp.models import Task +from pytask.taskapp.models import Task, TextBook from django.http import Http404 def getTask(tid): @@ -9,3 +9,11 @@ def getTask(tid): except Task.DoesNotExist: raise Http404 +def getTextBook(tid): + + try: + task = TextBook.objects.get(uniq_key=tid) + return task + except TextBook.DoesNotExist: + raise Http404 + |