Merge pull request #1202 from Mytherin/pythonblobs

Add blob support to the python api

tools/pythonpkg/duckdb_python.cpp

@@ -56,6 +56,12 @@ struct StringConvert {
 	}
 };
 
+struct BlobConvert {
+	template <class DUCKDB_T, class NUMPY_T> static py::str convert_value(string_t val) {
+		return py::bytes(val.GetString());
+	}
+};
+
 struct IntegralConvert {
 	template <class DUCKDB_T, class NUMPY_T> static NUMPY_T convert_value(DUCKDB_T val) {
 		return NUMPY_T(val);
@@ -544,7 +550,9 @@ struct DuckDBPyResult {
 			case LogicalTypeId::VARCHAR:
 				res[col_idx] = val.GetValue<string>();
 				break;
-
+			case LogicalTypeId::BLOB:
+				res[col_idx] = py::bytes(val.GetValue<string>());
+				break;
 			case LogicalTypeId::TIMESTAMP: {
 				D_ASSERT(result->types[col_idx].InternalType() == PhysicalType::INT64);
 
@@ -660,6 +668,10 @@ struct DuckDBPyResult {
 				col_res = duckdb_py_convert::fetch_column<string_t, py::str, duckdb_py_convert::StringConvert>(
 				    "object", mres->collection, col_idx);
 				break;
+			case LogicalTypeId::BLOB:
+				col_res = duckdb_py_convert::fetch_column<string_t, py::bytes, duckdb_py_convert::BlobConvert>(
+				    "object", mres->collection, col_idx);
+				break;
 			default:
 				throw runtime_error("unsupported type " + mres->types[col_idx].ToString());
 			}

tools/pythonpkg/tests/test_blob.py

@@ -0,0 +1,12 @@
+import duckdb
+import numpy
+
+class TestBlob(object):
+    def test_blob(self, duckdb_cursor):
+        duckdb_cursor.execute("SELECT BLOB 'hello'")
+        results = duckdb_cursor.fetchall()
+        assert results[0][0] == b'hello'
+
+        duckdb_cursor.execute("SELECT BLOB 'hello' AS a")
+        results = duckdb_cursor.fetchnumpy()
+        assert results['a'] == numpy.array([b'hello'], dtype=numpy.object)