Merge pull request #23 from preply/provides

update readme

README.md

@@ -1,7 +1,7 @@
 # graphene-federation
 Federation support for graphene
 
-Build: [![CircleCI](https://circleci.com/gh/preply/graphene-federation.svg?style=svg)](https://circleci.com/gh/erebus1/graphene-federation)
+Build: [![CircleCI](https://circleci.com/gh/preply/graphene-federation.svg?style=svg)](https://circleci.com/gh/preply/graphene-federation)
 
 
 Federation specs implementation on top of Python graphene lib 
@@ -100,7 +100,7 @@ For more details see [examples](examples/)
 
 Or better check [integration_tests](integration_tests/)
 
-Also cool [example](https://github.com/erebus1/graphene-federation/issues/1) of integration with Mongoengine
+Also cool [example](https://github.com/preply/graphene-federation/issues/1) of integration with Mongoengine
 
 
 ### For contribution:

graphene_federation/__init__.py

@@ -1,2 +1,3 @@
-from .main import key, build_schema
+from .main import build_schema
+from .entity import key
 from .extend import extend, external, requires

graphene_federation/entity.py

@@ -50,3 +50,17 @@ def resolve_entities(parent, info, representations):
             return entities
 
     return EntityQuery
+
+
+def key(fields: str):
+    def decorator(Type):
+        register_entity(Type.__name__, Type)
+
+        existing = getattr(Type, "_sdl", "")
+
+        key_sdl = f'@key(fields: "{fields}")'
+        updated = f"{key_sdl} {existing}" if existing else key_sdl
+
+        setattr(Type, '_sdl', updated)
+        return Type
+    return decorator

graphene_federation/main.py

@@ -19,17 +19,3 @@ def _get_query(schema, query_cls=None):
 def build_schema(query=None, mutation=None, **kwargs):
     schema = graphene.Schema(query=query, mutation=mutation, **kwargs)
     return graphene.Schema(query=_get_query(schema, query), mutation=mutation, **kwargs)
-
-
-def key(fields: str):
-    def decorator(Type):
-        register_entity(Type.__name__, Type)
-
-        existing = getattr(Type, "_sdl", "")
-
-        key_sdl = f'@key(fields: "{fields}")'
-        updated = f"{key_sdl} {existing}" if existing else key_sdl
-
-        setattr(Type, '_sdl', updated)
-        return Type
-    return decorator