forum/sitemap.py

   1 import re
   2
   3 from django.contrib.sitemaps import Sitemap
   4 from forum.models import Question
   5 from forum.settings import QUESTIONS_SITEMAP_LIMIT, QUESTIONS_SITEMAP_CHANGEFREQ
   6 from django.conf import settings
   7 from django.http import HttpResponse, Http404
   8 from django.template import loader
   9 from django.core import urlresolvers
  10 from django.core.urlresolvers import get_script_prefix
  11 from django.utils.encoding import smart_str
  12 from django.core.paginator import EmptyPage, PageNotAnInteger
  13
  14 def index(request, sitemaps):
  15     sites = []
  16     for section in sitemaps.keys():
  17         sitemap_url = urlresolvers.reverse('sitemap_section_index', prefix='/', kwargs={'section': section})
  18
  19         # Replace double forward slashes with single ones
  20         final_url = '%s%s' % (settings.APP_URL, sitemap_url)
  21         final_url = re.sub("/+", "/", final_url)
  22         final_url = final_url.replace('http:/', 'http://')
  23         final_url = final_url.replace('https:/', 'https://')
  24
  25         sites.append(final_url)
  26
  27     xml = loader.render_to_string('sitemap_index.xml', {'sitemaps': sites})
  28     return HttpResponse(xml, mimetype='application/xml')
  29
  30 def sitemap_section_index(request, section, sitemaps):
  31     try:
  32         sitemap = sitemaps[section]()
  33     except KeyError:
  34         raise Http404("Sitemap doesn't exist")
  35
  36     paginator = sitemap.paginator
  37
  38     locations = []
  39
  40     for page in paginator.page_range:
  41         location = urlresolvers.reverse('sitemap_section_page', prefix='/', kwargs={ 'page' : page, 'section' : section })
  42         location = '%s%s' % (settings.APP_URL, location)
  43         location = re.sub("/+", "/", location)
  44         location = location.replace('http:/', 'http://')
  45         location = location.replace('https:/', 'https://')
  46         locations.append(location)
  47
  48     xml = loader.render_to_string('sitemap_section_index.xml', { 'locations' : locations, })
  49     return HttpResponse(xml, mimetype='application/xml')
  50
  51 def sitemap(request, sitemaps, section=None, page=1):
  52     maps, urls = [], []
  53     if section is not None:
  54         if section not in sitemaps:
  55             raise Http404("No sitemap available for section: %r" % section)
  56         maps.append(sitemaps[section])
  57     else:
  58         maps = sitemaps.values()
  59
  60     for site in maps:
  61         try:
  62             if callable(site):
  63                 urls.extend(site().get_urls(page=page))
  64             else:
  65                 urls.extend(site.get_urls(page=page))
  66         except EmptyPage:
  67             raise Http404("Page %s empty" % page)
  68         except PageNotAnInteger:
  69             raise Http404("No page '%s'" % page)
  70     xml = smart_str(loader.render_to_string('sitemap.xml', {'urlset': urls}))
  71     return HttpResponse(xml, mimetype='application/xml')
  72
  73 class OsqaSitemap(Sitemap):
  74     limit = QUESTIONS_SITEMAP_LIMIT
  75     changefreq = QUESTIONS_SITEMAP_CHANGEFREQ
  76     priority = 0.5
  77     def items(self):
  78         return Question.objects.filter_state(deleted=False).order_by('id')
  79
  80     def lastmod(self, obj):
  81         return obj.last_activity_at
  82
  83     def location(self, obj):
  84         return obj.get_absolute_url()
  85
  86     def __get(self, name, obj, default=None):
  87         try:
  88             attr = getattr(self, name)
  89         except AttributeError:
  90             return default
  91         if callable(attr):
  92             return attr(obj)
  93         return attr
  94
  95     def get_urls(self, page=1):
  96         urls = []
  97         for item in self.paginator.page(page).object_list:
  98             root_relative_url = self.__get('location', item)
  99             relative_url = root_relative_url[len(get_script_prefix()):]
 100             loc = "%s/%s" % (settings.APP_URL, relative_url)
 101             url_info = {
 102                 'location':   loc,
 103                 'lastmod':    self.__get('lastmod', item, None),
 104                 'changefreq': self.__get('changefreq', item, None),
 105                 'priority':   self.__get('priority', item, None)
 106             }
 107             urls.append(url_info)
 108         return urls