You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
152 lines
5.0 KiB
Python
152 lines
5.0 KiB
Python
import string
|
|
from django.shortcuts import resolve_url
|
|
from django import forms
|
|
from django.contrib import admin
|
|
from django.contrib import messages
|
|
from django.contrib.auth.admin import UserAdmin
|
|
from django.contrib.auth.models import User
|
|
from django.utils.translation import ugettext as _
|
|
from django.utils.html import format_html
|
|
import django_lcore
|
|
import lcoreapi
|
|
|
|
from lambdainst.models import VPNUser, GiftCode, GiftCodeUser
|
|
|
|
|
|
def make_user_link(user):
|
|
change_url = resolve_url('admin:auth_user_change', user.id)
|
|
return format_html('<a href="{}">{}</a>', change_url, user.username)
|
|
|
|
|
|
class GiftCodeAdminForm(forms.ModelForm):
|
|
def clean(self):
|
|
input_code = self.cleaned_data.get('code', '')
|
|
code_charset = string.ascii_letters + string.digits
|
|
if any(c not in code_charset for c in input_code):
|
|
raise forms.ValidationError(_("Code must be [a-zA-Z0-9]"))
|
|
if not 1 <= len(input_code) <= 32:
|
|
raise forms.ValidationError(_("Code must be between 1 and 32 characters"))
|
|
return self.cleaned_data
|
|
|
|
|
|
class VPNUserInline(admin.StackedInline):
|
|
model = VPNUser
|
|
can_delete = False
|
|
fk_name = 'user'
|
|
|
|
fields = ('notes', 'expiration', 'last_expiry_notice', 'notify_expiration',
|
|
'last_vpn_auth_fail_notice', 'notify_vpn_auth_fail',
|
|
'trial_periods_given', 'referrer_a', 'campaign', 'last_vpn_auth', 'last_core_sync')
|
|
readonly_fields = ('referrer_a', 'last_vpn_auth', 'last_core_sync', 'campaign')
|
|
|
|
def referrer_a(self, object):
|
|
if not object.referrer:
|
|
return "-"
|
|
|
|
s = make_user_link(object.referrer) + " "
|
|
if object.referrer_used:
|
|
s += _("(rewarded)")
|
|
else:
|
|
s += _("(not rewarded)")
|
|
return s
|
|
referrer_a.allow_tags = True
|
|
referrer_a.short_description = _("Referrer")
|
|
|
|
def is_paid(self, object):
|
|
return object.is_paid
|
|
is_paid.boolean = True
|
|
is_paid.short_description = _("Is paid?")
|
|
|
|
|
|
class GiftCodeUserAdmin(admin.TabularInline):
|
|
model = GiftCodeUser
|
|
fields = ('user_link', 'code_link', 'date')
|
|
readonly_fields = ('user_link', 'code_link', 'date')
|
|
list_display = ('user', )
|
|
original = False
|
|
|
|
def user_link(self, object):
|
|
return make_user_link(object.user)
|
|
user_link.short_description = 'User'
|
|
|
|
def code_link(self, object):
|
|
change_url = resolve_url('admin:lambdainst_giftcode_change', object.code.id)
|
|
return format_html('<a href="{}">{}</a>', change_url, object.code.code)
|
|
code_link.short_description = 'Code'
|
|
|
|
def has_add_permission(self, request, obj):
|
|
return False
|
|
|
|
def has_delete_permission(self, request, obj=None):
|
|
return False
|
|
|
|
|
|
class UserAdmin(UserAdmin):
|
|
inlines = (VPNUserInline, GiftCodeUserAdmin)
|
|
list_display = ('username', 'email', 'is_staff', 'date_joined', 'is_paid')
|
|
ordering = ('-date_joined', )
|
|
fieldsets = (
|
|
(None, {'fields': ('username', 'password', 'email', 'links')}),
|
|
(_('Important dates'), {'fields': ('last_login', 'date_joined')}),
|
|
(_('Permissions'), {'fields': ('is_active', 'is_staff', 'is_superuser',
|
|
'groups', 'user_permissions')}),
|
|
)
|
|
readonly_fields = ('last_login', 'date_joined', 'links')
|
|
actions = (django_lcore.core_sync_action, )
|
|
|
|
def is_paid(self, object):
|
|
return object.vpnuser.is_paid
|
|
is_paid.boolean = True
|
|
is_paid.short_description = _("Is paid?")
|
|
|
|
def links(self, object):
|
|
payments_url = resolve_url('admin:payments_payment_changelist')
|
|
tickets_url = resolve_url('admin:tickets_ticket_changelist')
|
|
fmt = '<a href="{}?user__id__exact={}">{}</a>'
|
|
return format_html(fmt + " - " + fmt,
|
|
payments_url, object.id, "Payments",
|
|
tickets_url, object.id, "Tickets",
|
|
)
|
|
|
|
def save_model(self, request, obj, form, change):
|
|
super().save_model(request, obj, form, change)
|
|
|
|
# Notify core
|
|
if change:
|
|
django_lcore.sync_user(obj.vpnuser)
|
|
|
|
def delete_model(self, request, obj):
|
|
try:
|
|
django_lcore.api.get_user(obj.username).delete()
|
|
except lcoreapi.APIError as e:
|
|
messages.error(request, "failed to delete vpn user: %r" % e)
|
|
|
|
super().delete_model(request, obj)
|
|
|
|
|
|
class GiftCodeAdmin(admin.ModelAdmin):
|
|
fields = ('code', 'time', 'created', 'created_by', 'single_use', 'free_only',
|
|
'available', 'comment')
|
|
readonly_fields = ('created', 'created_by')
|
|
list_display = ('code', 'time', 'comment_head', 'available')
|
|
search_fields = ('code', 'comment', 'users__username')
|
|
inlines = (GiftCodeUserAdmin,)
|
|
list_filter = ('available', 'time')
|
|
form = GiftCodeAdminForm
|
|
|
|
def comment_head(self, object):
|
|
return object.comment_head
|
|
comment_head.short_description = _("Comment")
|
|
|
|
def save_model(self, request, obj, form, change):
|
|
if not change:
|
|
obj.created_by = request.user
|
|
obj.save()
|
|
|
|
|
|
admin.site.unregister(User)
|
|
admin.site.register(User, UserAdmin)
|
|
|
|
admin.site.register(GiftCode, GiftCodeAdmin)
|
|
|