X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/fe3838bfb645ccb7e2775ae709ac45044fd14d51..f12a9837a7a7a49507106a1f0d4d335b682fe1f6:/apps/newtagging/models.py

diff --git a/apps/newtagging/models.py b/apps/newtagging/models.py
index e988b6c9d..e121994c7 100644
--- a/apps/newtagging/models.py
+++ b/apps/newtagging/models.py
@@ -10,6 +10,7 @@ from django.contrib.contenttypes.models import ContentType
 from django.db import connection, models
 from django.utils.translation import ugettext_lazy as _
 from django.db.models.base import ModelBase
+from django.core.exceptions import ObjectDoesNotExist
 
 qn = connection.ops.quote_name
 
@@ -62,6 +63,14 @@ class TagManager(models.Manager):
             if tag not in current_tags:
                 self.intermediary_table_model._default_manager.create(tag=tag, content_object=obj)
     
+    def remove_tag(self, obj, tag):
+        """
+        Remove tag from an object.
+        """
+        content_type = ContentType.objects.get_for_model(obj)
+        self.intermediary_table_model._default_manager.filter(content_type__pk=content_type.pk,
+            object_id=obj.pk, tag=tag).delete()
+
     def get_for_object(self, obj):
         """
         Create a queryset matching all tags associated with the given
@@ -225,7 +234,7 @@ class TagManager(models.Manager):
         WHERE %(tagged_item)s.content_type_id = %(content_type_id)s
         AND %(tagged_item)s.object_id IN
         (
-            SELECT temporary.object_id 
+            SELECT *
             FROM (
                 SELECT %(tagged_item)s.object_id
                 FROM %(tagged_item)s, %(tag)s
@@ -236,6 +245,7 @@ class TagManager(models.Manager):
                 HAVING COUNT(%(tagged_item)s.object_id) = %(tag_count)s
             ) AS temporary
         )
+        AND %(tag)s.id NOT IN (%(tag_id_placeholders)s)
         %(extra_where)s
         GROUP BY %(tag_columns)s
         %(min_count_sql)s
@@ -465,8 +475,11 @@ def create_intermediary_table_model(model):
         unique_together = (('tag', 'content_type', 'object_id'),)
 
     def obj_unicode(self):
-        return u'%s [%s]' % (self.content_type.get_object_for_this_type(pk=self.object_id), self.tag)
-        
+        try:
+            return u'%s [%s]' % (self.content_type.get_object_for_this_type(pk=self.object_id), self.tag)
+        except ObjectDoesNotExist:
+            return u'<deleted> [%s]' % self.tag
+            
     # Set up a dictionary to simulate declarations within a class    
     attrs = {
         '__module__': model.__module__,