Add aliases to register_(hstore|json) methods for use with NamedTuples

FSX · Aug 3, 2015 · 87a4a7b · 87a4a7b
1 parent 99f69b9
commit 87a4a7b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/momoko/connection.py b/momoko/connection.py
@@ -887,7 +887,7 @@ def register_hstore(self, globally=False, unicode=False):
         registrator = partial(_psy_register_hstore, None, globally, unicode)
         callback = partial(self._register, future, registrator)
         self.ioloop.add_future(self.execute(
-            "SELECT 'hstore'::regtype::oid, 'hstore[]'::regtype::oid",
+            "SELECT 'hstore'::regtype::oid AS hstore_oid, 'hstore[]'::regtype::oid AS hstore_arr_oid",
         ), callback)
 
         return future
@@ -915,7 +915,7 @@ def register_json(self, globally=False, loads=None):
         registrator = partial(_psy_register_json, None, globally, loads)
         callback = partial(self._register, future, registrator)
         self.ioloop.add_future(self.execute(
-            "SELECT 'json'::regtype::oid, 'json[]'::regtype::oid"
+            "SELECT 'json'::regtype::oid AS json_oid, 'json[]'::regtype::oid AS json_arr_oid"
         ), callback)
 
         return future