Added automatic conversion of UUIDs to strings.

SStorm · SStorm · commit fac7355f6919 · 2023-07-28T10:57:14.000+01:00
diff --git a/.gitignore b/.gitignore
@@ -19,3 +19,4 @@ htmlcov/
 out/
 parts/
 tmp/
+env/
diff --git a/CHANGES.txt b/CHANGES.txt
@@ -5,6 +5,7 @@ Changes for crate
 Unreleased
 ==========
 
+- Properly handle Python-native UUID types in SQL parameters
 
 2023/07/17 0.33.0
 =================
diff --git a/src/crate/client/http.py b/src/crate/client/http.py
@@ -35,6 +35,8 @@
 from time import time
 from datetime import datetime, date, timezone
 from decimal import Decimal
+from uuid import UUID
+
 from urllib3 import connection_from_url
 from urllib3.connection import HTTPConnection
 from urllib3.exceptions import (
@@ -86,7 +88,7 @@ class CrateJsonEncoder(json.JSONEncoder):
     epoch_naive = datetime(1970, 1, 1)
 
     def default(self, o):
-        if isinstance(o, Decimal):
+        if isinstance(o, (Decimal, UUID)):
             return str(o)
         if isinstance(o, datetime):
             if o.tzinfo is not None:
diff --git a/src/crate/client/test_http.py b/src/crate/client/test_http.py
@@ -35,9 +35,12 @@
 from threading import Thread, Event
 from decimal import Decimal
 import datetime as dt
+
 import urllib3.exceptions
 from base64 import b64decode
 from urllib.parse import urlparse, parse_qs
+
+import uuid
 from setuptools.ssl_support import find_ca_bundle
 
 from .http import Client, CrateJsonEncoder, _get_socket_opts, _remove_certs_for_non_https
@@ -287,6 +290,18 @@ def test_socket_options_contain_keepalive(self):
         )
         client.close()
 
+    @patch(REQUEST, autospec=True)
+    def test_uuid_serialization(self, request):
+        client = Client(servers="localhost:4200")
+        request.return_value = fake_response(200)
+
+        uid = uuid.uuid4()
+        client.sql('insert into my_table (str_col) values (?)', (uid,))
+
+        data = json.loads(request.call_args[1]['data'])
+        self.assertEqual(data['args'], [str(uid)])
+        client.close()
+
 
 @patch(REQUEST, fail_sometimes)
 class ThreadSafeHttpClientTest(TestCase):

-Original file line number
+Diff line change
 out/
 parts/
 tmp/
 +env/