fixes

[wolnelektury.git] / src / waiter / models.py

diff --git a/src/waiter/models.py b/src/waiter/models.py
index 112e754..45ff80e 100644 (file)
--- a/src/waiter/models.py
+++ b/src/waiter/models.py
@@ -1,19 +1,16 @@
-# -*- coding: utf-8 -*-
 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
 from os.path import join, isfile
-from django.core.urlresolvers import reverse
+from django.urls import reverse
 from django.db import models
 from waiter.settings import WAITER_URL, WAITER_MAX_QUEUE
 from waiter.utils import check_abspath
-from picklefield import PickledObjectField
 
 
 class WaitedFile(models.Model):
     path = models.CharField(max_length=255, unique=True, db_index=True)
     task_id = models.CharField(max_length=128, db_index=True, null=True, blank=True)
-    task = PickledObjectField(null=True, editable=False)
     description = models.CharField(max_length=255, null=True, blank=True)
 
     @classmethod
@@ -53,8 +50,8 @@ class WaitedFile(models.Model):
         if not already:
             waited, created = cls.objects.get_or_create(path=path)
             if created:
-                waited.task = task_creator(check_abspath(path), waited.pk)
-                waited.task_id = waited.task.task_id
+                task = task_creator(check_abspath(path), waited.pk)
+                waited.task_id = task.task_id
                 waited.description = description
                 waited.save()
             return reverse("waiter", args=[path])