import os.path
from uuid import uuid4

from django.db import models
from djantic import ModelSchema


def upload_image_handler(instance, filename):
    _, ext = os.path.splitext(filename)
    _uuid = uuid4()
    return f'{_uuid}{ext}'


class NameMixin(models.Model):
    name = models.CharField(max_length=128)

    class Meta:
        abstract = True


class ImageMixin(models.Model):
    image = models.ImageField(blank=True, null=True, upload_to=upload_image_handler)

    class Meta:
        abstract = True


class A(NameMixin):
    a = models.IntegerField()
   
    
class B(ImageMixin, NameMixin):
    b = models.IntegerField()
    
    
class ASchema(ModelSchema):
    class Config:
        model = A


class BSchema(ModelSchema):
    class Config:
        model = B