128 lines
5.0 KiB
Python
128 lines
5.0 KiB
Python
import time
|
||
import logging
|
||
import requests
|
||
from rest_framework.views import APIView
|
||
from rest_framework.response import Response
|
||
from rest_framework import status, viewsets
|
||
from .models import Nomenclature, Sticker, StickerMovement
|
||
from .serializers import NomenclatureSerializer, StickerSerializer, StickerMovementSerializer
|
||
|
||
# Авторизационные данные
|
||
USERNAME = "superuser"
|
||
PASSWORD = "Superuser1105"
|
||
TOKEN_URL = "http://192.168.254.2:8280/api/login" # URL для получения токена
|
||
PRODUCT_LIST_URL = "http://192.168.254.2:8280/api/ProductType/list"
|
||
|
||
# Глобальный кэш для хранения токена
|
||
token_cache = {"token": None, "timestamp": 0}
|
||
TOKEN_LIFETIME = 1440 * 60 # 24 часа в секундах
|
||
|
||
|
||
def get_new_token():
|
||
"""Получение нового токена с кешированием."""
|
||
global token_cache
|
||
current_time = time.time()
|
||
|
||
# Если токен есть и он еще не истек, возвращаем его
|
||
if token_cache["token"] and (current_time - token_cache["timestamp"] < TOKEN_LIFETIME):
|
||
return token_cache["token"]
|
||
|
||
payload = {"UserName": USERNAME, "Password": PASSWORD}
|
||
try:
|
||
response = requests.post(TOKEN_URL, json=payload, timeout=5)
|
||
response_data = response.json()
|
||
|
||
if response.status_code == 200 and response_data.get("IsSuccess"):
|
||
token_cache["token"] = response_data["Value"]["Token"]
|
||
token_cache["timestamp"] = current_time
|
||
return token_cache["token"]
|
||
|
||
logging.error(f"Ошибка получения токена: {response_data}")
|
||
return None
|
||
except requests.RequestException as e:
|
||
logging.error(f"Ошибка сети при получении токена: {str(e)}")
|
||
return None
|
||
|
||
|
||
class ExternalNomenclatureView(APIView):
|
||
"""Получение списка номенклатур с внешнего API."""
|
||
|
||
def post(self, request):
|
||
product_line_id = request.data.get("product_line_id", "40ba525b-1686-4900-a2a9-443436812b1d")
|
||
token = get_new_token()
|
||
|
||
if not token:
|
||
return Response({"error": "Ошибка авторизации"}, status=status.HTTP_401_UNAUTHORIZED)
|
||
|
||
headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
|
||
payload = {
|
||
"Filter": {
|
||
"Filters": [
|
||
{
|
||
"Logic": "and",
|
||
"Operator": "linq",
|
||
"Field": "ProductType_ProductLines.Where(i=>i.ProductLineId=='{0}').Any()",
|
||
"Value": product_line_id,
|
||
}
|
||
],
|
||
"Logic": "and",
|
||
},
|
||
"Includes": ["ProductType_ProductLines"],
|
||
}
|
||
|
||
try:
|
||
response = requests.post(PRODUCT_LIST_URL, json=payload, headers=headers, timeout=10)
|
||
response.raise_for_status()
|
||
return Response(response.json(), status=status.HTTP_200_OK)
|
||
except requests.RequestException as e:
|
||
logging.error(f"Ошибка запроса {PRODUCT_LIST_URL}: {str(e)}")
|
||
return Response({"error": "Ошибка запроса к API"}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
|
||
|
||
|
||
# Основные API ViewSet'ы
|
||
class NomenclatureViewSet(viewsets.ModelViewSet):
|
||
queryset = Nomenclature.objects.all()
|
||
serializer_class = NomenclatureSerializer
|
||
|
||
|
||
class StickerViewSet(viewsets.ModelViewSet):
|
||
queryset = Sticker.objects.all().select_related('nomenclature')
|
||
serializer_class = StickerSerializer
|
||
|
||
def create(self, request, *args, **kwargs):
|
||
data = request.data
|
||
gtin = data.get("nomenclature_gtin") # Берем GTIN из запроса
|
||
|
||
if not gtin:
|
||
return Response({"error": "GTIN обязателен"}, status=status.HTTP_400_BAD_REQUEST)
|
||
|
||
# Проверяем, есть ли номенклатура с таким GTIN
|
||
nomenclature, created = Nomenclature.objects.get_or_create(
|
||
gtin=gtin,
|
||
defaults={"name": data.get("nomenclature_name", "Неизвестный продукт")}
|
||
)
|
||
|
||
# Обновляем данные запроса, чтобы стикер использовал ID найденной/созданной номенклатуры
|
||
data["nomenclature"] = nomenclature.id
|
||
|
||
serializer = self.get_serializer(data=data)
|
||
if serializer.is_valid():
|
||
serializer.save()
|
||
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
||
|
||
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
||
|
||
def destroy(self, request, *args, **kwargs):
|
||
instance = self.get_object()
|
||
self.perform_destroy(instance)
|
||
return Response({"message": "Стикер успешно удалён"}, status=status.HTTP_204_NO_CONTENT)
|
||
|
||
|
||
|
||
"""Передача стикеров"""
|
||
|
||
class StickerMovementViewSet(viewsets.ModelViewSet):
|
||
queryset = StickerMovement.objects.all()
|
||
serializer_class = StickerMovementSerializer
|
||
|