user.py 18.5 KB
Newer Older
Skia's avatar
Skia committed
1
2
# This file contains all the views that concern the user model
from django.shortcuts import render, redirect, get_object_or_404
Skia's avatar
Skia committed
3
from django.contrib.auth import logout as auth_logout, views
4
from django.utils.translation import ugettext as _
5
from django.core.urlresolvers import reverse
6
from django.core.exceptions import PermissionDenied, ObjectDoesNotExist, ValidationError
Skia's avatar
Skia committed
7
from django.http import Http404
Skia's avatar
Skia committed
8
from django.views.generic.edit import UpdateView
Sli's avatar
Sli committed
9
from django.views.generic import ListView, DetailView, TemplateView, DeleteView
10
11
from django.forms.models import modelform_factory
from django.forms import CheckboxSelectMultiple
Skia's avatar
Skia committed
12
from django.template.response import TemplateResponse
Skia's avatar
Skia committed
13
from django.conf import settings
Sli's avatar
Sli committed
14
from django.views.generic.dates import YearMixin, MonthMixin
15

Sli's avatar
Sli committed
16
17
from django.utils import timezone
from datetime import timedelta, datetime, date
Skia's avatar
Skia committed
18
19
import logging

20
from core.views import CanViewMixin, CanEditMixin, CanEditPropMixin, TabedViewMixin
Skia's avatar
Skia committed
21
from core.views.forms import RegisteringForm, UserPropForm, UserProfileForm, LoginForm, UserGodfathersForm
22
from core.models import User, SithFile
Lo-J's avatar
Lo-J committed
23
from club.models import Club
Skia's avatar
Skia committed
24
from subscription.models import Subscription
Skia's avatar
Skia committed
25

Skia's avatar
Skia committed
26
27
28
29
30
31
def login(request):
    """
    The login view

    Needs to be improve with correct handling of form exceptions
    """
Skia's avatar
Skia committed
32
    return views.login(request, template_name="core/login.jinja", authentication_form=LoginForm)
Skia's avatar
Skia committed
33
34
35
36
37
38
39
40
41
42
43

def logout(request):
    """
    The logout view
    """
    return views.logout_then_login(request)

def password_change(request):
    """
    Allows a user to change its password
    """
Skia's avatar
Skia committed
44
    return views.password_change(request, template_name="core/password_change.jinja", post_change_redirect=reverse("core:password_change_done"))
Skia's avatar
Skia committed
45
46
47
48
49

def password_change_done(request):
    """
    Allows a user to change its password
    """
Skia's avatar
Skia committed
50
    return views.password_change_done(request, template_name="core/password_change_done.jinja")
Skia's avatar
Skia committed
51

Skia's avatar
Skia committed
52
53
54
55
def password_root_change(request, user_id):
    """
    Allows a root user to change someone's password
    """
Skia's avatar
Skia committed
56
    if not request.user.is_root:
Skia's avatar
Skia committed
57
58
59
60
61
62
63
64
65
66
67
68
69
        raise PermissionDenied
    user = User.objects.filter(id=user_id).first()
    if not user:
        raise Http404("User not found")
    if request.method == "POST":
        form = views.SetPasswordForm(user=user, data=request.POST)
        if form.is_valid():
            form.save()
            return redirect("core:password_change_done")
    else:
        form = views.SetPasswordForm(user=user)
    return TemplateResponse(request, "core/password_change.jinja", {'form': form, 'target': user})

Skia's avatar
Skia committed
70
def password_reset(request):
Skia's avatar
Skia committed
71
72
73
    """
    Allows someone to enter an email adresse for resetting password
    """
Skia's avatar
Skia committed
74
    return views.password_reset(request,
Skia's avatar
Skia committed
75
76
                                template_name="core/password_reset.jinja",
                                email_template_name="core/password_reset_email.jinja",
Skia's avatar
Skia committed
77
                                post_reset_redirect="core:password_reset_done",
Skia's avatar
Skia committed
78
                               )
Skia's avatar
Skia committed
79
80

def password_reset_done(request):
Skia's avatar
Skia committed
81
82
83
    """
    Confirm that the reset email has been sent
    """
Skia's avatar
Skia committed
84
    return views.password_reset_done(request, template_name="core/password_reset_done.jinja")
Skia's avatar
Skia committed
85

Skia's avatar
Skia committed
86
def password_reset_confirm(request, uidb64=None, token=None):
Skia's avatar
Skia committed
87
88
89
    """
    Provide a reset password formular
    """
Skia's avatar
Skia committed
90
91
    return views.password_reset_confirm(request, uidb64=uidb64, token=token,
                                        post_reset_redirect="core:password_reset_complete",
Skia's avatar
Skia committed
92
                                        template_name="core/password_reset_confirm.jinja",
Skia's avatar
Skia committed
93
94
95
                                       )

def password_reset_complete(request):
Skia's avatar
Skia committed
96
97
98
    """
    Confirm the password has sucessfully been reset
    """
Skia's avatar
Skia committed
99
    return views.password_reset_complete(request,
Skia's avatar
Skia committed
100
                                         template_name="core/password_reset_complete.jinja",
Skia's avatar
Skia committed
101
                                        )
Skia's avatar
Skia committed
102

Skia's avatar
Skia committed
103
def register(request):
Skia's avatar
Skia committed
104
    context = {}
Skia's avatar
Skia committed
105
106
107
108
109
110
111
112
113
114
115
116
117
118
    if request.method == 'POST':
        form = RegisteringForm(request.POST)
        if form.is_valid():
            logging.debug("Registering "+form.cleaned_data['first_name']+form.cleaned_data['last_name'])
            u = form.save()
            context['user_registered'] = u
            context['tests'] = 'TEST_REGISTER_USER_FORM_OK'
            form = RegisteringForm()
        else:
            context['error'] = 'Erreur'
            context['tests'] = 'TEST_REGISTER_USER_FORM_FAIL'
    else:
        form = RegisteringForm()
    context['form'] = form.as_p()
Skia's avatar
Skia committed
119
    return render(request, "core/register.jinja", context)
Skia's avatar
Skia committed
120

121
122
123
124
125
126
127
128
129
130
131
class UserTabsMixin(TabedViewMixin):
    def get_tabs_title(self):
        return self.object.get_display_name()

    def get_list_of_tabs(self):
        tab_list = []
        tab_list.append({
                    'url': reverse('core:user_profile', kwargs={'user_id': self.object.id}),
                    'slug': 'infos',
                    'name': _("Infos"),
                    })
Skia's avatar
Skia committed
132
133
        tab_list.append({
                    'url': reverse('core:user_godfathers', kwargs={'user_id': self.object.id}),
Skia's avatar
Skia committed
134
                    'slug': 'godfathers',
Skia's avatar
Skia committed
135
136
                    'name': _("Godfathers"),
                    })
137
138
139
140
141
142
143
144
145
146
147
148
        if self.request.user == self.object:
            tab_list.append({
                        'url': reverse('core:user_tools'),
                        'slug': 'tools',
                        'name': _("Tools"),
                        })
        if self.request.user.can_edit(self.object):
            tab_list.append({
                        'url': reverse('core:user_edit', kwargs={'user_id': self.object.id}),
                        'slug': 'edit',
                        'name': _("Edit"),
                        })
Lo-J's avatar
Lo-J committed
149
150
151
152
153
154
        if self.request.user.can_view(self.object):
            tab_list.append({
                        'url': reverse('core:user_clubs', kwargs={'user_id': self.object.id}),
                        'slug': 'clubs',
                        'name': _("Clubs"),
                        })
155
156
157
158
159
160
161
162
163
        if self.request.user.is_owner(self.object):
            tab_list.append({
                        'url': reverse('core:user_groups', kwargs={'user_id': self.object.id}),
                        'slug': 'groups',
                        'name': _("Groups"),
                        })
        try:
            if (self.object.customer and (self.object == self.request.user
                or self.request.user.is_in_group(settings.SITH_GROUPS['accounting-admin']['name'])
164
                or self.request.user.is_in_group(settings.SITH_BAR_MANAGER['unix_name']+settings.SITH_BOARD_SUFFIX)
165
                or self.request.user.is_root)):
166
167
168
169
170
                tab_list.append({
                            'url': reverse('core:user_stats', kwargs={'user_id': self.object.id}),
                            'slug': 'stats',
                            'name': _("Stats"),
                            })
171
172
173
174
175
176
177
178
179
                tab_list.append({
                            'url': reverse('core:user_account', kwargs={'user_id': self.object.id}),
                            'slug': 'account',
                            'name': _("Account")+" (%s €)" % self.object.customer.amount,
                            })
        except: pass
        return tab_list

class UserView(UserTabsMixin, CanViewMixin, DetailView):
Skia's avatar
Skia committed
180
181
182
    """
    Display a user's profile
    """
Skia's avatar
Skia committed
183
184
185
    model = User
    pk_url_kwarg = "user_id"
    context_object_name = "profile"
Skia's avatar
Skia committed
186
    template_name = "core/user_detail.jinja"
187
    current_tab = 'infos'
Skia's avatar
Skia committed
188

Sli's avatar
Sli committed
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204

def DeleteUserGodfathers(request, user_id, godfather_id, is_father):
    user = User.objects.get(id=user_id)
    if ((user == request.user) or
         request.user.is_root or
         request.user.is_board_member):
        ud = get_object_or_404(User, id=godfather_id)
        if is_father == "True":
            user.godfathers.remove(ud)
        else:
            user.godchildren.remove(ud)
    else:
        raise PermissionDenied
    return redirect('core:user_godfathers', user_id=user_id)


Skia's avatar
Skia committed
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
class UserGodfathersView(UserTabsMixin, CanViewMixin, DetailView):
    """
    Display a user's godfathers
    """
    model = User
    pk_url_kwarg = "user_id"
    context_object_name = "profile"
    template_name = "core/user_godfathers.jinja"
    current_tab = 'godfathers'

    def post(self, request, *args, **kwargs):
        self.object = self.get_object()
        self.form = UserGodfathersForm(request.POST)
        if self.form.is_valid() and self.form.cleaned_data['user'] != self.object:
            if self.form.cleaned_data['type'] == 'godfather':
                self.object.godfathers.add(self.form.cleaned_data['user'])
                self.object.save()
            else:
                self.object.godchildren.add(self.form.cleaned_data['user'])
                self.object.save()
            self.form = UserGodfathersForm()
        return super(UserGodfathersView, self).get(request, *args, **kwargs)

    def get_context_data(self, **kwargs):
        kwargs = super(UserGodfathersView, self).get_context_data(**kwargs)
        try:
            kwargs['form'] = self.form
        except:
            kwargs['form'] = UserGodfathersForm()
        return kwargs

236
class UserStatsView(UserTabsMixin, CanViewMixin, DetailView):
237
238
239
240
241
242
243
    """
    Display a user's stats
    """
    model = User
    pk_url_kwarg = "user_id"
    context_object_name = "profile"
    template_name = "core/user_stats.jinja"
244
    current_tab = 'stats'
245
246
247

    def get_context_data(self, **kwargs):
        kwargs = super(UserStatsView, self).get_context_data(**kwargs)
Skia's avatar
Skia committed
248
249
        from counter.models import Counter, Product, Selling
        from django.db.models import Sum
Skia's avatar
Skia committed
250
251
252
        foyer = Counter.objects.filter(name="Foyer").first()
        mde = Counter.objects.filter(name="MDE").first()
        gommette = Counter.objects.filter(name="La Gommette").first()
Skia's avatar
Skia committed
253
        semester_start=Subscription.compute_start(d=date.today(), duration=3)
Skia's avatar
Skia committed
254
255
256
257
        kwargs['total_perm_time'] = sum([p.end-p.start for p in self.object.permanencies.exclude(end=None)], timedelta())
        kwargs['total_foyer_time'] = sum([p.end-p.start for p in self.object.permanencies.filter(counter=foyer).exclude(end=None)], timedelta())
        kwargs['total_mde_time'] = sum([p.end-p.start for p in self.object.permanencies.filter(counter=mde).exclude(end=None)], timedelta())
        kwargs['total_gommette_time'] = sum([p.end-p.start for p in self.object.permanencies.filter(counter=gommette).exclude(end=None)], timedelta())
Skia's avatar
Skia committed
258
259
260
261
262
263
        kwargs['total_foyer_buyings'] = sum([b.unit_price*b.quantity for b in
            self.object.customer.buyings.filter(counter=foyer, date__gte=semester_start)])
        kwargs['total_mde_buyings'] = sum([b.unit_price*b.quantity for b in self.object.customer.buyings.filter(counter=mde,
            date__gte=semester_start)])
        kwargs['total_gommette_buyings'] = sum([b.unit_price*b.quantity for b in
            self.object.customer.buyings.filter(counter=gommette, date__gte=semester_start)])
Skia's avatar
Skia committed
264
265
        kwargs['top_product'] = self.object.customer.buyings.values('product__name').annotate(
                product_sum=Sum('quantity')).exclude(product_sum=None).order_by('-product_sum').all()[:10]
266
267
        return kwargs

Skia's avatar
Skia committed
268
269
270
271
272
273
274
275
276
class UserMiniView(CanViewMixin, DetailView):
    """
    Display a user's profile
    """
    model = User
    pk_url_kwarg = "user_id"
    context_object_name = "profile"
    template_name = "core/user_mini.jinja"

Skia's avatar
Skia committed
277
278
279
280
281
class UserListView(ListView):
    """
    Displays the user list
    """
    model = User
Skia's avatar
Skia committed
282
    template_name = "core/user_list.jinja"
Skia's avatar
Skia committed
283

284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
class UserUploadProfilePictView(CanEditMixin, DetailView):
    """
    Handle the upload of the profile picture taken with webcam in navigator
    """
    model = User
    pk_url_kwarg = "user_id"
    template_name = "core/user_edit.jinja"

    def post(self, request, *args, **kwargs):
        from core.utils import resize_image
        from io import BytesIO
        from PIL import Image
        self.object = self.get_object()
        if self.object.profile_pict:
            raise ValidationError(_("User already has a profile picture"))
        f = request.FILES['new_profile_pict']
        parent = SithFile.objects.filter(parent=None, name="profiles").first()
        name = str(self.object.id) + "_profile.jpg" # Webcamejs uploads JPGs
        im = Image.open(BytesIO(f.read()))
        new_file = SithFile(parent=parent, name=name,
                file=resize_image(im, 400, f.content_type.split('/')[-1]),
                owner=self.object, is_folder=False, mime_type=f.content_type, size=f._size)
        new_file.file.name = name
        new_file.save()
        self.object.profile_pict = new_file
        self.object.save()
        return redirect("core:user_edit", user_id=self.object.id)

312
class UserUpdateProfileView(UserTabsMixin, CanEditMixin, UpdateView):
Skia's avatar
Skia committed
313
314
315
316
317
    """
    Edit a user's profile
    """
    model = User
    pk_url_kwarg = "user_id"
Skia's avatar
Skia committed
318
    template_name = "core/user_edit.jinja"
319
    form_class = UserProfileForm
320
    current_tab = "edit"
321
322
323
324

    def get(self, request, *args, **kwargs):
        self.object = self.get_object()
        self.form = self.get_form()
325
        if self.form.instance.profile_pict and not (request.user.is_board_member or request.user.is_root):
326
            self.form.fields.pop('profile_pict', None)
327
328
329
330
331
332
        if self.form.instance.date_of_birth and not (request.user.is_board_member or request.user.is_root):
            self.form.fields.pop('date_of_birth')
        if self.form.instance.first_name and not (request.user.is_board_member or request.user.is_root):
            self.form.fields.pop('first_name')
        if self.form.instance.last_name and not (request.user.is_board_member or request.user.is_root):
            self.form.fields.pop('last_name')
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
        return self.render_to_response(self.get_context_data(form=self.form))

    def post(self, request, *args, **kwargs):
        self.object = self.get_object()
        self.form = self.get_form()
        if self.form.instance.profile_pict and not request.user.is_in_group(settings.SITH_MAIN_BOARD_GROUP):
            self.form.fields.pop('profile_pict', None)
        files = request.FILES.items()
        self.form.process(files)
        if request.user.is_authenticated() and request.user.can_edit(self.object) and self.form.is_valid():
            return super(UserUpdateProfileView, self).form_valid(self.form)
        return self.form_invalid(self.form)

    def get_context_data(self, **kwargs):
        kwargs = super(UserUpdateProfileView, self).get_context_data(**kwargs)
        kwargs['profile'] = self.form.instance
        kwargs['form'] = self.form
        return kwargs
Skia's avatar
Skia committed
351

Lo-J's avatar
Lo-J committed
352
353
354
355
356
357
358
359
360
361
class UserClubView(UserTabsMixin, CanViewMixin, DetailView):
    """
    Display the user's club(s)
    """
    model = User
    context_object_name = "profile"
    pk_url_kwarg = "user_id"
    template_name = "core/user_clubs.jinja"
    current_tab = "clubs"

362
class UserUpdateGroupView(UserTabsMixin, CanEditPropMixin, UpdateView):
Skia's avatar
Skia committed
363
364
365
366
367
    """
    Edit a user's groups
    """
    model = User
    pk_url_kwarg = "user_id"
368
369
370
    template_name = "core/user_group.jinja"
    form_class = modelform_factory(User, fields=['groups'],
            widgets={'groups':CheckboxSelectMultiple})
Skia's avatar
Skia committed
371
    context_object_name = "profile"
372
    current_tab = "groups"
Skia's avatar
Skia committed
373

374
class UserToolsView(UserTabsMixin, TemplateView):
375
376
377
    """
    Displays the logged user's tools
    """
Skia's avatar
Skia committed
378
    template_name = "core/user_tools.jinja"
379
    current_tab = "tools"
Skia's avatar
Skia committed
380

381
    def get_context_data(self, **kwargs):
382
        self.object = self.request.user
383
384
385
        from launderette.models import Launderette
        kwargs = super(UserToolsView, self).get_context_data(**kwargs)
        kwargs['launderettes'] = Launderette.objects.all()
Skia's avatar
Skia committed
386
        kwargs['profile'] = self.request.user
387
        kwargs['object'] = self.request.user
388
389
        return kwargs

Sli's avatar
Sli committed
390
class UserAccountBase(UserTabsMixin, DetailView):
Skia's avatar
Skia committed
391
    """
Sli's avatar
Sli committed
392
    Base class for UserAccount
Skia's avatar
Skia committed
393
394
395
    """
    model = User
    pk_url_kwarg = "user_id"
396
    current_tab = "account"
Skia's avatar
Skia committed
397
398

    def dispatch(self, request, *arg, **kwargs): # Manually validates the rights
Sli's avatar
Sli committed
399
        res = super(UserAccountBase, self).dispatch(request, *arg, **kwargs)
Skia's avatar
Skia committed
400
401
        if (self.object == request.user
                or request.user.is_in_group(settings.SITH_GROUPS['accounting-admin']['name'])
402
                or request.user.is_in_group(settings.SITH_BAR_MANAGER['unix_name']+settings.SITH_BOARD_SUFFIX)
Skia's avatar
Skia committed
403
                or request.user.is_root):
Skia's avatar
Skia committed
404
405
406
            return res
        raise PermissionDenied

Sli's avatar
Sli committed
407
408
409
410
411
412
413
class UserAccountView(UserAccountBase):
    """
    Display a user's account
    """
    template_name = "core/user_account.jinja"

    def expense_by_month(self, obj, calc):
Sli's avatar
Sli committed
414
415
        stats = []

416
        for year in obj.datetimes('date', 'year', order='DESC'):
Sli's avatar
Sli committed
417
            stats.append([])
Sli's avatar
Sli committed
418
419
            i = 0
            for month in obj.filter(date__year=year.year).datetimes(
420
                'date', 'month', order='DESC'):
Sli's avatar
Sli committed
421
                q = obj.filter(
Sli's avatar
Sli committed
422
423
                    date__year=month.year,
                    date__month=month.month
Sli's avatar
Sli committed
424
                )
Sli's avatar
Sli committed
425
426
427
428
429
                stats[i].append((
                    sum([calc(p) for p in q]),
                    month
                ))
            i += 1
Sli's avatar
Sli committed
430
431
        return stats

Sli's avatar
Sli committed
432
433
434
435
436
437
    def invoices_calc(self, query):
        t = 0
        for it in query.items.all():
            t += it.quantity * it.product_unit_price
        return t

Skia's avatar
Skia committed
438
439
440
441
442
    def get_context_data(self, **kwargs):
        kwargs = super(UserAccountView, self).get_context_data(**kwargs)
        kwargs['profile'] = self.object
        try:
            kwargs['customer'] = self.object.customer
Sli's avatar
Sli committed
443
444
            kwargs['buyings_month'] = self.expense_by_month(
                self.object.customer.buyings,
Sli's avatar
Sli committed
445
                (lambda q: q.unit_price * q.quantity)
Sli's avatar
Sli committed
446
447
448
449
450
451
452
            )
            kwargs['invoices_month'] = self.expense_by_month(
                self.object.customer.user.invoices,
                self.invoices_calc
            )
            kwargs['refilling_month'] = self.expense_by_month(
                self.object.customer.refillings,
Sli's avatar
Sli committed
453
                (lambda q: q.amount)
Sli's avatar
Sli committed
454
            )
Skia's avatar
Skia committed
455
            kwargs['etickets'] = self.object.customer.buyings.exclude(product__eticket=None).all()
Skia's avatar
Skia committed
456
457
458
459
        except:
            pass
        return kwargs

Sli's avatar
Sli committed
460
461
462
463
464
465
466
467
468
class UserAccountDetailView(UserAccountBase, YearMixin, MonthMixin):
    """
    Display a user's account for month
    """
    template_name = "core/user_account_detail.jinja"

    def get_context_data(self, **kwargs):
        kwargs = super(UserAccountDetailView, self).get_context_data(**kwargs)
        kwargs['profile'] = self.object
469
470
        kwargs['year'] = self.get_year()
        kwargs['month'] = self.get_month()
Sli's avatar
Sli committed
471
472
473
474
475
476
        try:
            kwargs['customer'] = self.object.customer
        except:
            pass
        kwargs['tab'] = "account"
        return kwargs
477