from django.contrib import admin

from .models import (
    DepositTransaction, WithdrawalTransaction,
    AdminWalletAddress, UserProfile,
    Top10Deposit, Top10Withdrawal,
    ReferralBonus,
    TransactionRecord,EmailUser,
    UserEarningRecord, PendingDeposit,  
    PendingWithdrawl,CompanyYoutubeTubeVideo, AddBonus
)

# ADDING SEARCH FIELD
class DepositTransactionAdmin(admin.ModelAdmin):
    search_fields=("user__username",)
    list_display = ("user", "user_amount", "current_balance", "date_deposited", "status")


# ADDING SEARCH FIELD
class WithdrawalTransactionAdmin(admin.ModelAdmin):
    search_fields=("user__username",)
    list_display = ("user", "amount_to_withdraw", "withdrawal_date", "status")


# ADDING SEARCH FIELD
class UserProfileAdmin(admin.ModelAdmin):
    search_fields=("user__username",)
    list_display = ("user", "plan_type", "earning_rate", "referrer")

    
# ADDING SEARCH FIELD
class UserEarningRecordAdmin(admin.ModelAdmin):
    search_fields=("user",)
    list_display = ("user", "total_deposit", "total_withdrawal", "last_deposit", "last_withdrawal")
    
    
# ADDING SEARCH FIELD
class ReferralBonusAdmin(admin.ModelAdmin):
    search_fields=("username",)
    list_display = ("username", "total_person_referred", "accumulated_bonus")
    

# ADDING SEARCH FIELD
class TransactionRecordAdmin(admin.ModelAdmin):
    search_fields=("username",)
    list_display = ("username", "amount", "transaction_type", "date", "status")
    
# ADDING SEARCH FIELD
class AdminWalletAddressAdmin(admin.ModelAdmin):
    search_fields=("wallet_type",)
    list_display = ("wallet_type", "wallet_address")
    

# Adding Search Field
class AddBonusAdmin(admin.ModelAdmin):
    search_fields = ("user__username",)
    list_display = ("user", "current_bonus")
    
    
admin.site.register(UserProfile, UserProfileAdmin)
admin.site.register(DepositTransaction, DepositTransactionAdmin)
admin.site.register(WithdrawalTransaction, WithdrawalTransactionAdmin)
admin.site.register(ReferralBonus, ReferralBonusAdmin)
admin.site.register(TransactionRecord, TransactionRecordAdmin)
admin.site.register(AdminWalletAddress, AdminWalletAddressAdmin)
# admin.site.register(Top10Deposit)
# admin.site.register(Top10Withdrawal)
# admin.site.register(CompanyYoutubeTubeVideo)
admin.site.register(EmailUser)
admin.site.register(UserEarningRecord, UserEarningRecordAdmin)
# admin.site.register(AddBonus, AddBonusAdmin)
# admin.site.register(PendingDeposit)
# admin.site.register(PendingWithdrawl)

admin.site.site_header = "COMPANY ADMINISTRATOR"
admin.site.site_title = "COMPANY ADMIN"
admin.site.site_url = ""