fix graphql

Author: clavedeluna

integration_tests/sonar/test_sonar_disable_graphql_introspection.py

@@ -11,7 +11,7 @@ class TestSonarDisableGraphQLIntrospection(SonarIntegrationTest):
     codemod = SonarDisableGraphQLIntrospection
     code_path = "tests/samples/disable_graphql_introspection.py"
     expected_new_code = """\
-    from graphql_server.flask import GraphQLView
+    from graphql_server.flask.views import GraphQLView
     from flask import Flask
     from graphql import (
         GraphQLSchema, GraphQLObjectType, GraphQLField, GraphQLString)

integration_tests/test_disable_graphql_introspection.py

@@ -8,7 +8,7 @@
 class TestDisableGraphQLIntrospection(BaseIntegrationTest):
     codemod = DisableGraphQLIntrospection
     original_code = """
-        from graphql_server.flask import GraphQLView
+        from graphql_server.flask.views import GraphQLView
         from flask import Flask
         from graphql import (
             GraphQLSchema, GraphQLObjectType, GraphQLField, GraphQLString)
@@ -32,7 +32,7 @@ class TestDisableGraphQLIntrospection(BaseIntegrationTest):
         )
     """
     expected_new_code = """
-        from graphql_server.flask import GraphQLView
+        from graphql_server.flask.views import GraphQLView
         from flask import Flask
         from graphql import (
             GraphQLSchema, GraphQLObjectType, GraphQLField, GraphQLString)

pyproject.toml

@@ -84,7 +84,7 @@ test = [
     "numpy ~= 2.3.0; python_version > '3.10'",
     "flask_wtf~=1.2.0",
     "fickling~=0.1.0,>=0.1.3",
-    "graphql-server~=3.0.0b7",
+    "graphql-server~=3.0.0b9",
     "unidiff>=0.7.5",
     "semgrep>=1.134,<1.135",
 ]

src/core_codemods/disable_graphql_introspection.py

@@ -41,6 +41,8 @@ class FindGraphQLViewsWithIntrospection(
     supported_functions = {
         "graphql_server.flask.GraphQLView",
         "graphql_server.flask.GraphQLView.as_view",
+        "graphql_server.flask.views.GraphQLView",
+        "graphql_server.flask.views.GraphQLView.as_view",
         "graphql_server.sanic.GraphQLView",
         "graphql_server.aiohttp.GraphQLView",
         "graphql_server.webob.GraphQLView",

src/core_codemods/docs/pixee_python_disable-graphql-introspection.md

@@ -4,7 +4,7 @@ Introspection is often enabled by default in GraphQL without authentication. Thi
 
 Our changes look something like this:
 ```diff
-from graphql_server.flask import GraphQLView
+from graphql_server.flask.views import GraphQLView
 from flask import Flask
 from graphql import (
     GraphQLSchema, GraphQLObjectType, GraphQLField, GraphQLString)

tests/codemods/sonar/test_sonar_disable_graphql_introspection.py

@@ -15,7 +15,7 @@ def test_name(self):
 
     def test_simple(self, tmpdir):
         input_code = """
-        from graphql_server.flask import GraphQLView
+        from graphql_server.flask.views import GraphQLView
         from flask import Flask
         from .schemas import schema
 
@@ -29,7 +29,7 @@ def test_simple(self, tmpdir):
         )
         """
         expected = """
-        from graphql_server.flask import GraphQLView
+        from graphql_server.flask.views import GraphQLView
         from flask import Flask
         from .schemas import schema
         from graphql.validation import NoSchemaIntrospectionCustomRule

tests/codemods/test_disable_graphql_introspection.py

@@ -12,7 +12,7 @@ def test_name(self):
 
     def test_simple_flask(self, tmpdir):
         input_code = """
-        from graphql_server.flask import GraphQLView
+        from graphql_server.flask.views import GraphQLView
         from flask import Flask
         from .schemas import schema
 
@@ -26,7 +26,7 @@ def test_simple_flask(self, tmpdir):
         )
         """
         expected = """
-        from graphql_server.flask import GraphQLView
+        from graphql_server.flask.views import GraphQLView
         from flask import Flask
         from .schemas import schema
         from graphql.validation import NoSchemaIntrospectionCustomRule
@@ -77,7 +77,7 @@ def test_simple_constructor(self, tmpdir, module):
 
     def test_add_indirect(self, tmpdir):
         input_code = """
-        from graphql_server.flask import GraphQLView
+        from graphql_server.flask.views import GraphQLView
         from flask import Flask
         from .schemas import schema
 
@@ -89,7 +89,7 @@ def test_add_indirect(self, tmpdir):
         )
         """
         expected = """
-        from graphql_server.flask import GraphQLView
+        from graphql_server.flask.views import GraphQLView
         from flask import Flask
         from .schemas import schema
         from graphql.validation import NoSchemaIntrospectionCustomRule
@@ -105,7 +105,7 @@ def test_add_indirect(self, tmpdir):
 
     def test_add_list_double_indirect(self, tmpdir):
         input_code = """
-        from graphql_server.flask import GraphQLView
+        from graphql_server.flask.views import GraphQLView
         from flask import Flask
         from .schemas import schema
 
@@ -117,7 +117,7 @@ def test_add_list_double_indirect(self, tmpdir):
         )
         """
         expected = """
-        from graphql_server.flask import GraphQLView
+        from graphql_server.flask.views import GraphQLView
         from flask import Flask
         from .schemas import schema
         from graphql.validation import NoSchemaIntrospectionCustomRule
@@ -133,7 +133,7 @@ def test_add_list_double_indirect(self, tmpdir):
 
     def test_add_dict_indirect(self, tmpdir):
         input_code = """
-        from graphql_server.flask import GraphQLView
+        from graphql_server.flask.views import GraphQLView
         from flask import Flask
         from .schemas import schema
 
@@ -146,7 +146,7 @@ def test_add_dict_indirect(self, tmpdir):
         )
         """
         expected = """
-        from graphql_server.flask import GraphQLView
+        from graphql_server.flask.views import GraphQLView
         from flask import Flask
         from .schemas import schema
         from graphql.validation import NoSchemaIntrospectionCustomRule
@@ -163,7 +163,7 @@ def test_add_dict_indirect(self, tmpdir):
 
     def test_has_validation_rule(self, tmpdir):
         input_code = """
-        from graphql_server.flask import GraphQLView
+        from graphql_server.flask.views import GraphQLView
         from flask import Flask
         from .schemas import schema
         from graphql.validation import NoSchemaIntrospectionCustomRule
@@ -178,7 +178,7 @@ def test_has_validation_rule(self, tmpdir):
 
     def test_has_graphene_validation_rule(self, tmpdir):
         input_code = """
-        from graphql_server.flask import GraphQLView
+        from graphql_server.flask.views import GraphQLView
         from flask import Flask
         from .schemas import schema
         from graphene.validation import DisableIntrospection

tests/samples/disable_graphql_introspection.py

@@ -1,4 +1,4 @@
-from graphql_server.flask import GraphQLView
+from graphql_server.flask.views import GraphQLView
 from flask import Flask
 from graphql import (
     GraphQLSchema, GraphQLObjectType, GraphQLField, GraphQLString)