mirror of
https://github.com/jazzband/django-eav2.git
synced 2026-03-17 06:50:24 +00:00
b5b576aca5
Replace flake8 with ruff and apply consistent linting rules across the entire codebase. Update type annotations, quotation marks, and other style-related changes to comply with the new standards.
142 lines
3.4 KiB
Python
142 lines
3.4 KiB
Python
import uuid
|
|
from typing import Final, final
|
|
|
|
from django.db import models
|
|
|
|
from eav.decorators import register_eav
|
|
from eav.managers import EntityManager
|
|
from eav.models import EAVModelMeta
|
|
|
|
#: Constants
|
|
MAX_CHARFIELD_LEN: Final = 254
|
|
|
|
|
|
class TestBase(models.Model):
|
|
"""Base class for test models."""
|
|
|
|
class Meta:
|
|
"""Define common options."""
|
|
|
|
app_label = "test_project"
|
|
abstract = True
|
|
|
|
|
|
class DoctorManager(EntityManager):
|
|
"""
|
|
Custom manager for the Doctor model.
|
|
|
|
This manager extends the EntityManager and provides additional
|
|
methods specific to the Doctor model, and is expected to be the
|
|
default manager on the model.
|
|
"""
|
|
|
|
def get_by_name(self, name: str) -> models.QuerySet:
|
|
"""Returns a QuerySet of doctors with the given name.
|
|
|
|
Args:
|
|
name (str): The name of the doctor to search for.
|
|
|
|
Returns:
|
|
models.QuerySet: A QuerySet of doctors with the specified name.
|
|
"""
|
|
return self.filter(name=name)
|
|
|
|
|
|
class DoctorSubstringManager(models.Manager):
|
|
"""
|
|
Custom manager for the Doctor model.
|
|
|
|
This is a second manager used to ensure during testing that it's not replaced
|
|
as the default manager after eav.register().
|
|
"""
|
|
|
|
def get_by_name_contains(self, substring: str) -> models.QuerySet:
|
|
"""Returns a QuerySet of doctors whose names contain the given substring.
|
|
|
|
Args:
|
|
substring (str): The substring to search for in the doctor's name.
|
|
|
|
Returns:
|
|
models.QuerySet: A QuerySet of doctors whose names contain the
|
|
specified substring.
|
|
"""
|
|
return self.filter(name__icontains=substring)
|
|
|
|
|
|
@final
|
|
@register_eav()
|
|
class Doctor(TestBase):
|
|
"""Test model using UUID as primary key."""
|
|
|
|
id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
|
|
name = models.CharField(max_length=MAX_CHARFIELD_LEN)
|
|
|
|
objects = DoctorManager()
|
|
substrings = DoctorSubstringManager()
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
@final
|
|
class Patient(TestBase):
|
|
name = models.CharField(max_length=MAX_CHARFIELD_LEN)
|
|
email = models.EmailField(max_length=MAX_CHARFIELD_LEN, blank=True)
|
|
example = models.ForeignKey(
|
|
"ExampleModel",
|
|
null=True,
|
|
blank=True,
|
|
on_delete=models.PROTECT,
|
|
)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def __repr__(self):
|
|
return self.name
|
|
|
|
|
|
class Encounter(TestBase):
|
|
num = models.PositiveSmallIntegerField()
|
|
patient = models.ForeignKey(Patient, on_delete=models.PROTECT)
|
|
|
|
def __str__(self):
|
|
return "%s: encounter num %d" % (self.patient, self.num)
|
|
|
|
def __repr__(self):
|
|
return self.name
|
|
|
|
|
|
@register_eav()
|
|
@final
|
|
class ExampleModel(TestBase):
|
|
name = models.CharField(max_length=MAX_CHARFIELD_LEN)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
@register_eav()
|
|
@final
|
|
class M2MModel(TestBase):
|
|
name = models.CharField(max_length=MAX_CHARFIELD_LEN)
|
|
models = models.ManyToManyField(ExampleModel)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
@final
|
|
class ExampleMetaclassModel(TestBase, metaclass=EAVModelMeta):
|
|
name = models.CharField(max_length=MAX_CHARFIELD_LEN)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
@final
|
|
class RegisterTestModel(TestBase, metaclass=EAVModelMeta):
|
|
name = models.CharField(max_length=MAX_CHARFIELD_LEN)
|
|
|
|
def __str__(self):
|
|
return self.name
|