Source code for evennia.accounts.admin

#
# This sets up how models are displayed
# in the web admin interface.
#
from django import forms
from django.conf import settings
from django.contrib import admin, messages
from django.contrib.admin.options import IS_POPUP_VAR
from django.contrib.auth.admin import UserAdmin as BaseUserAdmin
from django.contrib.auth.forms import UserChangeForm, UserCreationForm
from django.contrib.admin.utils import unquote
from django.template.response import TemplateResponse
from django.http import Http404, HttpResponseRedirect
from django.core.exceptions import PermissionDenied
from django.views.decorators.debug import sensitive_post_parameters
from django.utils.decorators import method_decorator
from django.utils.html import escape
from django.urls import path, reverse
from django.contrib.auth import update_session_auth_hash

from evennia.accounts.models import AccountDB
from evennia.typeclasses.admin import AttributeInline, TagInline
from evennia.utils import create

sensitive_post_parameters_m = method_decorator(sensitive_post_parameters())


# handle the custom User editor
[docs]class AccountDBChangeForm(UserChangeForm): """ Modify the accountdb class. """
[docs] class Meta(object): model = AccountDB fields = "__all__"
username = forms.RegexField( label="Username", max_length=30, regex=r"^[\w. @+-]+$", widget=forms.TextInput(attrs={"size": "30"}), error_messages={ "invalid": "This value may contain only letters, spaces, numbers " "and @/./+/-/_ characters." }, help_text="30 characters or fewer. Letters, spaces, digits and " "@/./+/-/_ only.", )
[docs] def clean_username(self): """ Clean the username and check its existence. """ username = self.cleaned_data["username"] if username.upper() == self.instance.username.upper(): return username elif AccountDB.objects.filter(username__iexact=username): raise forms.ValidationError("An account with that name " "already exists.") return self.cleaned_data["username"]
[docs]class AccountDBCreationForm(UserCreationForm): """ Create a new AccountDB instance. """
[docs] class Meta(object): model = AccountDB fields = "__all__"
username = forms.RegexField( label="Username", max_length=30, regex=r"^[\w. @+-]+$", widget=forms.TextInput(attrs={"size": "30"}), error_messages={ "invalid": "This value may contain only letters, spaces, numbers " "and @/./+/-/_ characters." }, help_text="30 characters or fewer. Letters, spaces, digits and " "@/./+/-/_ only.", )
[docs] def clean_username(self): """ Cleanup username. """ username = self.cleaned_data["username"] if AccountDB.objects.filter(username__iexact=username): raise forms.ValidationError("An account with that name already " "exists.") return username
[docs]class AccountForm(forms.ModelForm): """ Defines how to display Accounts """
[docs] class Meta(object): model = AccountDB fields = "__all__" app_label = "accounts"
db_key = forms.RegexField( label="Username", initial="AccountDummy", max_length=30, regex=r"^[\w. @+-]+$", required=False, widget=forms.TextInput(attrs={"size": "30"}), error_messages={ "invalid": "This value may contain only letters, spaces, numbers" " and @/./+/-/_ characters." }, help_text="This should be the same as the connected Account's key " "name. 30 characters or fewer. Letters, spaces, digits and " "@/./+/-/_ only.", ) db_typeclass_path = forms.CharField( label="Typeclass", initial=settings.BASE_ACCOUNT_TYPECLASS, widget=forms.TextInput(attrs={"size": "78"}), help_text="Required. Defines what 'type' of entity this is. This " "variable holds a Python path to a module with a valid " "Evennia Typeclass. Defaults to " "settings.BASE_ACCOUNT_TYPECLASS.", ) db_permissions = forms.CharField( label="Permissions", initial=settings.PERMISSION_ACCOUNT_DEFAULT, required=False, widget=forms.TextInput(attrs={"size": "78"}), help_text="In-game permissions. A comma-separated list of text " "strings checked by certain locks. They are often used for " "hierarchies, such as letting an Account have permission " "'Admin', 'Builder' etc. An Account permission can be " "overloaded by the permissions of a controlled Character. " "Normal accounts use 'Accounts' by default.", ) db_lock_storage = forms.CharField( label="Locks", widget=forms.Textarea(attrs={"cols": "100", "rows": "2"}), required=False, help_text="In-game lock definition string. If not given, defaults " "will be used. This string should be on the form " "<i>type:lockfunction(args);type2:lockfunction2(args);...", ) db_cmdset_storage = forms.CharField( label="cmdset", initial=settings.CMDSET_ACCOUNT, widget=forms.TextInput(attrs={"size": "78"}), required=False, help_text="python path to account cmdset class (set in " "settings.CMDSET_ACCOUNT by default)", )
[docs]class AccountInline(admin.StackedInline): """ Inline creation of Account """ model = AccountDB template = "admin/accounts/stacked.html" form = AccountForm fieldsets = ( ( "In-game Permissions and Locks", { "fields": ("db_lock_storage",), # {'fields': ('db_permissions', 'db_lock_storage'), "description": "<i>These are permissions/locks for in-game use. " "They are unrelated to website access rights.</i>", }, ), ( "In-game Account data", { "fields": ("db_typeclass_path", "db_cmdset_storage"), "description": "<i>These fields define in-game-specific properties " "for the Account object in-game.</i>", }, ), ) extra = 1 max_num = 1
[docs]class AccountTagInline(TagInline): """ Inline Account Tags. """ model = AccountDB.db_tags.through related_field = "accountdb"
[docs]class AccountAttributeInline(AttributeInline): """ Inline Account Attributes. """ model = AccountDB.db_attributes.through related_field = "accountdb"
[docs]class AccountDBAdmin(BaseUserAdmin): """ This is the main creation screen for Users/accounts """ list_display = ("username", "email", "is_staff", "is_superuser") form = AccountDBChangeForm add_form = AccountDBCreationForm inlines = [AccountTagInline, AccountAttributeInline] fieldsets = ( (None, {"fields": ("username", "password", "email")}), ( "Website profile", { "fields": ("first_name", "last_name"), "description": "<i>These are not used " "in the default system.</i>", }, ), ( "Website dates", { "fields": ("last_login", "date_joined"), "description": "<i>Relevant only to the website.</i>", }, ), ( "Website Permissions", { "fields": ("is_active", "is_staff", "is_superuser", "user_permissions", "groups"), "description": "<i>These are permissions/permission groups for " "accessing the admin site. They are unrelated to " "in-game access rights.</i>", }, ), ( "Game Options", { "fields": ("db_typeclass_path", "db_cmdset_storage", "db_lock_storage"), "description": "<i>These are attributes that are more relevant " "to gameplay.</i>", }, ), ) # ('Game Options', {'fields': ( # 'db_typeclass_path', 'db_cmdset_storage', # 'db_permissions', 'db_lock_storage'), # 'description': '<i>These are attributes that are ' # 'more relevant to gameplay.</i>'})) add_fieldsets = ( ( None, { "fields": ("username", "password1", "password2", "email"), "description": "<i>These account details are shared by the admin " "system and the game.</i>", }, ), )
[docs] @sensitive_post_parameters_m def user_change_password(self, request, id, form_url=""): user = self.get_object(request, unquote(id)) if not self.has_change_permission(request, user): raise PermissionDenied if user is None: raise Http404("%(name)s object with primary key %(key)r does not exist.") % { "name": self.model._meta.verbose_name, "key": escape(id), } if request.method == "POST": form = self.change_password_form(user, request.POST) if form.is_valid(): form.save() change_message = self.construct_change_message(request, form, None) self.log_change(request, user, change_message) msg = "Password changed successfully." messages.success(request, msg) update_session_auth_hash(request, form.user) return HttpResponseRedirect( reverse( "%s:%s_%s_change" % ( self.admin_site.name, user._meta.app_label, # the model_name is something we need to hardcode # since our accountdb is a proxy: "accountdb", ), args=(user.pk,), ) ) else: form = self.change_password_form(user) fieldsets = [(None, {"fields": list(form.base_fields)})] adminForm = admin.helpers.AdminForm(form, fieldsets, {}) context = { "title": "Change password: %s" % escape(user.get_username()), "adminForm": adminForm, "form_url": form_url, "form": form, "is_popup": (IS_POPUP_VAR in request.POST or IS_POPUP_VAR in request.GET), "add": True, "change": False, "has_delete_permission": False, "has_change_permission": True, "has_absolute_url": False, "opts": self.model._meta, "original": user, "save_as": False, "show_save": True, **self.admin_site.each_context(request), } request.current_app = self.admin_site.name return TemplateResponse( request, self.change_user_password_template or "admin/auth/user/change_password.html", context, )
[docs] def save_model(self, request, obj, form, change): """ Custom save actions. Args: request (Request): Incoming request. obj (Object): Object to save. form (Form): Related form instance. change (bool): False if this is a new save and not an update. """ obj.save() if not change: # calling hooks for new account obj.set_class_from_typeclass(typeclass_path=settings.BASE_ACCOUNT_TYPECLASS) obj.basetype_setup() obj.at_account_creation()
[docs] def response_add(self, request, obj, post_url_continue=None): from django.http import HttpResponseRedirect from django.urls import reverse return HttpResponseRedirect(reverse("admin:accounts_accountdb_change", args=[obj.id]))
admin.site.register(AccountDB, AccountDBAdmin)