commit unversioned files ;)

common/utils/payment.py

@@ -0,0 +1,30 @@
+import datetime
+
+from models.enums import ContainerLength, ContainerType
+
+PAYMENT_SCHEMA = {
+    "standart": [[0, 5, 25], [5, 20, 40], [40, 1000000, 75]],
+    "reefer": [[0, 7, 35], [7, 30, 60], [30, 1000000, 115]],
+}
+
+
+def calc_storage_amount(depot, container):
+    result = (depot.expedited_on - depot.received_on + datetime.timedelta(days=1)).days
+    if container.container_type == ContainerType.ct_reefer:
+        data = PAYMENT_SCHEMA["reefer"]
+        result -= 1
+    else:
+        data = PAYMENT_SCHEMA["standart"]
+
+    for i in data:
+        start, end, price = i
+        if start <= result < end:
+            price_per_day = price
+            break
+    if container.container_length in (
+        ContainerLength.cl_40_ft,
+        ContainerLength.cl_45_ft,
+    ):
+        result *= 2
+    result = result * price_per_day
+    return result