mirror of
https://github.com/django/django.git
synced 2025-10-24 14:16:09 +00:00
Refs #33476 -- Reformatted code with Black.
This commit is contained in:
committed by
Mariusz Felisiak
parent
f68fa8b45d
commit
9c19aff7c7
@@ -2,7 +2,9 @@ from functools import partial
|
||||
|
||||
from django.db import models
|
||||
from django.db.models.fields.related import (
|
||||
RECURSIVE_RELATIONSHIP_CONSTANT, ManyToManyDescriptor, RelatedField,
|
||||
RECURSIVE_RELATIONSHIP_CONSTANT,
|
||||
ManyToManyDescriptor,
|
||||
RelatedField,
|
||||
create_many_to_many_intermediary_model,
|
||||
)
|
||||
|
||||
@@ -12,29 +14,45 @@ class CustomManyToManyField(RelatedField):
|
||||
Ticket #24104 - Need to have a custom ManyToManyField,
|
||||
which is not an inheritor of ManyToManyField.
|
||||
"""
|
||||
|
||||
many_to_many = True
|
||||
|
||||
def __init__(self, to, db_constraint=True, swappable=True, related_name=None, related_query_name=None,
|
||||
limit_choices_to=None, symmetrical=None, through=None, through_fields=None, db_table=None, **kwargs):
|
||||
def __init__(
|
||||
self,
|
||||
to,
|
||||
db_constraint=True,
|
||||
swappable=True,
|
||||
related_name=None,
|
||||
related_query_name=None,
|
||||
limit_choices_to=None,
|
||||
symmetrical=None,
|
||||
through=None,
|
||||
through_fields=None,
|
||||
db_table=None,
|
||||
**kwargs,
|
||||
):
|
||||
try:
|
||||
to._meta
|
||||
except AttributeError:
|
||||
to = str(to)
|
||||
kwargs['rel'] = models.ManyToManyRel(
|
||||
self, to,
|
||||
kwargs["rel"] = models.ManyToManyRel(
|
||||
self,
|
||||
to,
|
||||
related_name=related_name,
|
||||
related_query_name=related_query_name,
|
||||
limit_choices_to=limit_choices_to,
|
||||
symmetrical=symmetrical if symmetrical is not None else (to == RECURSIVE_RELATIONSHIP_CONSTANT),
|
||||
symmetrical=symmetrical
|
||||
if symmetrical is not None
|
||||
else (to == RECURSIVE_RELATIONSHIP_CONSTANT),
|
||||
through=through,
|
||||
through_fields=through_fields,
|
||||
db_constraint=db_constraint,
|
||||
)
|
||||
self.swappable = swappable
|
||||
self.db_table = db_table
|
||||
if kwargs['rel'].through is not None and self.db_table is not None:
|
||||
if kwargs["rel"].through is not None and self.db_table is not None:
|
||||
raise ValueError(
|
||||
'Cannot specify a db_table if an intermediary model is used.'
|
||||
"Cannot specify a db_table if an intermediary model is used."
|
||||
)
|
||||
super().__init__(
|
||||
related_name=related_name,
|
||||
@@ -45,22 +63,32 @@ class CustomManyToManyField(RelatedField):
|
||||
|
||||
def contribute_to_class(self, cls, name, **kwargs):
|
||||
if self.remote_field.symmetrical and (
|
||||
self.remote_field.model == "self" or self.remote_field.model == cls._meta.object_name):
|
||||
self.remote_field.model == "self"
|
||||
or self.remote_field.model == cls._meta.object_name
|
||||
):
|
||||
self.remote_field.related_name = "%s_rel_+" % name
|
||||
super().contribute_to_class(cls, name, **kwargs)
|
||||
if not self.remote_field.through and not cls._meta.abstract and not cls._meta.swapped:
|
||||
self.remote_field.through = create_many_to_many_intermediary_model(self, cls)
|
||||
if (
|
||||
not self.remote_field.through
|
||||
and not cls._meta.abstract
|
||||
and not cls._meta.swapped
|
||||
):
|
||||
self.remote_field.through = create_many_to_many_intermediary_model(
|
||||
self, cls
|
||||
)
|
||||
setattr(cls, self.name, ManyToManyDescriptor(self.remote_field))
|
||||
self.m2m_db_table = partial(self._get_m2m_db_table, cls._meta)
|
||||
|
||||
def get_internal_type(self):
|
||||
return 'ManyToManyField'
|
||||
return "ManyToManyField"
|
||||
|
||||
# Copy those methods from ManyToManyField because they don't call super() internally
|
||||
contribute_to_related_class = models.ManyToManyField.__dict__['contribute_to_related_class']
|
||||
_get_m2m_attr = models.ManyToManyField.__dict__['_get_m2m_attr']
|
||||
_get_m2m_reverse_attr = models.ManyToManyField.__dict__['_get_m2m_reverse_attr']
|
||||
_get_m2m_db_table = models.ManyToManyField.__dict__['_get_m2m_db_table']
|
||||
contribute_to_related_class = models.ManyToManyField.__dict__[
|
||||
"contribute_to_related_class"
|
||||
]
|
||||
_get_m2m_attr = models.ManyToManyField.__dict__["_get_m2m_attr"]
|
||||
_get_m2m_reverse_attr = models.ManyToManyField.__dict__["_get_m2m_reverse_attr"]
|
||||
_get_m2m_db_table = models.ManyToManyField.__dict__["_get_m2m_db_table"]
|
||||
|
||||
|
||||
class InheritedManyToManyField(models.ManyToManyField):
|
||||
@@ -71,5 +99,6 @@ class MediumBlobField(models.BinaryField):
|
||||
"""
|
||||
A MySQL BinaryField that uses a different blob size.
|
||||
"""
|
||||
|
||||
def db_type(self, connection):
|
||||
return 'MEDIUMBLOB'
|
||||
return "MEDIUMBLOB"
|
||||
|
||||
Reference in New Issue
Block a user