first

apps/shop/services/order.py

@@ -0,0 +1,111 @@
+import json
+from apps.shop.models import (
+    Order as OrderModel,
+    OrderItem as OrderItemModel,
+    Product as ProductModel,
+)
+from apps.customer.models import Customer as CustomerModel
+from rest_framework import exceptions
+from decimal import Decimal
+from django.db import transaction
+
+
+def create_order(responsible_group_id: int, customer_id: int,
+                 total_amount: Decimal, factor_type: int, factor_status: int, workspace) -> OrderModel:
+    try:
+        order_obj = OrderModel()
+        order_obj.responsible_group_id = responsible_group_id
+        order_obj.customer_id = customer_id
+        order_obj.total_amount = total_amount
+        order_obj.factor_type = factor_type
+        order_obj.factor_status = factor_status
+        order_obj.sys_ws = workspace
+
+        order_obj.full_clean()
+        order_obj.save()
+
+        return order_obj
+
+    except Exception as exp:
+        message = f'Error Create Order: {exp}'
+        exceptions.ValidationError({'error': message})
+
+
+def create_order_and_items(items: str, customer: CustomerModel, workspace) -> OrderModel:
+    with transaction.atomic():
+        try:
+            sum_price = 0
+            order_obj = OrderModel()
+            order_obj.responsible_group = customer.responsible_group
+            order_obj.customer = customer
+            order_obj.total_price = 0
+            order_obj.factor_type = 1
+            order_obj.factor_status = 1
+            order_obj.sys_ws = workspace
+
+            order_obj.full_clean()
+            order_obj.save()
+
+            json_items = json.loads(items)
+
+            for item in json_items:
+                product_obj = ProductModel.objects.get(id=int(item['product_id']))
+                product_price = product_obj.price
+                product_name = product_obj.name
+
+                order_item_obj = OrderItemModel()
+                order_item_obj.order = order_obj
+                order_item_obj.product_name = product_name
+                order_item_obj.price = product_price
+                order_item_obj.amount = int(item['amount'])
+                order_item_obj.product_id = int(item['product_id'])
+
+                order_item_obj.full_clean()
+                order_item_obj.save()
+
+                sum_price += product_price
+
+            order_obj.total_price = sum_price
+            order_obj.full_clean()
+            order_obj.save()
+
+            return order_obj
+
+        except Exception as exp:
+            transaction.set_rollback(True)
+            message = f'Error Create Order: {exp}'
+            raise exceptions.ValidationError({'error': message})
+
+
+def update_order(order_id: int, responsible_group_id: int = None, customer_id: int = None,
+                 total_amount: Decimal = None, factor_type: int = None, factor_status: int = None, workspace_uuid = None) -> OrderModel:
+    try:
+        order_obj = OrderModel.objects.get(id=order_id, sys_ws__uuid=workspace_uuid)
+
+        if responsible_group_id is not None:
+            order_obj.responsible_group_id = responsible_group_id
+        if customer_id is not None:
+            order_obj.customer_id = customer_id
+        if total_amount is not None:
+            order_obj.total_amount = total_amount
+        if factor_type is not None:
+            order_obj.factor_type = factor_type
+        if factor_status is not None:
+            order_obj.factor_status = factor_status
+
+        order_obj.full_clean()
+        order_obj.save()
+        return order_obj
+
+    except Exception as exp:
+        message = f'Error Updating Order: {exp}'
+        raise exceptions.ValidationError({'error': message})
+
+
+def delete_order(order_id: int, workspace_uuid: str) -> OrderModel:
+    try:
+        order_obj = OrderModel.objects.get(id=order_id, sys_ws__uuid=workspace_uuid).delete()
+        return order_obj
+    except Exception as exp:
+        message = f'Error Deleting Order: {exp}'
+        raise exceptions.ValidationError({'error': message})