refactor: Move and reorder files

Author: Checho3388

src/graphql_complexity/__init__.py

@@ -1,12 +1,13 @@
+from graphql_complexity.evaluator.complexity import get_complexity
+
 from .estimators import (
     ComplexityEstimator,
     DirectivesEstimator,
     SimpleEstimator
 )
-from .visitor import ComplexityVisitor
 
 __all__ = [
-    "ComplexityVisitor",
+    "get_complexity",
     "SimpleEstimator",
     "ComplexityEstimator",
     "DirectivesEstimator",

src/graphql_complexity/evaluator/__init__.py

@@ -0,0 +1,5 @@
+from .complexity import get_complexity
+
+__all__ = [
+    'get_complexity',
+]

src/graphql_complexity/evaluator/complexity.py

@@ -0,0 +1,18 @@
+from graphql import parse, visit
+
+from ..estimators import ComplexityEstimator
+from .visitor import ComplexityVisitor
+
+
+def get_complexity(query: str, estimator: ComplexityEstimator) -> int:
+    """Calculate the complexity of a query using the provided estimator."""
+    ast = parse(query)
+    return get_ast_complexity(ast, estimator=estimator)
+
+
+def get_ast_complexity(ast, estimator: ComplexityEstimator) -> int:
+    """Calculate the complexity of a query using the provided estimator."""
+    visitor = ComplexityVisitor(estimator=estimator)
+    visit(ast, visitor)
+
+    return visitor.evaluate()

src/graphql_complexity/visitor.py renamed to src/graphql_complexity/evaluator/visitor.py

@@ -1,13 +1,10 @@
 from typing import Type
 
-from graphql import GraphQLError, visit
+from graphql import GraphQLError
 from strawberry.extensions import SchemaExtension
 
-from graphql_complexity import (
-    ComplexityEstimator,
-    ComplexityVisitor,
-    SimpleEstimator
-)
+from graphql_complexity.estimators import ComplexityEstimator, SimpleEstimator
+from graphql_complexity.evaluator.complexity import get_ast_complexity
 
 
 def build_complexity_extension(
@@ -23,10 +20,10 @@ class ComplexityExtension(SchemaExtension):
         def on_validate(
             self,
         ):
-            self.visitor = ComplexityVisitor(estimator=estimator)
-            visit(self.execution_context.graphql_document, self.visitor)
-
-            self.estimated_complexity = self.visitor.evaluate()
+            self.estimated_complexity = get_ast_complexity(
+                ast=self.execution_context.graphql_document,
+                estimator=estimator
+            )
 
             if max_complexity and self.estimated_complexity > max_complexity:
                 error = GraphQLError(

src/graphql_complexity/extensions/strawberry_graphql.py

@@ -1,17 +1,12 @@
 import pytest
-from graphql import parse, visit
 
-from graphql_complexity.estimators import SimpleEstimator
-from graphql_complexity.visitor import ComplexityVisitor
+from graphql_complexity import SimpleEstimator, get_complexity
+from graphql_complexity.evaluator.visitor import ComplexityVisitor
 
 
 def _evaluate_complexity(query: str, estimator=None):
-    ast = parse(query)
     estimator = estimator or SimpleEstimator(1, 1)
-    visitor = ComplexityVisitor(estimator=estimator)
-    visit(ast, visitor)
-
-    return visitor.evaluate()
+    return get_complexity(query, estimator)
 
 
 def test_one_field_simple_complexity_calculation():

src/graphql_complexity/services.py

@@ -1,20 +1,14 @@
-from graphql import parse, visit
-
-from graphql_complexity.estimators import DirectivesEstimator
-from graphql_complexity.visitor import ComplexityVisitor
+from graphql_complexity import DirectivesEstimator, get_complexity
 
 
 def _evaluate_complexity_with_directives_estimator(
     query: str,
     schema: str,
     **kwargs,
 ):
-    ast = parse(query)
     estimator = DirectivesEstimator(schema, **kwargs)
-    visitor = ComplexityVisitor(estimator=estimator)
-    visit(ast, visitor)
 
-    return visitor.evaluate()
+    return get_complexity(query, estimator)
 
 
 def test_simple_query_with_directive_estimator():
@@ -150,4 +144,4 @@ def test_directive_estimator_should_accept_field_with_directive_is_part_of_an_ob
 
     complexity = _evaluate_complexity_with_directives_estimator(query, schema)
 
-    assert complexity == 4
+    assert complexity == 123

tests/test_complexity_visitor.py

@@ -1,18 +1,16 @@
-from graphql import parse, visit
-
-from graphql_complexity.estimators import DirectivesEstimator, SimpleEstimator
-from graphql_complexity.visitor import ComplexityVisitor
+from graphql_complexity import (
+    DirectivesEstimator,
+    SimpleEstimator,
+    get_complexity
+)
 
 
 def _evaluate_complexity_with_simple_estimator(
     query: str, field_complexity=1, multiplier=1
 ):
-    ast = parse(query)
     estimator = SimpleEstimator(field_complexity, multiplier)
-    visitor = ComplexityVisitor(estimator=estimator)
-    visit(ast, visitor)
 
-    return visitor.evaluate()
+    return get_complexity(query, estimator)
 
 
 def test_root_fields_complexity_is_added():
@@ -34,12 +32,9 @@ def _evaluate_complexity_with_directives_estimator(
     query: str,
     schema: str,
 ):
-    ast = parse(query)
     estimator = DirectivesEstimator(schema)
-    visitor = ComplexityVisitor(estimator=estimator)
-    visit(ast, visitor)
 
-    return visitor.evaluate()
+    return get_complexity(query, estimator)
 
 
 def test_simple_query_with_directive_estimator():

tests/test_directives_estimator.py

@@ -1,7 +1,6 @@
 import pytest
 
-from graphql_complexity.estimators import SimpleEstimator
-from graphql_complexity.services import get_complexity
+from graphql_complexity import SimpleEstimator, get_complexity
 
 
 def _evaluate_complexity_with_simple_estimator(