88 lines
3.1 KiB
Python
88 lines
3.1 KiB
Python
# -*- mode: python; coding: utf-8 -*-
|
|
# Copyright (C) 2017-2019 by BDE ENS Paris-Saclay
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
from django.contrib import admin
|
|
from django.contrib import messages
|
|
from django.contrib.auth.admin import UserAdmin as BaseUserAdmin
|
|
from django.contrib.auth.forms import PasswordResetForm
|
|
from django.urls import reverse
|
|
from django.utils.html import format_html
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from reversion.admin import VersionAdmin
|
|
from med.admin import admin_site
|
|
|
|
from .models import User
|
|
|
|
|
|
class IsMemberFilter(admin.SimpleListFilter):
|
|
title = _('membership status')
|
|
parameter_name = 'is_member'
|
|
|
|
def lookups(self, request, model_admin):
|
|
return (
|
|
('Yes', _('Yes')),
|
|
)
|
|
|
|
def queryset(self, request, queryset):
|
|
# FIXME Replace with imported Note Kfet memberships
|
|
return queryset
|
|
|
|
|
|
class UserAdmin(VersionAdmin, BaseUserAdmin):
|
|
# Customize admin to add more fields
|
|
fieldsets = (
|
|
(None, {'fields': ('username', 'password')}),
|
|
(_('Personal info'), {'fields': ('first_name', 'last_name', 'email',
|
|
'phone_number', 'address',
|
|
'comment')}),
|
|
(_('Permissions'), {'fields': ('is_active', 'is_staff', 'is_superuser',
|
|
'groups', 'user_permissions')}),
|
|
(_('Important dates'), {'fields': ('last_login', 'date_joined')}),
|
|
)
|
|
list_display = ('username', 'email', 'first_name', 'last_name',
|
|
'is_member', 'is_staff')
|
|
list_filter = (IsMemberFilter, 'is_staff', 'is_superuser', 'is_active',
|
|
'groups')
|
|
|
|
def save_model(self, request, obj, form, change):
|
|
"""
|
|
On creation, send a password init mail
|
|
"""
|
|
super().save_model(request, obj, form, change)
|
|
|
|
if not change:
|
|
# Virtually fill the password reset form
|
|
password_reset = PasswordResetForm(data={'email': obj.email})
|
|
if password_reset.is_valid():
|
|
password_reset.save(request=request,
|
|
use_https=request.is_secure())
|
|
messages.success(request, _("An email to set the password"
|
|
" was sent."))
|
|
else:
|
|
messages.error(request, _("The email is invalid."))
|
|
|
|
def is_member(self, obj):
|
|
"""
|
|
Get current membership year and check if user is there
|
|
"""
|
|
# FIXME Use NK20
|
|
is_member = True
|
|
if is_member:
|
|
return format_html(
|
|
'<img src="/static/admin/img/icon-yes.svg" alt="True">'
|
|
)
|
|
else:
|
|
return format_html(
|
|
'<img src="/static/admin/img/icon-no.svg" alt="False"> '
|
|
'<a class="button" href="{}">{}</a>',
|
|
reverse('users:adherer', args=[obj.pk]),
|
|
_('Adhere')
|
|
)
|
|
|
|
is_member.short_description = _('is member')
|
|
is_member.allow_tags = True
|
|
|
|
|
|
admin_site.register(User, UserAdmin)
|