cms.forms.utils: 78 total statements, 0.0% covered

Generated: Wed 2013-03-13 10:33 CET

Source file: /media/Envs/Envs/filer-gallery/lib/python2.7/site-packages/cms/forms/utils.py

Stats: 0 executed, 67 missed, 11 excluded, 21 ignored

1. # -*- coding: utf-8 -*-
2. from cms.models import Page
3. from cms.models.titlemodels import Title
4. from cms.utils import i18n
5. from collections import defaultdict
6. from django.conf import settings
7. from django.contrib.sites.models import Site
8. from django.core.cache import cache
9. from django.db.models.signals import post_save, post_delete
10. from django.utils import translation
11. from django.utils.datastructures import SortedDict
12. from django.utils.safestring import mark_safe
15. def update_site_and_page_choices(lang=None):
16. lang = lang or translation.get_language()
17. SITE_CHOICES_KEY = get_site_cache_key(lang)
18. PAGE_CHOICES_KEY = get_page_cache_key(lang)
19. if settings.CMS_MODERATOR:
20. title_queryset = Title.objects.filter(page__publisher_is_draft=False)
21. else:
22. title_queryset = Title.objects.filter(page__publisher_is_draft=True)
23. title_queryset = title_queryset.select_related('page', 'page__site').order_by('page__tree_id', 'page__lft', 'page__rght')
24. pages = defaultdict(SortedDict)
25. sites = {}
26. for title in title_queryset:
27. page = pages[title.page.site.pk].get(title.page.pk, {})
28. page[title.language] = title
29. pages[title.page.site.pk][title.page.pk] = page
30. sites[title.page.site.pk] = title.page.site.name
32. site_choices = []
33. page_choices = [('', '----')]
35. language_order = [lang] + i18n.get_fallback_languages(lang)
37. for sitepk, sitename in sites.items():
38. site_choices.append((sitepk, sitename))
40. site_page_choices = []
41. for titles in pages[sitepk].values():
42. title = None
43. for language in language_order:
44. title = titles.get(language)
45. if title:
46. break
47. if not title:
48. continue
50. indent = u"  " * title.page.level
51. page_title = mark_safe(u"%s%s" % (indent, title.title))
52. site_page_choices.append((title.page.pk, page_title))
54. page_choices.append((sitename, site_page_choices))
56. # We set it to 1 day here because we actively invalidate this cache.
57. cache.set(SITE_CHOICES_KEY, site_choices, 86400)
58. cache.set(PAGE_CHOICES_KEY, page_choices, 86400)
59. return site_choices, page_choices
61. def get_site_choices(lang=None):
62. lang = lang or translation.get_language()
63. site_choices = cache.get(get_site_cache_key(lang))
64. if site_choices is None:
65. site_choices, page_choices = update_site_and_page_choices(lang)
66. return site_choices
68. def get_page_choices(lang=None):
69. lang = lang or translation.get_language()
70. page_choices = cache.get(get_page_cache_key(lang))
71. if page_choices is None:
72. site_choices, page_choices = update_site_and_page_choices(lang)
73. return page_choices
75. def _get_key(prefix, lang):
76. return "%s-%s" % (prefix, lang)
78. def get_site_cache_key(lang):
79. return _get_key(settings.CMS_SITE_CHOICES_CACHE_KEY, lang)
81. def get_page_cache_key(lang):
82. return _get_key(settings.CMS_PAGE_CHOICES_CACHE_KEY, lang)
84. def _clean_many(prefix):
85. keys = []
86. for lang in [language[0] for language in settings.LANGUAGES]:
87. keys.append(_get_key(prefix, lang))
88. cache.delete_many(keys)
90. def clean_site_choices_cache(sender, **kwargs):
91. _clean_many(settings.CMS_SITE_CHOICES_CACHE_KEY)
93. def clean_page_choices_cache(sender, **kwargs):
94. _clean_many(settings.CMS_PAGE_CHOICES_CACHE_KEY)
96. post_save.connect(clean_page_choices_cache, sender=Page)
97. post_save.connect(clean_site_choices_cache, sender=Site)
98. post_delete.connect(clean_page_choices_cache, sender=Page)
99. post_delete.connect(clean_site_choices_cache, sender=Site)