black.parsing.InvalidInput: Cannot parse: 1041:8: def return(cls, id: str) -> ReturnFields:

[Anton-Shutik]

Fails when trying to generate Sync client for Shopify GraphQL, but works fine with enable_custom_operations=false

env:

Python 3.9.13
ariadne-codegen==0.14.0
isort==5.13.2
black==24.8.0

[tool.ariadne-codegen]
remote_schema_url = "https://{}.myshopify.com/admin/api/2024-07/graphql.json"
remote_schema_headers = {"X-Shopify-Access-Token" = "$SHOPIFY_API_TOKEN"}
include_all_enums = true
base_client_file_path  = "/usr/local/lib/python3.9/site-packages/ariadne_codegen/client_generators/dependencies/base_client.py"
plugins = []
**enable_custom_operations=true**
convert_to_snake_case = true
opentelemetry_client = false
client_name = "ShopifyGQLClient"
include_comments = "stable"
base_client_name = "BaseClient"
target_package_name = "gql"
client_file_name = "client"
enums_module_name = "enums"
async_client = false

Traceback (most recent call last):
  File "/usr/local/bin/ariadne-codegen", line 8, in <module>
    sys.exit(main())
  File "/usr/local/lib/python3.9/site-packages/click/core.py", line 1157, in __call__
    return self.main(*args, **kwargs)
  File "/usr/local/lib/python3.9/site-packages/click/core.py", line 1078, in main
    rv = self.invoke(ctx)
  File "/usr/local/lib/python3.9/site-packages/click/core.py", line 1434, in invoke
    return ctx.invoke(self.callback, **ctx.params)
  File "/usr/local/lib/python3.9/site-packages/click/core.py", line 783, in invoke
    return __callback(*args, **kwargs)
  File "/usr/local/lib/python3.9/site-packages/ariadne_codegen/main.py", line 37, in main
    client(config_dict)
  File "/usr/local/lib/python3.9/site-packages/ariadne_codegen/main.py", line 81, in client
    generated_files = package_generator.generate()
  File "/usr/local/lib/python3.9/site-packages/ariadne_codegen/client_generators/package.py", line 161, in generate
    self._generate_custom_queries()
  File "/usr/local/lib/python3.9/site-packages/ariadne_codegen/client_generators/package.py", line 384, in _generate_custom_queries
    code = self._add_comments_to_code(ast_to_str(module, False))
  File "/usr/local/lib/python3.9/site-packages/ariadne_codegen/utils.py", line 33, in ast_to_str
    return format_str(isort.code(code), mode=Mode())
  File "/usr/local/lib/python3.9/site-packages/black/__init__.py", line 1067, in format_str
    dst_contents = _format_str_once(src_contents, mode=mode)
  File "/usr/local/lib/python3.9/site-packages/black/__init__.py", line 1077, in _format_str_once
    src_node = lib2to3_parse(src_contents.lstrip(), mode.target_versions)
  File "/usr/local/lib/python3.9/site-packages/black/parsing.py", line 126, in lib2to3_parse
    raise exc from None
black.parsing.InvalidInput: Cannot parse: 1041:8:     def return(cls, id: str) -> ReturnFields:

[Anton-Shutik]

Seems like here is the problem, they have query named return