internal 1.1.31__tar.gz → 1.1.32.1__tar.gz

{internal-1.1.31 → internal-1.1.32.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: internal
-Version: 1.1.31
+Version: 1.1.32.1
 Summary: 
 Author: Ray
 Author-email: ray@cruisys.com

{internal-1.1.31 → internal-1.1.32.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "internal"
-version = "1.1.31"
+version = "1.1.32.1"
 description = ""
 authors = [{ name = "Ray", email = "ray@cruisys.com" }]
 readme = "README.md"

internal-1.1.32.1/src/internal/common_enum/point_type.py

@@ -0,0 +1,6 @@
+from enum import Enum
+
+
+class PointTypeEnum(str, Enum):
+    INVOICE_REWARD = "invoice_reward"
+    ACTIVITY = "activity"

{internal-1.1.31 → internal-1.1.32.1}/src/internal/database.py

@@ -2,6 +2,18 @@ from motor.motor_asyncio import AsyncIOMotorClient
 from pymongo.errors import ServerSelectionTimeoutError
 
 from .exception.internal_exception import DatabaseInitializeFailureException, DatabaseConnectFailureException
+from bson import Decimal128
+from bson.codec_options import CodecOptions, TypeDecoder, TypeRegistry
+from decimal import Decimal
+
+
+class Decimal128Decoder(TypeDecoder):
+    """自定義 Decimal128 解碼器，自動轉換為 Python Decimal"""
+    bson_type = Decimal128
+    python_type = Decimal
+
+    def transform_bson(self, value):
+        return value.to_decimal()
 
 
 class MongoDB:
@@ -31,7 +43,7 @@ class MongoDB:
             if self.ssl_ca_certs and self.ssl_ca_certs != "":
                 db_settings["tlsCAFile"] = self.ssl_ca_certs
 
-            self.client = AsyncIOMotorClient(**db_settings)
+            self.client = AsyncIOMotorClient(**db_settings).get_database()
         except ServerSelectionTimeoutError:
             raise DatabaseInitializeFailureException()
 
@@ -42,7 +54,11 @@ class MongoDB:
     def get_database(self):
         if not self.client:
             raise DatabaseConnectFailureException()
-        return self.client[self.db_name]
+
+        type_registry = TypeRegistry([Decimal128Decoder()])
+        codec_options = CodecOptions(type_registry=type_registry)
+
+        return self.client.get_database(self.db_name, codec_options=codec_options)
 
     async def get_mongodb_uri(self) -> str:
         if self.user_name and self.password: