Revert "feat!: check django model has a default ordering when used in a relay connection (#1495)"

This reverts commit 96c09ac439.
This commit is contained in:
Kien Dang 2024-04-11 21:53:08 +08:00
parent ea45de02ad
commit fede977f8a
6 changed files with 16 additions and 69 deletions

View File

@ -24,9 +24,6 @@ class Faction(models.Model):
class Ship(models.Model):
class Meta:
ordering = ["pk"]
name = models.CharField(max_length=50)
faction = models.ForeignKey(Faction, on_delete=models.CASCADE, related_name="ships")

View File

@ -101,19 +101,13 @@ class DjangoConnectionField(ConnectionField):
non_null = True
assert issubclass(
_type, DjangoObjectType
), "DjangoConnectionField only accepts DjangoObjectType types as underlying type"
), "DjangoConnectionField only accepts DjangoObjectType types"
assert _type._meta.connection, "The type {} doesn't have a connection".format(
_type.__name__
)
connection_type = _type._meta.connection
if non_null:
return NonNull(connection_type)
# Since Relay Connections require to have a predictible ordering for pagination,
# check on init that the Django model provided has a default ordering declared.
model = connection_type._meta.node._meta.model
assert (
len(getattr(model._meta, "ordering", [])) > 0
), f"Django model {model._meta.app_label}.{model.__name__} has to have a default ordering to be used in a Connection."
return connection_type
@property

View File

@ -26,9 +26,6 @@ else:
class Event(models.Model):
class Meta:
ordering = ["pk"]
name = models.CharField(max_length=50)
tags = ArrayField(models.CharField(max_length=50))
tag_ids = ArrayField(models.IntegerField())

View File

@ -5,9 +5,6 @@ CHOICES = ((1, "this"), (2, _("that")))
class Person(models.Model):
class Meta:
ordering = ["pk"]
name = models.CharField(max_length=30)
parent = models.ForeignKey(
"self", on_delete=models.CASCADE, null=True, blank=True, related_name="children"
@ -15,9 +12,6 @@ class Person(models.Model):
class Pet(models.Model):
class Meta:
ordering = ["pk"]
name = models.CharField(max_length=30)
age = models.PositiveIntegerField()
owner = models.ForeignKey(
@ -37,9 +31,6 @@ class FilmDetails(models.Model):
class Film(models.Model):
class Meta:
ordering = ["pk"]
genre = models.CharField(
max_length=2,
help_text="Genre",
@ -55,9 +46,6 @@ class DoeReporterManager(models.Manager):
class Reporter(models.Model):
class Meta:
ordering = ["pk"]
first_name = models.CharField(max_length=30)
last_name = models.CharField(max_length=30)
email = models.EmailField()

View File

@ -2,12 +2,11 @@ import datetime
import re
import pytest
from django.db.models import Count, Model, Prefetch
from django.db.models import Count, Prefetch
from graphene import List, NonNull, ObjectType, Schema, String
from graphene.relay import Node
from ..fields import DjangoConnectionField, DjangoListField
from ..fields import DjangoListField
from ..types import DjangoObjectType
from .models import (
Article as ArticleModel,
@ -717,34 +716,3 @@ class TestDjangoListField:
r'SELECT .* FROM "tests_film" INNER JOIN "tests_film_reporters" .* LEFT OUTER JOIN "tests_filmdetails"',
captured.captured_queries[1]["sql"],
)
class TestDjangoConnectionField:
def test_model_ordering_assertion(self):
class Chaos(Model):
class Meta:
app_label = "test"
class ChaosType(DjangoObjectType):
class Meta:
model = Chaos
interfaces = (Node,)
class Query(ObjectType):
chaos = DjangoConnectionField(ChaosType)
with pytest.raises(
TypeError,
match=r"Django model test\.Chaos has to have a default ordering to be used in a Connection\.",
):
Schema(query=Query)
def test_only_django_object_types(self):
class Query(ObjectType):
something = DjangoConnectionField(String)
with pytest.raises(
TypeError,
match="DjangoConnectionField only accepts DjangoObjectType types as underlying type",
):
Schema(query=Query)

View File

@ -1,4 +1,3 @@
import warnings
from collections import OrderedDict, defaultdict
from textwrap import dedent
from unittest.mock import patch
@ -400,7 +399,7 @@ def test_django_objecttype_fields_exist_on_model():
with pytest.warns(
UserWarning,
match=r"Field name .* matches an attribute on Django model .* but it's not a model field",
):
) as record:
class Reporter2(DjangoObjectType):
class Meta:
@ -408,8 +407,7 @@ def test_django_objecttype_fields_exist_on_model():
fields = ["first_name", "some_method", "email"]
# Don't warn if selecting a custom field
with warnings.catch_warnings():
warnings.simplefilter("error")
with pytest.warns(None) as record:
class Reporter3(DjangoObjectType):
custom_field = String()
@ -418,6 +416,8 @@ def test_django_objecttype_fields_exist_on_model():
model = ReporterModel
fields = ["first_name", "custom_field", "email"]
assert len(record) == 0
@with_local_registry
def test_django_objecttype_exclude_fields_exist_on_model():
@ -445,14 +445,15 @@ def test_django_objecttype_exclude_fields_exist_on_model():
exclude = ["custom_field"]
# Don't warn on exclude fields
with warnings.catch_warnings():
warnings.simplefilter("error")
with pytest.warns(None) as record:
class Reporter4(DjangoObjectType):
class Meta:
model = ReporterModel
exclude = ["email", "first_name"]
assert len(record) == 0
@with_local_registry
def test_django_objecttype_neither_fields_nor_exclude():
@ -466,22 +467,24 @@ def test_django_objecttype_neither_fields_nor_exclude():
class Meta:
model = ReporterModel
with warnings.catch_warnings():
warnings.simplefilter("error")
with pytest.warns(None) as record:
class Reporter2(DjangoObjectType):
class Meta:
model = ReporterModel
fields = ["email"]
with warnings.catch_warnings():
warnings.simplefilter("error")
assert len(record) == 0
with pytest.warns(None) as record:
class Reporter3(DjangoObjectType):
class Meta:
model = ReporterModel
exclude = ["email"]
assert len(record) == 0
def custom_enum_name(field):
return f"CustomEnum{field.name.title()}"