from django.contrib import admin

# Register your models here.
from choose.models import Tea, TeaType, TeaCategory


class TeaCategoryAdmin(admin.ModelAdmin):
    list_display = ('name', 'shop_name')


class TeaTypeAdmin(admin.ModelAdmin):
    list_display = ('name', 'shop_name', 'category')


class TeaAdmin(admin.ModelAdmin):
    list_display = ('name', 'price', 'pic', 'type')


admin.site.register(Tea, TeaAdmin)
admin.site.register(TeaType, TeaTypeAdmin)
admin.site.register(TeaCategory, TeaCategoryAdmin)