32 lines
930 B
Python
32 lines
930 B
Python
from apps.project.models import Project as ProjectModel
|
|
from apps.order.models import Order as OrderModel
|
|
from rest_framework.exceptions import ValidationError
|
|
|
|
|
|
def get_order(order_id):
|
|
try:
|
|
order = OrderModel.objects.get(id=order_id)
|
|
return order
|
|
except Exception as exp:
|
|
raise ValidationError({'error': f'error get order: {str(exp)}'})
|
|
|
|
|
|
def get_orders_list(status: str = 'waiting'):
|
|
try:
|
|
base_filter = {}
|
|
if status:
|
|
base_filter['status'] = status
|
|
|
|
orders = OrderModel.objects.filter(**base_filter)
|
|
return orders
|
|
except Exception as exp:
|
|
raise ValidationError({'error': f'error get orders: {str(exp)}'})
|
|
|
|
|
|
def get_project_by_id(project_id):
|
|
try:
|
|
project = ProjectModel.objects.get(id=project_id)
|
|
return project
|
|
except Exception as exp:
|
|
raise ValidationError({'error': f'error get project: {str(exp)}'})
|