fixing arguments inconsistency

Author: jhnnsrs

.vscode/settings.json

@@ -36,5 +36,9 @@
   "python.testing.pytestEnabled": true,
   "python.linting.pylintEnabled": true,
   "python.linting.pylintArgs": ["--enable=W0614"],
-  "python.linting.enabled": false
+  "python.linting.enabled": false,
+  "[python]": {
+    "editor.defaultFormatter": "ms-python.black-formatter"
+  },
+  "python.formatting.provider": "none"
 }

gg/schema.py

@@ -0,0 +1,101 @@
+from pydantic import BaseModel, Field
+from typing import Optional, Literal, List
+from api.proxies import execute
+from gql.client import AsyncClientSession
+from enum import Enum
+
+
+class Beast(BaseModel):
+    typename: Optional[Literal["Beast"]] = Field(alias="__typename")
+    common_name: Optional[str] = Field(alias="commonName")
+    "a beast's name to you and I"
+    tax_class: Optional[str] = Field(alias="taxClass")
+    "taxonomy grouping"
+
+
+class CreateBeastCreatebeast(BaseModel):
+    typename: Optional[Literal["Beast"]] = Field(alias="__typename")
+    binomial: Optional[str]
+    "a beast's name in Latin"
+    common_name: Optional[str] = Field(alias="commonName")
+    "a beast's name to you and I"
+
+
+class CreateBeast(BaseModel):
+    create_beast: CreateBeastCreatebeast = Field(alias="createBeast")
+    "create a massive beast on the server"
+
+    class Arguments(BaseModel):
+        id: str
+        legs: int
+        binomial: str
+        common_name: str = Field(alias="commonName")
+        tax_class: str = Field(alias="taxClass")
+        eats: Optional[List[Optional[str]]] = Field(default=None)
+
+    class Meta:
+        document = "mutation createBeast($id: ID!, $legs: Int!, $binomial: String!, $commonName: String!, $taxClass: String!, $eats: [ID]) {\n  createBeast(\n    id: $id\n    legs: $legs\n    binomial: $binomial\n    commonName: $commonName\n    taxClass: $taxClass\n    eats: $eats\n  ) {\n    binomial\n    commonName\n  }\n}"
+
+
+class Get_beasts(BaseModel):
+    beasts: Optional[List[Optional[Beast]]]
+    "get all the beasts on the server"
+
+    class Arguments(BaseModel):
+        pass
+
+    class Meta:
+        document = "fragment Beast on Beast {\n  commonName\n  taxClass\n}\n\nquery get_beasts {\n  beasts {\n    ...Beast\n  }\n}"
+
+
+async def acreate_beast(
+    client: AsyncClientSession,
+    id: str,
+    legs: int,
+    binomial: str,
+    common_name: str,
+    tax_class: str,
+    eats: Optional[List[Optional[str]]] = None,
+) -> CreateBeastCreatebeast:
+    """createBeast
+
+
+
+    Arguments:
+        client (gql.client.AsyncClientSession): Specify that in turms.plugin.funcs.OperationsFuncPlugin
+        id (str): id
+        legs (int): legs
+        binomial (str): binomial
+        common_name (str): commonName
+        tax_class (str): taxClass
+        eats (Optional[List[Optional[str]]], optional): eats.
+
+    Returns:
+        CreateBeastCreatebeast"""
+    return (
+        await execute(
+            client,
+            CreateBeast,
+            {
+                "id": id,
+                "legs": legs,
+                "binomial": binomial,
+                "commonName": common_name,
+                "taxClass": tax_class,
+                "eats": eats,
+            },
+        )
+    ).create_beast
+
+
+async def aget_beasts(client: AsyncClientSession) -> Optional[List[Optional[Beast]]]:
+    """get_beasts
+
+
+
+    Arguments:
+        client (gql.client.AsyncClientSession): Specify that in turms.plugin.funcs.OperationsFuncPlugin
+
+    Returns:
+        Optional[List[Optional[Beast]]]"""
+    return (await execute(client, Get_beasts, {})).beasts

graphql.config.yaml

@@ -1,4 +1,3 @@
-
 projects:
   default:
     schema: https://countries.trevorblades.com/
@@ -16,3 +15,33 @@ projects:
           - type: turms.processors.black.BlackProcessor
         scalar_definitions:
           uuid: str
+  beast:
+    schema: tests/schemas/beasts.graphql
+    documents: tests/documents/beasts/**.graphql
+    extensions:
+      turms:
+        out_dir: gg
+        plugins:
+          - type: turms.plugins.enums.EnumsPlugin
+          - type: turms.plugins.inputs.InputsPlugin
+          - type: turms.plugins.fragments.FragmentsPlugin
+          - type: turms.plugins.operations.OperationsPlugin
+          - type: turms.plugins.funcs.FuncsPlugin
+          - type: turms.plugins.funcs.FuncsPlugin
+            global_args:
+              - type: gql.client.AsyncClientSession
+                key: client
+            definitions:
+              - type: query
+                use: api.proxies.execute
+                is_async: True
+              - type: mutation
+                use: api.proxies.execute
+                is_async: True
+        stylers:
+          - type: turms.stylers.capitalize.CapitalizeStyler
+          - type: turms.stylers.snake_case.SnakeCaseStyler
+        processors:
+          - type: turms.processors.black.BlackProcessor
+        scalar_definitions:
+          uuid: str

tests/test_funcs.py

@@ -235,3 +235,92 @@ def test_extra_args(beast_schema):
         generated_ast,
         "",
     )
+
+
+
+def test_extra_args(beast_schema):
+    config = GeneratorConfig(
+        documents=build_relative_glob("/documents/beasts/**/*.graphql"),
+    )
+
+    extra_args = [
+        Arg(
+            key="client",
+            type="mocks.ExtraArg",
+            description="An Extra Arg",
+        )
+    ]
+
+    extra_kwargs = [
+        Kwarg(
+            key="extra",
+            type="int",
+            description="An Extra Arg",
+            default=5,
+        )
+    ]
+
+    generated_ast = generate_ast(
+        config,
+        beast_schema,
+        stylers=[DefaultStyler()],
+        plugins=[
+            EnumsPlugin(),
+            InputsPlugin(),
+            FragmentsPlugin(),
+            OperationsPlugin(),
+            FuncsPlugin(
+                config=FuncsPluginConfig(
+                    definitions=[
+                        FunctionDefinition(
+                            type="query",
+                            use="mocks.query",
+                            is_async=False,
+                            extra_args=extra_args,
+                            extra_kwargs=extra_kwargs,
+                        ),
+                        FunctionDefinition(
+                            type="query",
+                            use="mocks.query",
+                            is_async=True,
+                            extra_args=extra_args,
+                            extra_kwargs=extra_kwargs,
+                        ),
+                        FunctionDefinition(
+                            type="mutation",
+                            use="mocks.aquery",
+                            is_async=False,
+                            extra_args=extra_args,
+                            extra_kwargs=extra_kwargs,
+                        ),
+                        FunctionDefinition(
+                            type="mutation",
+                            use="mocks.aquery",
+                            is_async=True,
+                            extra_args=extra_args,
+                            extra_kwargs=extra_kwargs,
+                        ),
+                        FunctionDefinition(
+                            type="subscription",
+                            use="mocks.subscribe",
+                            is_async=False,
+                            extra_args=extra_args,
+                            extra_kwargs=extra_kwargs,
+                        ),
+                        FunctionDefinition(
+                            type="subscription",
+                            use="mocks.asubscribe",
+                            is_async=True,
+                            extra_args=extra_args,
+                            extra_kwargs=extra_kwargs,
+                        ),
+                    ]
+                ),
+            ),
+        ],
+    )
+
+    unit_test_with(
+        generated_ast,
+        "",
+    )