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
from payments.admin import UserCouponInline, UserLedgerInline
def make_user_link(user):
change_url = resolve_url("admin:auth_user_change", user.id)
return format_html('{}', change_url, user.username)
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 UserAdmin(UserAdmin):
inlines = (VPNUserInline, UserLedgerInline, UserCouponInline)
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 = '{}'
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)
admin.site.unregister(User)
admin.site.register(User, UserAdmin)