Generic management of baskets in owns items

1 files changed, 6 insertions, 11 deletions

diff --git a/archaeological_finds/models.py b/archaeological_finds/models.py
index 9c972f68c..417dd3929 100644
--- a/archaeological_finds/models.py
+++ b/archaeological_finds/models.py
@@ -240,6 +240,11 @@ CHECK_CHOICES = (('NC', _(u"Not checked")),
                  )
 
 
+class FindBasket(Basket):
+    items = models.ManyToManyField('Find', blank=True, null=True,
+                                   related_name='basket')
+
+
 class Find(BaseHistorizedItem, ImageModel, OwnPerms, ShortMenuItem):
     CHECK_DICT = dict(CHECK_CHOICES)
     SHOW_URL = 'show-find'
@@ -329,6 +334,7 @@ class Find(BaseHistorizedItem, ImageModel, OwnPerms, ShortMenuItem):
     check_date = models.DateField(_(u"Check date"),
                                   default=datetime.date.today)
     history = HistoricalRecords()
+    BASKET_MODEL = FindBasket
 
     def __init__(self, *args, **kwargs):
         super(Find, self).__init__(*args, **kwargs)
@@ -384,12 +390,6 @@ class Find(BaseHistorizedItem, ImageModel, OwnPerms, ShortMenuItem):
             lbl += u' ({})'.format(base)
         return lbl
 
-    @classmethod
-    def get_owns(cls, user):
-        if user.is_anonymous():
-            return FindBasket.objects.none()
-        return FindBasket.objects.filter(user=user)
-
     def get_first_base_find(self):
         q = self.base_finds
         if not q.count():
@@ -546,11 +546,6 @@ class Find(BaseHistorizedItem, ImageModel, OwnPerms, ShortMenuItem):
             #        idx and idx['material_index__max'] + 1 or 1
 
 
-class FindBasket(Basket):
-    items = models.ManyToManyField(Find, blank=True, null=True,
-                                   related_name='basket')
-
-
 class FindSource(Source):
     SHOW_URL = 'show-findsource'