feat: support make schema from multi files

Author: syfun

gql/__init__.py

@@ -11,8 +11,8 @@
     type_resolver,
 )
 from .scalar import scalar_type  # noqa
-from .schema import make_schema, make_schema_from_file  # noqa
+from .schema import make_schema, make_schema_from_file, make_schema_from_files  # noqa
 from .schema_visitor import SchemaDirectiveVisitor  # noqa
 from .utils import gql  # noqa
 
-__version__ = '0.2.2'
+__version__ = '0.2.3'

gql/schema.py

@@ -74,6 +74,42 @@ def make_schema_from_file(
         return schema
 
 
+def parse_from_file(file: str):
+    with open(file, 'r') as f:
+        return parse(f.read())
+
+
+base_type_def = """
+type Query
+type Mutation
+"""
+
+
+def make_schema_from_files(
+    files: List[str],
+    assume_valid: bool = False,
+    assume_valid_sdl: bool = False,
+    no_location: bool = False,
+    experimental_fragment_variables: bool = False,
+    federation: bool = False,
+    directives: Dict[str, Type[SchemaDirectiveVisitor]] = None,
+):
+    if not files:
+        return None
+    schema = make_schema(
+        base_type_def,
+        assume_valid=assume_valid,
+        assume_valid_sdl=assume_valid_sdl,
+        no_location=no_location,
+        experimental_fragment_variables=experimental_fragment_variables,
+        federation=federation,
+        directives=directives,
+    )
+    for file in files:
+        schema = extend_schema(schema, parse_from_file(file))
+    return schema
+
+
 def make_federation_schema(
     type_defs: str,
     assume_valid: bool = False,

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "python-gql"
-version = "0.2.2"
+version = "0.2.3"
 description = "Python schema-first GraphQL library based on GraphQL-core"
 authors = ["ysun <sunyu418@gmail.com>"]
 license = "MIT"

tests/hero.gql tests/hero.graphqltests/hero.gql renamed to tests/hero.graphql

@@ -0,0 +1,7 @@
+type User {
+    id: ID!
+    name: String!
+}
+extend type Query {
+    me: User!
+}

tests/schema1.graphql

@@ -0,0 +1,12 @@
+type Address {
+    id: ID!
+    name: String!
+}
+
+extend type Query {
+    addresses: [Address!]!
+}
+
+extend type Mutation {
+    createAddress(name: String!): Address!
+}

tests/schema2.graphql

@@ -0,0 +1,16 @@
+from gql.schema import make_schema_from_files
+
+from pathlib import Path
+
+
+def test_make_schema_from_files():
+    assert make_schema_from_files([]) is None
+
+    current_dir = Path(__file__).parent
+    files = [
+        str(current_dir / 'schema1.graphql'),
+        str(current_dir / 'schema2.graphql')
+    ]
+    schema = make_schema_from_files(files)
+    assert set(schema.query_type.fields.keys()) == {'me', 'addresses'}
+    assert set(schema.mutation_type.fields.keys()) == {'createAddress'}