wio

streams/meili.py

@@ -1,7 +1,6 @@
-import json
-from decimal import Decimal
 from typing import Self
 
+from aws_lambda_powertools.shared.json_encoder import Encoder
 from aws_lambda_powertools.utilities.data_classes.dynamo_db_stream_event import (
     DynamoDBRecordEventName,
 )
@@ -50,20 +49,8 @@ class Op:
         return self.op[index][op].append(data)
 
 
-class DecimalEncoder(json.JSONEncoder):
-    def default(self, o):
-        if isinstance(o, Decimal):
-            if o % 1 != 0:
-                return float(o.quantize(Decimal('0.00')))
-            return int(o)
-        return super(__class__, self).default(o)
-
-
-class SetEncoder(json.JSONEncoder):
-    def default(self, o):
-        if isinstance(o, set):
-            return list(o)
-        return super(__class__, self).default(o)
-
-
-class JSONEncoder(SetEncoder, DecimalEncoder): ...
+class JSONEncoder(Encoder):
+    def default(self, obj):
+        if isinstance(obj, set):
+            return list(obj)
+        return super(__class__, self).default(obj)