修改删除账目管理

api/bill.py

-import datetime
-import re
 from typing import Union, List, Any
-
-import pytz
 from fastapi import APIRouter, Depends, Query
 from motor.core import AgnosticCollection
-from pymongo import UpdateOne
 from dependencies import get_current_user, get_fund_collect, get_bill_collect, get_permission_user_collect, \
     get_permission_role_collect, get_nav_collect
+from exception.api import APIError
 from model import Response, Page, PageResponse, SortParams, FilterTime
 from model.bill import PCFBill, ExchangeBill, AdjustBill, StakingBill
 from model.node import BaseNode
 from schema.bill import CreatePCFBill, PCFBillType, CreateExchangeBill, CreateAdjustBill, CreateStakingBill, \
-    UpdatePCFBill, UpdateExchangeBill, UpdateStakingBill, UpdateAdjustBill, AllBillType, StakingDirection
+    UpdatePCFBill, UpdateExchangeBill, UpdateStakingBill, UpdateAdjustBill, AllBillType, StakingDirection, BillType
 from schema.fund import FundStatus
 from schema.node import BindNode
 from service.beacon import BeaconChaService
@@ -159,7 +155,7 @@ async def create_adjust(
         fund_id=create_adjust_bill.fund_id,
         record_time=create_adjust_bill.record_time,
         settlement_time=fund_data["settlement_time"],
-        asset_changed={
+        adjust_assets_changed={
             create_adjust_bill.currency: create_adjust_bill.volume,
             "fund_share": create_adjust_bill.fund_share
         }
@@ -244,6 +240,7 @@ async def update_pcf_bill(
         user: User = Depends(get_current_user),
         fund_collect: AgnosticCollection = Depends(get_fund_collect),
         bill_collect: AgnosticCollection = Depends(get_bill_collect),
+        nav_collect: AgnosticCollection = Depends(get_nav_collect),
         permission_user_collect: AgnosticCollection = Depends(get_permission_user_collect),
         permission_role_collect: AgnosticCollection = Depends(get_permission_role_collect)
 ):
@@ -251,6 +248,25 @@ async def update_pcf_bill(
                            fund_id,
                            user.email, permission_user_collect,
                            permission_role_collect)
+    fund_data = await fund_collect.find_one({"id": fund_id})
+    bill_data = await bill_collect.find_one({"id": bill_id, "bill_type": {"$in": [PCFBillType.sub, PCFBillType.redemption]}})
+    A = {bill_data["currency"]: bill_data["volume"]}
+    B = {update_item.currency: update_item.volume}
+    changed = {k: B.get(k, 0) - A.get(k, 0) for k in set(A) | set(B)}
+    bulk_list = await build_nav_bulk_list(
+        nav_collect=nav_collect,
+        fund_id=fund_id,
+        record_time=bill_data["record_time"],
+        settlement_time=fund_data["settlement_time"],
+        asset_changed=changed
+    )
+    assets = fund_data["assets"]
+    for key, value in changed.items():
+        assets.setdefault(key, 0)
+        assets[key] += value
+    await update_fund(fund_collect, fund_id, assets=assets)
+    if bulk_list:
+        await nav_collect.bulk_write(bulk_list)
     response = await update_bill(
         bill_id=bill_id,
         fund_id=fund_id,
@@ -274,6 +290,7 @@ async def update_exchange_bill(
         user: User = Depends(get_current_user),
         fund_collect: AgnosticCollection = Depends(get_fund_collect),
         bill_collect: AgnosticCollection = Depends(get_bill_collect),
+        nav_collect: AgnosticCollection = Depends(get_nav_collect),
         permission_user_collect: AgnosticCollection = Depends(get_permission_user_collect),
         permission_role_collect: AgnosticCollection = Depends(get_permission_role_collect)
 ):
@@ -281,6 +298,25 @@ async def update_exchange_bill(
                            fund_id,
                            user.email, permission_user_collect,
                            permission_role_collect)
+    fund_data = await fund_collect.find_one({"id": fund_id})
+    bill_data = await bill_collect.find_one({"id": bill_id, "bill_type": BillType.exchange})
+    A = {bill_data["input_currency"]: -bill_data["input_volume"], bill_data["output_currency"]: bill_data["output_volume"]}
+    B = {update_item.input_currency: -update_item.input_volume, update_item.output_currency: update_item.output_volume}
+    changed = {k: B.get(k, 0) - A.get(k, 0) for k in set(A) | set(B)}
+    bulk_list = await build_nav_bulk_list(
+        nav_collect=nav_collect,
+        fund_id=fund_id,
+        record_time=bill_data["record_time"],
+        settlement_time=fund_data["settlement_time"],
+        asset_changed=changed
+    )
+    assets = fund_data["assets"]
+    for key, value in changed.items():
+        assets.setdefault(key, 0)
+        assets[key] += value
+    await update_fund(fund_collect, fund_id, assets=assets)
+    if bulk_list:
+        await nav_collect.bulk_write(bulk_list)
     response = await update_bill(
         bill_id=bill_id,
         fund_id=fund_id,
@@ -304,6 +340,7 @@ async def update_adjust_bill(
         user: User = Depends(get_current_user),
         fund_collect: AgnosticCollection = Depends(get_fund_collect),
         bill_collect: AgnosticCollection = Depends(get_bill_collect),
+        nav_collect: AgnosticCollection = Depends(get_nav_collect),
         permission_user_collect: AgnosticCollection = Depends(get_permission_user_collect),
         permission_role_collect: AgnosticCollection = Depends(get_permission_role_collect)
 ):
@@ -311,6 +348,25 @@ async def update_adjust_bill(
                            fund_id,
                            user.email, permission_user_collect,
                            permission_role_collect)
+    fund_data = await fund_collect.find_one({"id": fund_id})
+    bill_data = await bill_collect.find_one({"id": bill_id, "bill_type": BillType.adjust})
+    A = {bill_data["currency"]: bill_data["volume"], "fund_share": bill_data["fund_share"]}
+    B = {update_item.currency: update_item.volume, "fund_share": update_item.fund_share}
+    changed = {k: B.get(k, 0) - A.get(k, 0) for k in set(A) | set(B)}
+    bulk_list = await build_nav_bulk_list(
+        nav_collect=nav_collect,
+        fund_id=fund_id,
+        record_time=bill_data["record_time"],
+        settlement_time=fund_data["settlement_time"],
+        asset_changed=changed
+    )
+    adjust_assets = fund_data["adjust_assets"]
+    for key, value in changed.items():
+        adjust_assets.setdefault(key, 0)
+        adjust_assets[key] += value
+    await update_fund(fund_collect, fund_id, adjust_assets=adjust_assets)
+    if bulk_list:
+        await nav_collect.bulk_write(bulk_list)
     response = await update_bill(
         bill_id=bill_id,
         fund_id=fund_id,
@@ -394,7 +450,9 @@ async def query_bill(
         fund_id: str,
         bill_id: str,
         bill_type: AllBillType,
+        fund_collect: AgnosticCollection = Depends(get_fund_collect),
         bill_collect: AgnosticCollection = Depends(get_bill_collect),
+        nav_collect: AgnosticCollection = Depends(get_nav_collect),
         user: User = Depends(get_current_user),
         permission_user_collect: AgnosticCollection = Depends(get_permission_user_collect),
         permission_role_collect: AgnosticCollection = Depends(get_permission_role_collect)
@@ -405,5 +463,40 @@ async def query_bill(
                            permission_role_collect)
 
     query = {"fund_id": fund_id, "bill_type": bill_type, "id": bill_id}
+    fund_data = await fund_collect.find_one({"id": fund_id})
+    bill_data = await bill_collect.find_one({"id": bill_id})
+    assets_changed, adjust_changed = {}, {}
+    assets, adjust_assets = fund_data["assets"], fund_data["adjust_assets"]
+    if bill_data["bill_type"] == PCFBillType.sub or bill_data["bill_type"] == PCFBillType.redemption:
+        delta_volume = -bill_data["volume"] if bill_data["bill_type"] == PCFBillType.sub else bill_data["volume"]
+        assets_changed = {bill_data["currency"]: delta_volume}
+        assets[bill_data["currency"]] += delta_volume
+    elif bill_data["bill_type"] == BillType.exchange:
+        assets_changed.update({
+            bill_data["input_currency"]: bill_data["input_volume"],
+            bill_data["output_currency"]: -bill_data["output_volume"]
+        })
+        assets[bill_data["input_currency"]] += bill_data["input_volume"]
+        assets[bill_data["output_currency"]] -= bill_data["output_volume"]
+    elif bill_data["bill_type"] == BillType.adjust:
+        adjust_changed.update({
+            bill_data["currency"]: -bill_data["volume"],
+            "fund_share": -bill_data["fund_share"]
+        })
+        adjust_assets[bill_data["currency"]] -= bill_data["volume"]
+        adjust_assets["fund_share"] -= bill_data["fund_share"]
+    else:
+        raise APIError(message="该账单不支持删除")
+    bulk_list = await build_nav_bulk_list(
+        nav_collect=nav_collect,
+        fund_id=fund_id,
+        record_time=bill_data["record_time"],
+        settlement_time=fund_data["settlement_time"],
+        asset_changed=assets_changed,
+        adjust_assets_changed=adjust_changed
+    )
+    await update_fund(fund_collect, fund_id, assets=assets, adjust_assets=adjust_assets)
+    if bulk_list:
+        await nav_collect.bulk_write(bulk_list)
     await bill_collect.delete_one(query)
     return Response()