from django.contrib import admin

from apps.referrals.models import PointsBalance, Referral, ReferralCode


@admin.register(ReferralCode)
class ReferralCodeAdmin(admin.ModelAdmin):
    list_display = ["user", "code", "created_at"]
    search_fields = ["code", "user__phone_number"]


@admin.register(Referral)
class ReferralAdmin(admin.ModelAdmin):
    list_display = ["referrer", "referred", "awarded_points", "created_at"]


@admin.register(PointsBalance)
class PointsBalanceAdmin(admin.ModelAdmin):
    list_display = ["user", "points"]