添加修改assets方法

service/fund.py

-from typing import Union
-from model.fund import StakingFund, NormalFund
 from schema.fund import FundStatus
 
 
-async def query_fund(fund_collect, fund_id, user_id=None, fund_status=None) -> Union[StakingFund, NormalFund]:
+async def query_fund_assets(fund_collect, fund_id, user_id=None, fund_status=None):
     query = {'id': fund_id}
     if user_id:
         query.update({"user_id": user_id})
     if fund_status:
         query.update({"fund_status": FundStatus.active})
     fund = await fund_collect.find_one(query)
-    return fund
+    return fund['assets'], fund['adjust_assets'], fund['pending_assets'], fund['staking_assets']
 
 
 # 修改资产
-# assets[symbol] += amount
-
+async def update_assets(fund_collect, fund_id, assets=None, adjust_assets=None, pending_assets=None, staking_assets=None):
+    query = {'id': fund_id}
+    update_data = {}
+    if assets:
+        update_data.update({"assets": assets})
+    if adjust_assets:
+        update_data.update({"adjust_assets": adjust_assets})
+    if pending_assets:
+        update_data.update({"pending_assets": pending_assets})
+    if staking_assets:
+        update_data.update({"staking_assets": staking_assets})
+    await fund_collect.update_one(query, {"$set": update_data})