statikk 0.1.19__tar.gz → 0.1.21__tar.gz

{statikk-0.1.19 → statikk-0.1.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: statikk
-Version: 0.1.19
+Version: 0.1.21
 Summary: statikk is a single table application (STA) library for DynamoDb.
 Home-page: https://github.com/terinia/statikk
 Author: Balint Biro

{statikk-0.1.19 → statikk-0.1.21}/src/statikk/engine.py

@@ -1,4 +1,5 @@
 import os
+import decimal
 from datetime import datetime
 from typing import Any, Dict, Type, Optional, List, Union
 
@@ -23,6 +24,14 @@ from aws_xray_sdk.core import patch_all
 patch_all()
 
 
+DYNAMODB_CONTEXT = decimal.Context(
+    prec=38,  # precision
+    rounding=decimal.ROUND_HALF_EVEN,
+    traps=[decimal.Overflow, decimal.Underflow, decimal.InvalidOperation],
+)
+decimal.setcontext(DYNAMODB_CONTEXT)
+
+
 class InvalidIndexNameError(Exception):
     pass
 
@@ -535,6 +544,8 @@ class Table:
 
         if actual_annotation is datetime or "datetime" in str(annotation) and value is not None:
             return datetime.fromtimestamp(int(value))
+        if isinstance(value, decimal.Decimal):
+            return float(value)
         if actual_annotation is float:
             return float(value)
         if actual_annotation is list:
@@ -555,7 +566,7 @@ class Table:
         if isinstance(value, datetime):
             return int(value.timestamp())
         if isinstance(value, float):
-            return Decimal(value)
+            return DYNAMODB_CONTEXT.create_decimal_from_float(value)
         if isinstance(value, list):
             return [self._serialize_value(item) for item in value]
         if isinstance(value, set):

{statikk-0.1.19 → statikk-0.1.21}/src/statikk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: statikk
-Version: 0.1.19
+Version: 0.1.21
 Summary: statikk is a single table application (STA) library for DynamoDb.
 Home-page: https://github.com/terinia/statikk
 Author: Balint Biro

{statikk-0.1.19 → statikk-0.1.21}/tests/test_engine.py

@@ -1,6 +1,6 @@
 from _decimal import Decimal
 from datetime import datetime, timezone
-from typing import List, Optional, Type
+from typing import List, Optional, Type, Any
 
 import pytest
 from boto3.dynamodb.conditions import Attr
@@ -996,4 +996,22 @@ def test_add_child_node():
     my_database_model = MyDatabaseModel.query_hierarchy(hash_key=Equals("foo"))
     assert len(my_database_model.nested.list_nested) == 1
     assert len(my_database_model.nested.other_list_nested) == 1
-    hierarchy = MyDatabaseModel.query_hierarchy(hash_key=Equals("foo"))
+
+
+def test_nested_model_with_decimal():
+    class MyModel(DatabaseModel):
+        id: str
+        foo: str = "foo"
+        values: dict[str, Any]
+
+        @classmethod
+        def index_definitions(cls) -> dict[str, IndexFieldConfig]:
+            return {"main-index": IndexFieldConfig(pk_fields=["foo"], sk_fields=[FIELD_STATIKK_TYPE])}
+
+    _create_default_dynamodb_table([MyModel])
+
+    my_model = MyModel(id="foo", values={"bar": 1.23})
+    my_model.save()
+    my_model_deserialized = MyModel.get("foo")
+    assert type(my_model_deserialized.values["bar"]) == float
+    assert my_model_deserialized.values["bar"] == 1.23