init

app/user/admin.py

@@ -0,0 +1,51 @@
+from django.contrib import admin
+from django.contrib.auth.admin import UserAdmin as BaseUserAdmin
+from django.template.defaultfilters import filesizeformat
+
+from .forms import UserCreationForm, UserChangeForm
+from .models import User, FriendRequest, Invitation
+
+
+@admin.register(User)
+class UserAdmin(BaseUserAdmin):
+    # add_form = UserCreationForm
+    form = UserChangeForm
+    fieldsets = BaseUserAdmin.fieldsets + (
+        ["Custom Fields", {
+            "fields": ["max_size", "friends"]
+        }]
+    ,)
+    list_display = ["username", "email", "is_superuser", "is_active", "is_staff", "display_max_size", "size_used"]
+    add_fieldsets = (
+        (None, {
+            "classes": ("wide",),
+            "fields": ("username", "email", "max_size", "password1", "password2"),
+        }),
+    )
+
+    def display_max_size(self, obj: User):
+        return filesizeformat(obj.max_size)
+    display_max_size.short_description = "Max size"
+
+    def size_used(self, obj: User):
+        return filesizeformat(obj.size_used)
+    size_used.short_description = "Size used"
+
+    def save_formset(self, request, form, formset, change):
+        print("save_formset")
+        return super().save_formset(request, form, formset, change)
+
+    def save_model(self, request, obj, form, change):
+        print("save_model")
+        return super().save_model(request, obj, form, change)
+
+
+
+@admin.register(Invitation)
+class InvitationAdmin(admin.ModelAdmin):
+    pass
+
+
+@admin.register(FriendRequest)
+class FriendRequestAdmin(admin.ModelAdmin):
+    pass