from django.contrib import admin
from apps.root.admin import custom_admin_site
from import_export.admin import ImportExportModelAdmin

from apps.subscriptions.models import Plan, Subscription, Payment, Extention

# Register your models here.

@admin.register(Plan,site=custom_admin_site)
class PlanAdmin(ImportExportModelAdmin, admin.ModelAdmin):
    list_display = ("plan_name", "price","plan_duration", "paymentSource", "duration_days", "videos_per_day", "time_per_video", "upload_youtube_perm", "download_youtube_perm", "file_size", "parts_limits","created")
    save_as = True

@admin.register(Subscription,site=custom_admin_site)
class SubscriptionAdmin(ImportExportModelAdmin, admin.ModelAdmin):
     list_display = ("plan", "user", "created_at", "ends_time")

@admin.register(Payment,site=custom_admin_site)
class PaymentAdmin(ImportExportModelAdmin, admin.ModelAdmin):
     list_display = ("transactionID", "conversationID", "subscription", "is_active") 


@admin.register(Extention,site=custom_admin_site)
class ExtentionAdmin(ImportExportModelAdmin, admin.ModelAdmin):
    pass