django-cqrs/tests/dj_master/models.py

#  Copyright © 2023 Ingram Micro Inc. All rights reserved.

from django.contrib.contenttypes.fields import GenericForeignKey
from django.contrib.contenttypes.models import ContentType
from django.db import models
from mptt.models import MPTTModel, TreeForeignKey

from dj_cqrs.constants import ALL_BASIC_FIELDS
from dj_cqrs.metas import MasterMeta
from dj_cqrs.mixins import MasterMixin, RawMasterMixin


class BasicFieldsModel(MasterMixin, models.Model):
    CQRS_ID = 'basic'
    CQRS_TRACKED_FIELDS = ALL_BASIC_FIELDS

    int_field = models.IntegerField(primary_key=True)
    bool_field = models.BooleanField(null=True)
    char_field = models.CharField(max_length=200, null=True)
    date_field = models.DateField(null=True)
    datetime_field = models.DateTimeField(null=True)
    float_field = models.FloatField(null=True)
    url_field = models.URLField(null=True)
    uuid_field = models.UUIDField(null=True)


class AllFieldsModel(MasterMixin, models.Model):
    CQRS_FIELDS = ALL_BASIC_FIELDS
    CQRS_ID = 'all'
    CQRS_TRACKED_FIELDS = ALL_BASIC_FIELDS

    int_field = models.IntegerField(null=True)
    char_field = models.CharField(max_length=200, null=True)


class ChosenFieldsModel(MasterMixin, models.Model):
    CQRS_FIELDS = ('char_field', 'id')
    CQRS_ID = 'chosen'

    float_field = models.IntegerField(null=True)
    char_field = models.CharField(max_length=200, null=True)


class AutoFieldsModel(MasterMixin, models.Model):
    CQRS_ID = 'auto'

    created = models.DateTimeField(auto_now_add=True)
    updated = models.DateTimeField(auto_now=True)


class SimplestModel(MasterMixin, models.Model):
    CQRS_ID = 'pk'

    id = models.IntegerField(primary_key=True)
    name = models.CharField(max_length=200, null=True)


class SimplestTrackedModel(MasterMixin):
    CQRS_ID = 'pk_tracked'
    CQRS_TRACKED_FIELDS = ('status', 'description')

    id = models.IntegerField(primary_key=True)
    status = models.CharField(max_length=200, null=True)
    description = models.CharField(max_length=200, null=True)


class FilteredSimplestModel(MasterMixin, models.Model):
    CQRS_ID = 'filter'

    name = models.CharField(max_length=200)

    def is_sync_instance(self):
        return len(str(self.name)) > 2


class Publisher(models.Model):
    id = models.IntegerField(primary_key=True)
    name = models.CharField(max_length=20)


class Author(MasterMixin, models.Model):
    CQRS_ID = 'author'
    CQRS_SERIALIZER = 'tests.dj_master.serializers.AuthorSerializer'

    id = models.IntegerField(primary_key=True)
    name = models.CharField(max_length=20)

    publisher = models.ForeignKey(
        Publisher,
        related_name='authors',
        on_delete=models.SET_NULL,
        null=True,
    )

    @classmethod
    def relate_cqrs_serialization(cls, queryset):
        return queryset.select_related('publisher').prefetch_related('books')


class Book(models.Model):
    id = models.IntegerField(primary_key=True)
    title = models.CharField(max_length=20)

    author = models.ForeignKey(Author, related_name='books', on_delete=models.CASCADE)


class BadSerializationClassModel(MasterMixin, models.Model):
    CQRS_ID = 'bad_serialization'
    CQRS_SERIALIZER = 'tests.Serializer'


class BadQuerySetSerializationClassModel(MasterMixin, models.Model):
    CQRS_ID = 'bad_queryset'
    CQRS_SERIALIZER = 'tests.dj_master.serializers.AuthorSerializer'

    @classmethod
    def relate_cqrs_serialization(cls, queryset):
        return queryset.none()


class NonMetaClassModel(MPTTModel, RawMasterMixin):
    CQRS_ID = 'non_meta'

    name = models.CharField(max_length=50, unique=True)
    parent = TreeForeignKey(
        'self',
        on_delete=models.CASCADE,
        null=True,
        blank=True,
        related_name='children',
    )


MasterMeta.register(NonMetaClassModel)


class NonSentModel(MasterMixin, models.Model):
    CQRS_ID = 'non_sent'
    CQRS_PRODUCE = False


class TrackedFieldsParentModel(MasterMixin, models.Model):
    CQRS_ID = 'tracked_parent'
    CQRS_TRACKED_FIELDS = ALL_BASIC_FIELDS

    char_field = models.CharField(max_length=10)


class TrackedFieldsChildModel(MasterMixin):
    CQRS_ID = 'tracked_child'
    CQRS_TRACKED_FIELDS = ('char_field', 'parent')

    char_field = models.CharField(max_length=10)
    parent = models.ForeignKey(TrackedFieldsParentModel, on_delete=models.CASCADE, null=True)


class TrackedFieldsAllWithChildModel(MasterMixin):
    CQRS_ID = 'tracked_child_all'
    CQRS_TRACKED_FIELDS = '__all__'

    char_field = models.CharField(max_length=10)
    parent = models.ForeignKey(TrackedFieldsParentModel, on_delete=models.CASCADE)


class MPTTWithTrackingModel(MPTTModel, RawMasterMixin):
    CQRS_ID = 'with_tracking'

    CQRS_TRACKED_FIELDS = '__all__'

    name = models.CharField(max_length=50, unique=True)
    parent = TreeForeignKey(
        'self',
        on_delete=models.CASCADE,
        null=True,
        blank=True,
        related_name='children',
    )


MasterMeta.register(MPTTWithTrackingModel)


class WithGenericFKModel(MasterMixin):
    CQRS_ID = 'with_generic_fk'

    CQRS_TRACKED_FIELDS = '__all__'

    content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE, null=True, blank=True)
    object_id = models.PositiveIntegerField(null=True)
    content_object = GenericForeignKey('content_type', 'object_id')


class M2MModel(models.Model):
    CQRS_ID = 'm2m_model'

    id = models.IntegerField(primary_key=True)
    name = models.CharField(max_length=200, null=True)


class WithM2MModel(MasterMixin):
    CQRS_ID = 'with_m2m_fk'

    CQRS_TRACKED_FIELDS = '__all__'

    char_field = models.CharField(max_length=100)
    m2m_field = models.ManyToManyField(M2MModel)


class FailModel(MasterMixin, models.Model):
    CQRS_ID = 'fail'