vector type params

nakagami · Aug 23, 2024 · da2b948 · da2b948
1 parent 18d5cb1
commit da2b948
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/cymysql/converters.py b/cymysql/converters.py
@@ -92,6 +92,11 @@ def escape_decimal(obj):
     return str(obj)
 
 
+def escape_vector(obj):
+    import numpy as np
+    return escape_bytes(obj.astype(np.float32).tobytes())
+
+
 def convert_datetime(obj):
     """Returns a DATETIME or TIMESTAMP column value as a datetime object:
 
@@ -355,6 +360,12 @@ def convert_decimal(obj):
     encoders[unicode] = escape_string
     encoders[long] = escape_long
 
+try:
+    import numpy as np
+    encoders[np.ndarray] = escape_vector
+except ImportError:
+    pass
+
 
 def escape_item(val, charset, encoders=encoders):
     if type(val) in [tuple, list, set]: