graphql-python / graphql-python/graphene

FragmentSpread' object has no attribute 'selection_set' with custom Middleware

Offen
#1,354 0 Kommentare 0 Reaktionen 0 zugewiesene Personen Auf GitHub ansehen
🐛 bug
Vorherrschende Sprache
Python
Sterne
8.2k
Forks
818
PR-Merge-Kennzahlen
Keine gemergten PRs in 30 T.

Beschreibung

Hello, I am using custom middleware to mesure complexity of graphql query, but it throws an error `FragmentSpread' object has no attribute 'selection_set' with custom Middleware` when using fragments.
Query example
```
query{
getProfile(id:1){
...test
}
}

fragment test on ProfileType{
id
}
```
The middleware
```python
from graphql.backend.core import GraphQLCoreBackend

def measure_depth(selection_set, level=1):
current_depth = level
for field in selection_set.selections:
if field.selection_set:
new_depth = measure_depth(field.selection_set, level=level + 1)
if new_depth > current_depth:
current_depth = new_depth
if new_depth > 3:
raise Exception('Query is too nested')
return current_depth

class DepthAnalysisBackend(GraphQLCoreBackend):
def document_from_string(self, schema, document_string):
document = super().document_from_string(schema, document_string)
ast = document.document_ast
for definition in ast.definitions:
if len(definition.selection_set.selections) > 1:
raise Exception("Query is too complex")
measure_depth(definition.selection_set)

return document
```

How to fix this, and what Fragments use instead of selection_set

Beitragsleitfaden

Für dieses Repository ist kein Beitragsleitfaden indexiert

Bewertung

Dieses Issue wurde noch nicht bewertet.

Neue Issues direkt in Ihr Postfach

Eine kurze Übersicht über anfängerfreundliche GitHub-Issues.