Add I/O support for jsonb type

elprans · elprans · commit 2f2df7a3abf0 · 2016-06-09T17:15:42.000-04:00
diff --git a/postgresql/test/test_driver.py b/postgresql/test/test_driver.py
@@ -284,6 +284,9 @@
 			['00:00:00:00:00:01', '00:00:00:00:00:00', 'ff:ff:ff:ff:ff:ff', '10:00:00:00:00:00'],
 		],
 	),
+	('jsonb', [
+		'{"foo": "bar", "spam": ["ham"]}'
+	])
 ]
 
 try:
diff --git a/postgresql/types/__init__.py b/postgresql/types/__init__.py
@@ -31,6 +31,7 @@
 REGDICTIONARYOID = 3769
 
 JSONOID = 114
+JSONBOID = 3802
 XMLOID = 142
 
 MACADDROID = 829
diff --git a/postgresql/types/io/__init__.py b/postgresql/types/io/__init__.py
@@ -79,6 +79,10 @@
 		pg_types.XMLOID,
 	),
 
+	'stdlib_jsonb' : (
+		pg_types.JSONBOID,
+	),
+
 	# Must be db.typio.identify(contrib_hstore = 'hstore')'d
 	'contrib_hstore' : (
 		'contrib_hstore',
diff --git a/postgresql/types/io/stdlib_jsonb.py b/postgresql/types/io/stdlib_jsonb.py
@@ -0,0 +1,24 @@
+from ...types import JSONBOID
+
+
+def jsonb_pack(x, typeio):
+	jsonb = typeio.encode(x)
+	return b'\x01' + jsonb
+
+
+def jsonb_unpack(x, typeio):
+	if x[0] != 1:
+		raise ValueError('unexpected JSONB format version: {!r}'.format(x[0]))
+	return typeio.decode(x[1:])
+
+
+def _jsonb_io_factory(oid, typeio):
+	_pack = lambda x: jsonb_pack(x, typeio)
+	_unpack = lambda x: jsonb_unpack(x, typeio)
+
+	return (_pack, _unpack, str)
+
+
+oid_to_io = {
+	JSONBOID: _jsonb_io_factory
+}
