summaryrefslogtreecommitdiff
path: root/tests/serializers/models/natural.py
blob: 3b01385e8f81ed843aa778f2d4d42c4b23ab36c8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
"""Models for test_natural.py"""

import uuid

from django.contrib.auth.base_user import AbstractBaseUser
from django.db import models


class NaturalKeyAnchorManager(models.Manager):
    def get_by_natural_key(self, data):
        return self.get(data=data)


class NaturalKeyAnchor(models.Model):
    data = models.CharField(max_length=100, unique=True)
    title = models.CharField(max_length=100, null=True)

    objects = NaturalKeyAnchorManager()

    def natural_key(self):
        return (self.data,)


class FKDataNaturalKey(models.Model):
    data = models.ForeignKey(NaturalKeyAnchor, models.SET_NULL, null=True)


class NaturalKeyThing(models.Model):
    key = models.CharField(max_length=100, unique=True)
    other_thing = models.ForeignKey(
        "NaturalKeyThing", on_delete=models.CASCADE, null=True
    )
    other_things = models.ManyToManyField(
        "NaturalKeyThing", related_name="thing_m2m_set"
    )

    class Manager(models.Manager):
        def get_by_natural_key(self, key):
            return self.get(key=key)

    objects = Manager()

    def natural_key(self):
        return (self.key,)

    def __str__(self):
        return self.key


class NaturalPKWithDefault(models.Model):
    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
    name = models.CharField(max_length=100, unique=True)

    class Manager(models.Manager):
        def get_by_natural_key(self, name):
            return self.get(name=name)

    objects = Manager()

    def natural_key(self):
        return (self.name,)


class FKAsPKNoNaturalKeyManager(models.Manager):
    def get_by_natural_key(self, *args, **kwargs):
        return super().get_by_natural_key(*args, **kwargs)


class FKAsPKNoNaturalKey(models.Model):
    pk_fk = models.ForeignKey(
        NaturalKeyAnchor, on_delete=models.CASCADE, primary_key=True
    )

    objects = FKAsPKNoNaturalKeyManager()

    def natural_key(self):
        raise NotImplementedError("This method was not expected to be called.")


class SubclassNaturalKeyOptOutUser(AbstractBaseUser):
    email = models.EmailField(unique=False, null=True, blank=True)
    USERNAME_FIELD = "email"

    def natural_key(self):
        return ()


class PostToOptOutSubclassUser(models.Model):
    author = models.ForeignKey(SubclassNaturalKeyOptOutUser, on_delete=models.CASCADE)
    title = models.CharField(max_length=100)
    subscribers = models.ManyToManyField(
        SubclassNaturalKeyOptOutUser, related_name="subscribed_posts", blank=True
    )


class NaturalKeyWithNullableFieldManager(models.Manager):
    def get_by_natural_key(self, name, optional_id):
        return self.get(name=name, optional_id=optional_id)


class NaturalKeyWithNullableField(models.Model):
    name = models.CharField(max_length=100)
    optional_id = models.CharField(max_length=100, null=True, blank=True)

    objects = NaturalKeyWithNullableFieldManager()

    class Meta:
        unique_together = [["name", "optional_id"]]

    def natural_key(self):
        return (self.name, self.optional_id)


class FKToNaturalKeyWithNullable(models.Model):
    ref = models.ForeignKey(
        NaturalKeyWithNullableField, on_delete=models.CASCADE, null=True
    )
    refs = models.ManyToManyField(
        NaturalKeyWithNullableField, related_name="m2m_referrers"
    )
    data = models.CharField(max_length=100, default="")