111 lines
4.2 KiB
Python
111 lines
4.2 KiB
Python
from fastapi import APIRouter, Depends, HTTPException
|
||
from bson import ObjectId
|
||
from datetime import datetime
|
||
from models import DeliveryAcceptRequest, DeliverySendRequest
|
||
from db import delivery_collection, notify_collection
|
||
from auth import get_current_user
|
||
from tasks import notify_in_profile
|
||
from models import NotifyItem
|
||
|
||
|
||
router = APIRouter(prefix="/api/delivery", tags=["delivery"])
|
||
|
||
|
||
@router.post("/send")
|
||
async def send_delivery(
|
||
payload: DeliverySendRequest,
|
||
current_user: dict = Depends(get_current_user)
|
||
):
|
||
"""
|
||
Создаёт новый документ отправки с обновлёнными статусом и полями.
|
||
"""
|
||
# Проверяем, что текущий пользователь — отправитель
|
||
if str(current_user["_id"]) != payload.sender:
|
||
raise HTTPException(status_code=403, detail="Не авторизован как отправитель")
|
||
|
||
# Валидация delivery_id
|
||
try:
|
||
delivery_id = ObjectId(payload.item)
|
||
except Exception:
|
||
raise HTTPException(status_code=400, detail="Неверный delivery ID")
|
||
|
||
update_data = {
|
||
"status": "Ожидает получения",
|
||
"delivery_type": payload.delivery_type,
|
||
"sent_at": datetime.utcnow(),
|
||
"updated_at": datetime.utcnow()
|
||
}
|
||
|
||
# Добавляем трек только для СДЭК (если передан)
|
||
if payload.delivery_type == "СДЭК" and payload.track:
|
||
update_data["track"] = payload.track
|
||
|
||
# Ищем существующую запись, чтобы создать новую на её основе
|
||
existing = await delivery_collection.find_one(
|
||
{"_id": delivery_id, "sender._id": payload.sender},
|
||
{"_id": 0}
|
||
)
|
||
|
||
if not existing:
|
||
raise HTTPException(status_code=404, detail="Доставка не найдена или не для вас")
|
||
|
||
# Собираем новый документ для вставки
|
||
notify = await notify_in_profile(NotifyItem(
|
||
user_id=str(existing['receiver']['_id']),
|
||
type_item='delivery',
|
||
title='Доставка картин',
|
||
link=f'/profile/delivery',
|
||
text=f'{str(existing['sender']['username'])} — прислал вам картины.'
|
||
))
|
||
new_doc = {**existing, **update_data, 'notify':notify.upserted_id}
|
||
|
||
# Вставляем новый документ
|
||
result = await delivery_collection.insert_one(new_doc)
|
||
if (iserted_id:=result.inserted_id):
|
||
pass
|
||
# send email
|
||
|
||
return {"status": "success", "delivery_id": str(iserted_id)}
|
||
|
||
|
||
@router.post("/accept")
|
||
async def accept_delivery(
|
||
payload: DeliveryAcceptRequest,
|
||
current_user: dict = Depends(get_current_user)
|
||
):
|
||
"""
|
||
Создаёт новый документ отправки со статусом 'Получено'.
|
||
"""
|
||
try:
|
||
delivery_id = ObjectId(payload.item)
|
||
except Exception:
|
||
raise HTTPException(status_code=400, detail="Неверный delivery ID")
|
||
|
||
# Берём исходный документ, чтобы вставить новый
|
||
existing = await delivery_collection.find_one(
|
||
{"_id": delivery_id, "receiver._id": current_user["_id"]},
|
||
{"_id": 0}
|
||
)
|
||
|
||
if not existing:
|
||
raise HTTPException(status_code=404, detail="Доставка не найдена или не для вас")
|
||
notify = await notify_in_profile(NotifyItem(
|
||
user_id=str(existing['sender']['_id']),
|
||
type_item='delivery',
|
||
title='Доставка картин',
|
||
link=f'/profile/delivery',
|
||
text=f'Картина «{existing['title']}» получена — {str(existing['receiver']['username'])}.'
|
||
))
|
||
new_doc = {
|
||
**existing,
|
||
"status": "Получено",
|
||
"notify": notify.upserted_id,
|
||
"received_at": datetime.utcnow(),
|
||
"updated_at": datetime.utcnow(),
|
||
}
|
||
|
||
result = await delivery_collection.insert_one(new_doc)
|
||
|
||
await notify_collection.update_one({'_id': existing['notify']}, {'$set': {'read': False}})
|
||
return {"status": "success", "delivery_id": str(result.inserted_id)}
|