mainapp/admin.py - main - eb/django_pycoding_shop

mainapp/admin.py
73 строки3 KB
EvgeniyBudaev
Merge branch 'main' of https://github.com/EvgeniyBudaev/django_pycoding_shop into main
30 май 2021, 15:52
30 май 2021, 15:52801eee7
from django.contrib import admin
from django.core.exceptions import ValidationError
from django.forms import ModelChoiceField, ModelForm
from django.utils.safestring import mark_safe

from .models import *


class NotebookAdminForm(ModelForm):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields['image'].help_text = mark_safe(
            f'<span style="color:red;font'
            f'-size:12px;">При'
            f' загрузке изображения с '
            f'разрешением больше '
            f'{Product.MAX_RESOLUTION} оно '
            f'будет обрезано! </span>')

        # def clean_image(self):
    #   image = self.cleaned_data['image']
    #   img = Image.open(image)
    #   min_height, min_width = Product.MIN_RESOLUTION
    #   max_height, max_width = Product.MAX_RESOLUTION
    #   if image.size > Product.MAX_IMAGE_SIZE:
    #     raise ValidationError('Размер изображения не должен превышать 3Mb!')
    #   if img.height < min_height or img.width < min_width:
    #     raise ValidationError('Разрешение изображения меньше минимального!')
    #   if img.height > max_height or img.width > max_width:
    #     raise ValidationError('Разрешение изображения больше максимального!')
    #   return image

class SmartphoneAdminForm(ModelForm):
  def __init__(self, *args, **kwargs):
    super().__init__(*args, **kwargs)
    instance = kwargs.get('instance')
    if not instance.sd:
      self.fields['sd_volume_max'].widgeet.attrs.update({
        'readonly': True,
        'style': 'background: lightgray;'
      })

  def clean(self):
    if not self.cleaned_data['sd']:
      self.cleaned_data['sd_volume_max'] = None
    return self.cleaned_data


class NotebookAdmin(admin.ModelAdmin):
    form = NotebookAdminForm

    def formfield_for_foreignkey(self, db_field, request, **kwargs):
        if db_field.name == 'category':
            return ModelChoiceField(Category.objects.filter(slug='notebooks'))
        return super().formfield_for_foreignkey(db_field, request, **kwargs)


class SmartphoneAdmin(admin.ModelAdmin):
  change_form_template = 'admin.html'
  form = SmartphoneAdminForm

  def formfield_for_foreignkey(self, db_field, request, **kwargs):
    if db_field.name == 'category':
      return ModelChoiceField(Category.objects.filter(slug='smartphones'))
    return super().formfield_for_foreignkey(db_field, request, **kwargs)

admin.site.register(Category)
admin.site.register(Notebook, NotebookAdmin)
admin.site.register(Smartphone, SmartphoneAdmin)
admin.site.register(CartProduct)
admin.site.register(Cart)
admin.site.register(Customer)
admin.site.register(Order)