from django.contrib import admin
from django.contrib.auth.admin import UserAdmin as BaseUserAdmin
from django.contrib.auth.models import User
from django.utils.html import format_html
from django.urls import reverse
from .models import GoalignProfile, GoalignAccount
from apps.audit.export import export_as_csv_action
class GoalignProfileInline(admin.StackedInline):
model = GoalignProfile
can_delete = False
verbose_name_plural = 'Goalign Profile'
class CustomUserAdmin(BaseUserAdmin):
inlines = [GoalignProfileInline]
admin.site.unregister(User)
admin.site.register(User, CustomUserAdmin)
@admin.register(GoalignProfile)
class GoalignProfileAdmin(admin.ModelAdmin):
list_display = ('user', 'firebase_uid_short')
search_fields = ('user__username', 'user__email', 'firebase_uid')
readonly_fields = ('firebase_uid',)
def firebase_uid_short(self, obj):
val = obj.firebase_uid or '-'
return format_html('{}...', val, val[:20])
firebase_uid_short.short_description = "Firebase UID"
@admin.register(GoalignAccount)
class GoalignAccountAdmin(admin.ModelAdmin):
list_display = ('id_formatted', 'email_cell', 'display_name', 'vip_badge', 'role_badge', 'kyc_status_badge', 'quick_actions', 'created_at')
list_filter = ('vip_level', 'is_admin', 'kyc_status', 'created_at')
search_fields = ('email', 'display_name', 'firebase_uid', 'web3_eth_address')
readonly_fields = ('firebase_uid', 'web3_eth_address', 'created_at', 'updated_at')
actions = [export_as_csv_action("Export Selected Accounts to CSV", filename_prefix="goalign_accounts")]
def id_formatted(self, obj):
return format_html('#{}', obj.id)
id_formatted.short_description = "ID"
def email_cell(self, obj):
return format_html('{}', obj.email)
email_cell.short_description = "Email"
def vip_badge(self, obj):
return format_html('VIP {}', obj.vip_level)
vip_badge.short_description = "VIP Tier"
def role_badge(self, obj):
if obj.is_admin:
return format_html('SUPER ADMIN')
return format_html('TRADER')
role_badge.short_description = "Role"
def kyc_status_badge(self, obj):
colors = {
'verified': '#10B981',
'pending': '#F59E0B',
'rejected': '#EF4444',
'unverified': '#64748B'
}
color = colors.get(obj.kyc_status, '#64748B')
return format_html('{}', color, obj.kyc_status)
kyc_status_badge.short_description = "KYC Status"
def quick_actions(self, obj):
if obj.kyc_status == 'pending':
url = reverse('admin:kyc_queue')
return format_html('Review KYC', url)
return "-"
quick_actions.short_description = "Review Action"