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