Skip to content

Commit b4ca836

Browse files
committed
Remove warnings for GenericForeignKey
1 parent af61878 commit b4ca836

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

tagging/models.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,12 @@
55
from django.db import connection
66
from django.utils.encoding import smart_text
77
from django.utils.encoding import python_2_unicode_compatible
8-
from django.contrib.contenttypes import generic
9-
from django.contrib.contenttypes.models import ContentType
108
from django.utils.translation import ugettext_lazy as _
9+
from django.contrib.contenttypes.models import ContentType
10+
try:
11+
from django.contrib.contenttypes.fields import GenericForeignKey
12+
except ImportError:
13+
from django.contrib.contenttypes.generic import GenericForeignKey
1114

1215
from . import settings
1316
from .utils import LOGARITHMIC
@@ -484,7 +487,7 @@ class TaggedItem(models.Model):
484487
tag = models.ForeignKey(Tag, verbose_name=_('tag'), related_name='items')
485488
content_type = models.ForeignKey(ContentType, verbose_name=_('content type'))
486489
object_id = models.PositiveIntegerField(_('object id'), db_index=True)
487-
object = generic.GenericForeignKey('content_type', 'object_id')
490+
object = GenericForeignKey('content_type', 'object_id')
488491

489492
objects = TaggedItemManager()
490493

0 commit comments

Comments
 (0)