from django.contrib import admin

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


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


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


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


class PicAdmin(admin.ModelAdmin):
    list_display = ('href',)


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