quillsql 2.2.9__py3-none-any.whl → 2.2.10__py3-none-any.whl

quillsql/core.py

@@ -1,5 +1,8 @@
 import os
 import codecs
+import uuid
+from datetime import date, datetime
+from decimal import Decimal
 from dotenv import load_dotenv
 
 import requests
@@ -600,6 +603,17 @@ class Quill:
 
         return {"error": f"Unknown tool: {tool_name}"}
 
+    def _json_default(self, obj):
+        if isinstance(obj, Enum):
+            return obj.value
+        if isinstance(obj, uuid.UUID):
+            return str(obj)
+        if isinstance(obj, (datetime, date)):
+            return obj.isoformat()
+        if isinstance(obj, Decimal):
+            return float(obj)
+        raise TypeError(f"Object of type {type(obj).__name__} is not JSON serializable")
+
     def _stream_sse(self, endpoint, tenants, payload, flags, filters, admin_enabled):
         tenant_flags = self._normalize_tenant_flags(tenants, flags)
         request_payload = {
@@ -612,20 +626,13 @@ class Quill:
         if admin_enabled is not None:
             request_payload["adminEnabled"] = admin_enabled
 
-        # Custom JSON Encoder to handle Enums
-        class EnumEncoder(json.JSONEncoder):
-            def default(self, obj):
-                if isinstance(obj, Enum):
-                    return obj.value  # Convert enum to its value (string in this case)
-                return super().default(obj)
-
         url = f"{self.baseUrl}/sdk/{endpoint}"
         headers = {
             "Authorization": f"Bearer {self.private_key}",
             "Content-Type": "application/json",
             "Accept": "text/event-stream",
         }
-        encoded = json.dumps(request_payload, cls=EnumEncoder)
+        encoded = json.dumps(request_payload, default=self._json_default)
 
         resp = requests.post(url, data=encoded, headers=headers, stream=True)
         decoder = codecs.getincrementaldecoder("utf-8")()
@@ -828,16 +835,9 @@ class Quill:
         return results
     
     def post_quill(self, path, payload):
-        # Custom JSON Encoder to handle Enums
-        class EnumEncoder(json.JSONEncoder):
-            def default(self, obj):
-                if isinstance(obj, Enum):
-                    return obj.value  # Convert enum to its value (string in this case)
-                return super().default(obj)
-        
         url = f"{self.baseUrl}/sdk/{path}"
         # Set content type to application/json
         headers = {"Authorization": f"Bearer {self.private_key}", "Content-Type": "application/json"}
-        encoded = json.dumps(payload, cls=EnumEncoder)
+        encoded = json.dumps(payload, default=self._json_default)
         response = requests.post(url, data=encoded, headers=headers)
         return response.json()

{quillsql-2.2.9.dist-info → quillsql-2.2.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: quillsql
-Version: 2.2.9
+Version: 2.2.10
 Summary: Quill SDK for Python.
 Home-page: https://github.com/quill-sql/quill-python
 Author: Quill

{quillsql-2.2.9.dist-info → quillsql-2.2.10.dist-info}/RECORD

@@ -1,5 +1,5 @@
 quillsql/__init__.py,sha256=wjJfszle5vheUbgUfJMHQqtqhx2W3UaDN4ndcRIfmkQ,236
-quillsql/core.py,sha256=VzWtu0up8tcSxijHt_x3bSbZnPjbZFIyGKXCwerpPrg,33704
+quillsql/core.py,sha256=cxuAfOgQ3AMt26hQgNTFejGv7L_fEAVXtV7dsNdiT7k,33592
 quillsql/error.py,sha256=n9VKHw4FAgg7ZEAz2YQ8L_8FdRG_1shwGngf2iWhUSM,175
 quillsql/assets/__init__.py,sha256=oXQ2ZS5XDXkXTYjADxNfGt55cIn_rqfgWL2EDqjTyoI,45
 quillsql/assets/pgtypes.py,sha256=-B_2wUaoAsdX7_HnJhUlx4ptZQ6x-cXwuST9ACgGFdE,33820
@@ -15,7 +15,7 @@ quillsql/utils/post_quill_executor.py,sha256=DB1RHNfqHPYarMM10vSv--UjpCZqe4qYTjq
 quillsql/utils/run_query_processes.py,sha256=QwnMr5UwXdtO_W88lv5nBaf6pJ_h5oWQnYd8K9oHQ5s,1030
 quillsql/utils/schema_conversion.py,sha256=TFfMibN9nOsxNRhHw5YIFl3jGTvipG81bxX4LFDulUY,314
 quillsql/utils/tenants.py,sha256=ZD2FuKz0gjBVSsThHDv1P8PU6EL8E009NWihE5hAH-Q,2022
-quillsql-2.2.9.dist-info/METADATA,sha256=zVlzKEUZAAIQR30Re6w6tdebu6hFjtTQnOeIXlOwZz4,3052
-quillsql-2.2.9.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-quillsql-2.2.9.dist-info/top_level.txt,sha256=eU2vHnVqwpYQJ3ADl1Q-DIBzbYejZRUhcMdN_4zMCz8,9
-quillsql-2.2.9.dist-info/RECORD,,
+quillsql-2.2.10.dist-info/METADATA,sha256=LiIaD8DgC3gUuUTkW5Opgk2Sh5alIdVh-2UzhIJxAHQ,3053
+quillsql-2.2.10.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+quillsql-2.2.10.dist-info/top_level.txt,sha256=eU2vHnVqwpYQJ3ADl1Q-DIBzbYejZRUhcMdN_4zMCz8,9
+quillsql-2.2.10.dist-info/RECORD,,