from django.contrib import admin

from .models import Investment, Reward


@admin.register(Investment)
class InvestmentAdmin(admin.ModelAdmin):
    list_display = (
        "id",
        "user",
        "amount",
        "tier",
        "daily_reward_rate",
        "duration_days",
        "status",
        "start_date",
        "end_date",
        "created_at",
    )
    list_filter = ("status", "tier")
    search_fields = ("user__telegram_user_id", "user__username")
    ordering = ("-created_at",)


@admin.register(Reward)
class RewardAdmin(admin.ModelAdmin):
    list_display = (
        "id",
        "investment",
        "amount",
        "reward_date",
        "calculated_at",
        "distributed_at",
    )
    list_filter = ("reward_date",)
    search_fields = ("investment__user__telegram_user_id",)
    ordering = ("-reward_date",)