Use __key__ instead for the attribute name for custom keys

Author: john2x

appengine_fixture_loader/loader.py

@@ -48,13 +48,14 @@ def _load(od):
             for attribute_name in [k for k in od.keys()
                                    if not k.startswith('__') and
                                    not k.endswith('__')]:
-                if attribute_name == '_key':
-                    obj.key = ndb.Key(*od[attribute_name])
-                else:
-                    attribute_type = objtype.__dict__[attribute_name]
-                    attribute_value = _sensible_value(attribute_type,
-                                                      od[attribute_name])
-                    obj.__dict__['_values'][attribute_name] = attribute_value
+                attribute_type = objtype.__dict__[attribute_name]
+                attribute_value = _sensible_value(attribute_type,
+                                                  od[attribute_name])
+                obj.__dict__['_values'][attribute_name] = attribute_value
+
+            # set custom key if specified
+            if '__key__' in od.keys():
+                obj.key = ndb.Key(*od['__key__'])
 
             if post_processor:
                 post_processor(obj)

tests/customers.json

@@ -1,10 +1,10 @@
 [
     {
         "name": "John",
-        "_key": ["Customer", "john"]
+        "__key__": ["Customer", "john"]
     },
     {
         "name": "Jane",
-        "_key": ["Customer", "jane"]
+        "__key__": ["Customer", "jane"]
     }
 ]

tests/purchases.json

@@ -1,10 +1,10 @@
 [
     {
         "price": 100,
-        "_key": ["Customer", "john", "Purchase", "100"]
+        "__key__": ["Customer", "john", "Purchase", "100"]
     },
     {
         "price": 50,
-        "_key": ["Customer", "jane", "Purchase", "50"]
+        "__key__": ["Customer", "jane", "Purchase", "50"]
     }
 ]