get_current_site compatibility (#23)

rizumu · Mar 23, 2017 · 89f74b2 · 89f74b2
1 parent 27b6f47
commit 89f74b2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/podcasting/feeds.py b/podcasting/feeds.py
@@ -7,7 +7,10 @@
 from django.utils.feedgenerator import rfc2822_date, Rss201rev2Feed, Atom1Feed
 from django.shortcuts import get_object_or_404
 
-from django.contrib.sites.models import get_current_site
+try:
+    from django.contrib.sites.models import get_current_site
+except ImportError:
+    from django.contrib.sites.shortcuts import get_current_site
 from django.contrib.syndication.views import Feed
 from django.views.generic.base import RedirectView