26 lines
743 B
Python
26 lines
743 B
Python
from django.contrib import admin
|
|
from django.contrib.auth.admin import UserAdmin
|
|
from django.contrib.auth.models import User
|
|
from .models import UserProfile, Signal, Trade, Backtest
|
|
|
|
class UserProfileInline(admin.StackedInline):
|
|
model = UserProfile
|
|
can_delete = False
|
|
verbose_name_plural = 'Profile'
|
|
fk_name = 'user'
|
|
|
|
class CustomUserAdmin(UserAdmin):
|
|
inlines = (UserProfileInline, )
|
|
|
|
def get_inline_instances(self, request, obj=None):
|
|
if not obj:
|
|
return list()
|
|
return super(CustomUserAdmin, self).get_inline_instances(request, obj)
|
|
|
|
|
|
admin.site.unregister(User)
|
|
admin.site.register(User, CustomUserAdmin)
|
|
|
|
admin.site.register(Signal)
|
|
admin.site.register(Trade)
|
|
admin.site.register(Backtest) |