# noinspection PyUnresolvedReferences from constance.admin import Config from constance.admin import ConstanceAdmin as BaseConstanceAdmin from django.apps import apps from django.contrib import admin from django.contrib.admin import ModelAdmin, TabularInline from django.contrib.admin.options import InlineModelAdmin from django.contrib.gis.admin import GISModelAdmin from django.db.models import Model from django.forms import modelform_factory from django.urls import path from django.utils.translation import gettext_lazy as _ from modeltranslation.translator import translator from modeltranslation.utils import get_translation_fields from mptt.admin import DraggableMPTTAdmin from evibes.settings import CONSTANCE_CONFIG from .forms import OrderForm, OrderProductForm, VendorForm from .models import ( Address, Attribute, AttributeGroup, AttributeValue, Brand, Category, CategoryTag, Feedback, Order, OrderProduct, Product, ProductImage, ProductTag, PromoCode, Promotion, Stock, Vendor, Wishlist, ) SECTION_GENERAL = _("general") SECTION_I18N = _("I18N") SECTION_META = _("metadata") SECTION_DATES = _("timestamps") SECTION_RELATIONS = _("relations") class I18NFieldsetMixin: model: type[Model] def get_inline_instances(self, request, obj=None): inlines = super().get_inline_instances(request, obj) trans_opts = translator.get_options_for_model(self.model) translation_fields = [] for orig in trans_opts.local_fields: translation_fields += get_translation_fields(orig) class _TranslationInline(InlineModelAdmin[self.model, self.model]): model = self.model form = modelform_factory( self.model, fields=translation_fields, formfield_callback=lambda f, **kw: super( I18NFieldsetMixin, self ).formfield_for_dbfield(f, request, **kw), ) template = "admin/edit_inline/tabular.html" is_navtab = True verbose_name = _("translations") verbose_name_plural = _("translations") extra = 0 def get_queryset(self, request): return ( self.model.objects.filter(pk=obj.pk) if obj else self.model.objects.none() ) def get_formset(self, request, obj=None, **kw): return super().get_formset( request, obj, **{ **kw, "form": self.form, "fields": translation_fields, "exclude": [], }, ) inlines.append(_TranslationInline(self.model, self.admin_site)) return inlines class BasicModelAdmin(ModelAdmin): @admin.action(description=str(_("activate selected %(verbose_name_plural)s"))) def activate_selected(self, _request, queryset) -> str: queryset.update(is_active=True) return "" @admin.action(description=str(_("deactivate selected %(verbose_name_plural)s"))) def deactivate_selected(self, _request, queryset) -> str: queryset.update(is_active=False) return "" def get_actions(self, request): actions = super().get_actions(request) actions["activate_selected"] = ( self.activate_selected, "activate_selected", str(_("activate selected %(verbose_name_plural)s")), ) actions["deactivate_selected"] = ( self.deactivate_selected, "deactivate_selected", str(_("deactivate selected %(verbose_name_plural)s")), ) return actions class AttributeValueInline(TabularInline): model = AttributeValue extra = 0 is_navtab = True verbose_name = _("attribute value") verbose_name_plural = _("attribute values") autocomplete_fields = ["attribute"] @admin.register(AttributeGroup) class AttributeGroupAdmin(BasicModelAdmin, I18NFieldsetMixin): list_display = ("name", "modified") search_fields = ( "uuid", "name", ) fieldsets = ( ( SECTION_GENERAL, { "fields": ("name", "parent", "is_active"), }, ), ( SECTION_META, { "fields": ("uuid",), }, ), ) @admin.register(Attribute) class AttributeAdmin(BasicModelAdmin, I18NFieldsetMixin): list_display = ("name", "group", "value_type", "modified") list_filter = ("value_type", "group", "is_active") search_fields = ("uuid", "name", "group__name") autocomplete_fields = ["categories", "group"] @admin.register(AttributeValue) class AttributeValueAdmin(BasicModelAdmin, I18NFieldsetMixin): list_display = ("attribute", "value", "modified") list_filter = ("attribute__group", "is_active") search_fields = ("uuid", "value", "attribute__name") autocomplete_fields = ["attribute"] class CategoryChildrenInline(admin.TabularInline): model = Category fk_name = "parent" extra = 0 fields = ("name", "description", "is_active", "image", "markup_percent") @admin.register(Category) class CategoryAdmin(DraggableMPTTAdmin, BasicModelAdmin, I18NFieldsetMixin): mptt_indent_field = "name" list_display = ("indented_title", "parent", "is_active", "modified") # noinspection PyUnresolvedReferences list_filter = ("is_active", "level", "created", "modified") list_display_links = ("indented_title",) search_fields = ( "uuid", "name", ) inlines = [CategoryChildrenInline] fieldsets = ( ( SECTION_GENERAL, { "fields": ( "name", "slug", "parent", "is_active", ), }, ), ( SECTION_RELATIONS, { "fields": ("tags",), }, ), ( SECTION_META, { "fields": ("uuid",), }, ), ( SECTION_DATES, { "fields": ("created", "modified"), }, ), ) autocomplete_fields = ["parent", "tags"] readonly_fields = ( "uuid", "slug", "created", "modified", ) def indented_title(self, instance): return instance.name indented_title.short_description = _("name") # type: ignore indented_title.admin_order_field = "name" # type: ignore @admin.register(Brand) class BrandAdmin(BasicModelAdmin, I18NFieldsetMixin): list_display = ("name",) list_filter = ("categories", "is_active") search_fields = ( "uuid", "name", "categories__name", ) readonly_fields = ( "uuid", "slug", ) class ProductImageInline(TabularInline): model = ProductImage extra = 0 is_navtab = True verbose_name = _("image") verbose_name_plural = _("images") class StockInline(TabularInline): model = Stock extra = 0 is_navtab = True verbose_name = _("stock") verbose_name_plural = _("stocks") @admin.register(Product) class ProductAdmin(BasicModelAdmin, I18NFieldsetMixin): list_display = ( "name", "partnumber", "is_active", "category", "brand", "price", "rating", "modified", ) list_filter = ( "is_active", "is_digital", ("tags", admin.RelatedOnlyFieldListFilter), ("stocks__vendor", admin.RelatedOnlyFieldListFilter), "created", "modified", ) search_fields = ( "name", "partnumber", "brand__name", "category__name", "uuid", "slug", ) readonly_fields = ("created", "modified", "uuid", "rating", "price", "slug") autocomplete_fields = ("category", "brand", "tags") def price(self, obj): return obj.price price.short_description = _("price") # type: ignore def rating(self, obj): return obj.rating rating.short_description = _("rating") # type: ignore fieldsets = ( ( SECTION_GENERAL, { "fields": ( "name", "slug", "partnumber", "is_active", "is_digital", ), }, ), ( SECTION_RELATIONS, { "fields": ("category", "brand", "tags"), }, ), ( SECTION_META, { "fields": ("uuid",), }, ), ( SECTION_DATES, { "fields": ("created", "modified"), }, ), ) inlines = [AttributeValueInline, ProductImageInline, StockInline] def get_changelist(self, request, **kwargs): changelist = super().get_changelist(request, **kwargs) changelist.filter_input_length = 64 return changelist @admin.register(ProductTag) class ProductTagAdmin(BasicModelAdmin, I18NFieldsetMixin): list_display = ("name",) search_fields = ("name",) @admin.register(CategoryTag) class CategoryTagAdmin(BasicModelAdmin, I18NFieldsetMixin): list_display = ("name",) search_fields = ("name",) @admin.register(Vendor) class VendorAdmin(BasicModelAdmin): list_display = ("name", "markup_percent", "modified") list_filter = ("markup_percent", "is_active") search_fields = ("name",) form = VendorForm @admin.register(Feedback) class FeedbackAdmin(BasicModelAdmin): list_display = ("order_product", "rating", "comment", "modified") list_filter = ("rating", "is_active") search_fields = ("order_product__product__name", "comment") class OrderProductInline(admin.TabularInline): model = OrderProduct extra = 0 readonly_fields = ("product", "quantity", "buy_price") form = OrderProductForm is_navtab = True verbose_name = _("order product") verbose_name_plural = _("order products") def get_queryset(self, request): qs = super().get_queryset(request) return qs.select_related("product").only("product__name") @admin.register(Order) class OrderAdmin(BasicModelAdmin): list_display = ( "human_readable_id", "user", "is_business", "status", "total_price", "buy_time", "modified", ) list_filter = ("status", "buy_time", "modified", "created") search_fields = ("user__email", "status", "uuid", "human_readable_id") inlines = [OrderProductInline] form = OrderForm readonly_fields = ("total_price", "total_quantity", "buy_time", "human_readable_id") def is_business(self, obj): return obj.is_business is_business.short_description = _("is business") # type: ignore def get_queryset(self, request): qs = super().get_queryset(request) return qs.prefetch_related( "user", "shipping_address", "billing_address", "order_products", "promo_code", ) def save_model(self, request, obj, form, change): if form.cleaned_data.get("attributes") is None: obj.attributes = None if form.cleaned_data.get("notifications") is None: obj.attributes = None super().save_model(request, obj, form, change) @admin.register(OrderProduct) class OrderProductAdmin(BasicModelAdmin): list_display = ("order", "product", "quantity", "buy_price", "status", "modified") list_filter = ("status",) search_fields = ("order__user__email", "product__name") form = OrderProductForm def get_queryset(self, request): qs = super().get_queryset(request) return qs.prefetch_related("order", "product") def save_model(self, request, obj, form, change): if form.cleaned_data.get("attributes") is None: obj.attributes = None if form.cleaned_data.get("notifications") is None: obj.attributes = None super().save_model(request, obj, form, change) @admin.register(PromoCode) class PromoCodeAdmin(BasicModelAdmin): list_display = ( "code", "discount_percent", "discount_amount", "start_time", "end_time", "used_on", ) list_filter = ("discount_percent", "discount_amount", "start_time", "end_time") search_fields = ("code",) def get_queryset(self, request): qs = super().get_queryset(request) return qs.prefetch_related("user") @admin.register(Promotion) class PromotionAdmin(BasicModelAdmin, I18NFieldsetMixin): list_display = ("name", "discount_percent", "modified") search_fields = ("name",) autocomplete_fields = ("products",) def get_queryset(self, request): qs = super().get_queryset(request) return qs.prefetch_related("products") @admin.register(Stock) class StockAdmin(BasicModelAdmin): list_display = ("product", "vendor", "sku", "quantity", "price", "modified") list_filter = ("vendor", "quantity") search_fields = ("product__name", "vendor__name", "sku") autocomplete_fields = ("product", "vendor") @admin.register(Wishlist) class WishlistAdmin(BasicModelAdmin): list_display = ("user", "modified") search_fields = ("user__email",) @admin.register(ProductImage) class ProductImageAdmin(BasicModelAdmin): list_display = ("alt", "product", "priority", "modified") list_filter = ("priority",) search_fields = ("alt", "product__name") autocomplete_fields = ("product",) @admin.register(Address) class AddressAdmin(GISModelAdmin): list_display = ("street", "city", "region", "country", "user") list_filter = ("country", "region") search_fields = ( "raw_data", "street", "city", "postal_code", "user__email", "address_line", ) gis_widget_kwargs = { "attrs": { "default_lon": 37.61556, "default_lat": 55.75222, "default_zoom": 6, } } class ConstanceAdmin(BaseConstanceAdmin): def get_urls(self): info = f"{self.model._meta.app_label}_{self.model._meta.model_name}" return [ path( "", self.admin_site.admin_view(self.changelist_view), name=f"{info}_changelist", ), path( "", self.admin_site.admin_view(self.changelist_view), name=f"{info}_add" ), ] class ConstanceConfig: class Meta: app_label = "core" object_name = "Config" concrete_model = None model_name = module_name = "config" verbose_name_plural = _("config") abstract = False swapped = False is_composite_pk = False def get_ordered_objects(self): return False def get_change_permission(self): return f"change_{self.model_name}" @property def app_config(self): return apps.get_app_config(self.app_label) @property def label(self): return f"{self.app_label}.{self.object_name}" @property def label_lower(self): return f"{self.app_label}.{self.model_name}" _meta = Meta() admin.site.unregister([Config]) # type: ignore admin.site.register([ConstanceConfig], ConstanceAdmin) # type: ignore admin.site.site_title = f"{CONSTANCE_CONFIG.get('PROJECT_NAME')[0]}" # type: ignore admin.site.site_header = "eVibes" admin.site.index_title = f"{CONSTANCE_CONFIG.get('PROJECT_NAME')[0]}" # type: ignore