[v0.2a3] Add methods needed for some relay mutation operations.

Author: jhgg

epoxy/contrib/relay/connections/sorted_collection.py

@@ -99,6 +99,10 @@ def empty_connection(relay, type_name):
             )
         )
 
+    def get_edge(self, relay, type_name, node):
+        Connection, Edge = relay.get_connection_and_edge_types(type_name)
+        return Edge(node=node, cursor=cursor.from_offset(self._key(node)))
+
     def get_connection(self, relay, type_name, args):
         Connection, Edge = relay.get_connection_and_edge_types(type_name)
         before = args.get('before')

epoxy/contrib/relay/data_source/memory.py

@@ -15,6 +15,13 @@ def add(self, obj):
         self.objects_by_type_and_id[obj.T][text_type(obj.id)] = obj
         self.objects_by_type[obj.T].insert(obj)
 
+    def remove(self, obj):
+        del self.objects_by_type_and_id[obj.T][text_type(obj.id)]
+        self.objects_by_type[obj.T].remove(obj)
+
+    def get_edge(self, relay, obj):
+        return self.objects_by_type[obj.T].get_edge(relay, obj.T.name, obj)
+
     def fetch_node(self, object_type, id, resolve_info):
         return self.objects_by_type_and_id[object_type].get(text_type(id))
 

epoxy/contrib/relay/mixin.py

@@ -38,7 +38,7 @@ def register_types(self):
         class Node(R.Interface):
             id = R.ID.NonNull(description='The id of the object.')
 
-            resolve_id = self.resolve_node_id
+            resolve_id = self._resolve_node_id
 
         class PageInfo(R.ObjectType):
             has_next_page = R.Boolean.NonNull(description='When paginating forwards, are there more items?')
@@ -55,9 +55,12 @@ def fetch_node(self, id, info):
         assert isinstance(object_type, GraphQLObjectType)
         return self.data_source.fetch_node(object_type, object_id, info)
 
-    def resolve_node_id(self, obj, args, info):
-        type = self.R.Node().resolve_type(obj, info)
-        return base64('%s:%s' % (type, obj.id))
+    def _resolve_node_id(self, obj, args, info):
+        return self.node_id_for(obj, info)
+
+    def node_id_for(self, obj, info=None):
+        object_type = self.Node.T.resolve_type(obj, info)
+        return base64('%s:%s' % (object_type, obj.id))
 
     def connection_definitions(self, name, object_type):
         R = self.R

epoxy/metaclasses/mutation.py

@@ -40,7 +40,8 @@ def __new__(mcs, name, bases, attrs):
             args={
                 'input': GraphQLArgument(GraphQLNonNull(R[Input]))
             },
-            resolver=functools.partial(mcs._process_resolver, resolver, Input)
+            resolver=functools.partial(mcs._process_resolver, resolver, Input),
+            description=attrs.get('__doc__', None)
         )))
 
     @staticmethod

setup.py

@@ -11,7 +11,7 @@
 
 setup(
     name='graphql-epoxy',
-    version='0.2a1',
+    version='0.2a3',
     description='GraphQL implementation for Python',
     url='https://github.com/graphql-python/graphql-core',
     download_url='https://github.com/graphql-python/graphql-core/releases',