Merge pull request #15 from graphql-python/feature/id_to_key

NdbNode.global_id_to_key utility method

Author: ekampf

HISTORY.rst

@@ -3,6 +3,10 @@
 History
 -------
 
+0.1.6 (TBD)
+---------------------
+* Added NdbNode.global_id_to_key [PR #15](https://github.com/graphql-python/graphene-gae/pull/15)
+
 0.1.5 (2016-06-08)
 ---------------------
 * Fixed behavior of ndb.KeyProperty ([PR #14](https://github.com/graphql-python/graphene-gae/pull/14))

graphene_gae/__init__.py

@@ -12,7 +12,7 @@
 )
 
 __author__ = 'Eran Kampf'
-__version__ = '0.1.5'
+__version__ = '0.1.6'
 
 __all__ = [
     NdbObjectType,

graphene_gae/ndb/types.py

@@ -5,6 +5,7 @@
 from graphene.core.classtypes.objecttype import ObjectType, ObjectTypeMeta
 
 from google.appengine.ext import ndb
+from graphql_relay import from_global_id
 
 from .options import NdbOptions
 
@@ -96,6 +97,11 @@ def to_global_id(self):
         entity_id = self.key.urlsafe() if self.key else None
         return self.global_id(entity_id)
 
+    @classmethod
+    def global_id_to_key(cls, global_id):
+        _, urlsafe_key = from_global_id(global_id)
+        return ndb.Key(urlsafe=urlsafe_key)
+
     @classmethod
     def get_node(cls, urlsafe_key, info=None):
         try:

tests/_ndb/test_types.py

@@ -325,4 +325,3 @@ def testQuery_repeatedKeyProperty(self):
         self.assertLength(article['tags'], 4)
         for i in range(0, 3):
             self.assertEqual(article['tags'][i]['name'], 't%s' % (i + 1))
-

tests/_ndb/test_types_relay.py

@@ -83,6 +83,16 @@ def testNdbNode_getNode_validID_entityDoes_shouldReturnEntity(self):
         self.assertIsNotNone(result)
         self.assertEqual(result.instance, article_key.get())
 
+    def testNdbNode_globalIdToKey_returnsNdbKey(self):
+        article_key = Article(
+            headline="TestGetNode",
+            summary="1",
+            author_key=Author(name="John Dow", email="[email protected]").put(),
+        ).put()
+
+        node = ArticleType.get_node(article_key.urlsafe())
+        self.assertEqual(NdbNode.global_id_to_key(node.to_global_id()), article_key)
+
     def test_keyProperty(self):
         Article(
             headline="Test1",