Enable A/B test for blackout.
[wolnelektury.git] / src / wolnelektury / views.py
1 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
3 #
4 from datetime import date, datetime
5 import feedparser
6 from allauth.socialaccount.views import SignupView
7
8 from django.conf import settings
9 from django.contrib import auth
10 from django.contrib.auth.decorators import login_required
11 from django.contrib.auth.forms import AuthenticationForm
12 from django.core.cache import cache
13 from django.http import HttpResponse, HttpResponseRedirect
14 from django.shortcuts import render
15 from django.utils.http import urlquote_plus
16 from django.utils.translation import ugettext_lazy as _
17 from django.views.decorators.cache import never_cache
18
19 from ajaxable.utils import AjaxableFormView
20 from ajaxable.utils import placeholdized
21 from catalogue.models import Book, Collection, Tag, Fragment
22
23 from social.utils import get_or_choose_cite
24 from wolnelektury.forms import RegistrationForm, SocialSignupForm
25
26
27 @never_cache
28 def main_page(request):
29     ctx = {
30         'last_published': Book.objects.exclude(cover_thumb='').filter(findable=True, parent=None).order_by('-created_at')[:6],
31         'theme_books': [],
32     }
33
34     # FIXME: find this theme and books properly.
35     if Fragment.objects.exists():
36         while True:
37             ctx['theme'] = Tag.objects.filter(category='theme').order_by('?')[:1][0]
38             tf = Fragment.tagged.with_any([ctx['theme']]).select_related('book').filter(book__findable=True).order_by('?')[:100]
39             if not tf:
40                 continue
41             ctx['theme_fragment'] = tf[0]
42             for f in tf:
43                 if f.book not in ctx['theme_books']:
44                     ctx['theme_books'].append(f.book)
45                 if len(ctx['theme_books']) == 3:
46                     break
47             break
48
49     # Choose a collection for main.
50     try:
51         ctx['collection'] = Collection.objects.order_by('?')[:1][0]
52     except IndexError:
53         pass
54
55     best = []
56     best_places = 5
57     for recommended in Collection.objects.filter(role='recommend').order_by('?'):
58         books = list(recommended.get_books().exclude(id__in=[b.id for b in best]).order_by('?')[:best_places])
59         best.extend(books)
60         best_places -= len(books)
61         if not best_places:
62             break
63     if best_places:
64         best.extend(
65             list(
66                 Book.objects.filter(findable=True).exclude(id__in=[b.id for b in best]).order_by('?')[:best_places]
67             )
68         )
69     ctx['best'] = best
70
71     return render(request, "main_page.html", ctx)
72
73
74 class LoginFormView(AjaxableFormView):
75     form_class = AuthenticationForm
76     template = "auth/login.html"
77     placeholdize = True
78     title = _('Sign in')
79     submit = _('Sign in')
80     ajax_redirect = True
81
82     def __call__(self, request):
83         if request.user.is_authenticated:
84             return self.redirect_or_refresh(
85                 request, '/',
86                 message=_('Already logged in as user %(user)s', ) % {'user': request.user.username})
87         return super(LoginFormView, self).__call__(request)
88
89     def success(self, form, request):
90         auth.login(request, form.get_user())
91
92
93 class RegisterFormView(AjaxableFormView):
94     form_class = RegistrationForm
95     template = "auth/register.html"
96     placeholdize = True
97     title = _('Register')
98     submit = _('Register')
99     ajax_redirect = True
100     form_prefix = 'register'
101     honeypot = True
102
103     def __call__(self, request):
104         if request.user.is_authenticated:
105             return self.redirect_or_refresh(
106                 request, '/',
107                 message=_('Already logged in as user %(user)s', ) % {'user': request.user.username})
108         return super(RegisterFormView, self).__call__(request)
109
110     def success(self, form, request):
111         form.save()
112         user = auth.authenticate(
113             username=form.cleaned_data['username'],
114             password=form.cleaned_data['password1']
115         )
116         auth.login(request, user)
117
118
119 class LoginRegisterFormView(LoginFormView):
120     template = 'auth/login_register.html'
121     title = _('You have to be logged in to continue')
122
123     def extra_context(self, request, obj):
124         return {
125             "register_form": placeholdized(RegistrationForm(prefix='register')),
126             "register_submit": _('Register'),
127         }
128
129
130 @never_cache
131 def logout_then_redirect(request):
132     auth.logout(request)
133     return HttpResponseRedirect(urlquote_plus(request.GET.get('next', '/'), safe='/?='))
134
135
136 @never_cache
137 def clock(request):
138     """ Provides server UTC time for jquery.countdown,
139     in a format suitable for Date.parse()
140     """
141     return HttpResponse(datetime.utcnow().strftime('%Y/%m/%d %H:%M:%S UTC'))
142
143
144 def publish_plan(request):
145     cache_key = "publish_plan"
146     plan = cache.get(cache_key)
147
148     if plan is None:
149         plan = []
150         try:
151             feed = feedparser.parse(settings.PUBLISH_PLAN_FEED)
152         except:
153             pass
154         else:
155             for i in range(len(feed['entries'])):
156                 plan.append({
157                     'title': feed['entries'][i].title,
158                     'link': feed['entries'][i].link,
159                     })
160         cache.set(cache_key, plan, 1800)
161
162     return render(request, "publish_plan.html", {'plan': plan})
163
164
165 @login_required
166 def user_settings(request):
167     return render(request, "user.html")
168
169
170 def widget(request):
171     return render(request, 'widget.html')
172
173
174 class SocialSignupView(SignupView):
175     form_class = SocialSignupForm
176
177
178 def exception_test(request):
179     msg = request.GET.get('msg')
180     if msg:
181         raise Exception('Exception test: %s' % msg)
182     else:
183         raise Exception('Exception test')
184
185
186 def post_test(request):
187     return render(request, 'post_test.html', {'action': '/api/reading/jego-zasady/complete/'})