81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
"""资金账户服务：余额查询 + 余额加减（路由层只编排，不碰数据/逻辑）。"""
|
||
from __future__ import annotations
|
||
|
||
from decimal import Decimal
|
||
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
||
from model.fin_account import FinAccount
|
||
from model.sys_user import SysUser
|
||
from repositories.fin_account import FinAccountRepo
|
||
from schemas.account import BalanceResp
|
||
from utils.exceptions import ForbiddenError, NotFoundError, ParamError
|
||
|
||
_ZERO = Decimal("0.00")
|
||
_DEFAULT_CURRENCY = "CNY"
|
||
|
||
|
||
def _build_balance_resp(account: FinAccount) -> BalanceResp:
|
||
return BalanceResp(
|
||
customer_id=account.customer_id,
|
||
balance=account.balance,
|
||
available_balance=account.balance - account.frozen_amount,
|
||
frozen_amount=account.frozen_amount,
|
||
currency=account.currency,
|
||
status=account.status,
|
||
update_time=account.update_time,
|
||
)
|
||
|
||
|
||
async def get_balance(db: AsyncSession, user: SysUser) -> BalanceResp:
|
||
"""查询当前用户现金余额。
|
||
|
||
- 仅客户账号可查（员工共用 sys_user，但无资金账户）；
|
||
- 未开资金户时按零余额返回，不报错。
|
||
"""
|
||
if user.user_type != "CUSTOMER":
|
||
raise ForbiddenError("仅客户账号可查询资金余额")
|
||
|
||
account = await FinAccountRepo(db).get_by_customer_id(user.id)
|
||
if account is None:
|
||
return BalanceResp(
|
||
customer_id=user.id,
|
||
balance=_ZERO,
|
||
available_balance=_ZERO,
|
||
frozen_amount=_ZERO,
|
||
currency=_DEFAULT_CURRENCY,
|
||
status="正常",
|
||
)
|
||
|
||
return _build_balance_resp(account)
|
||
|
||
|
||
async def adjust_balance(
|
||
db: AsyncSession, user: SysUser, direction: str, amount: Decimal
|
||
) -> BalanceResp:
|
||
"""加减余额：direction=add 入账 / sub 出账。
|
||
|
||
- add：balance += amount，未开资金户时自动开户入账；
|
||
- sub：balance -= amount，可用余额（balance - frozen_amount）不足时报错。
|
||
"""
|
||
if user.user_type != "CUSTOMER":
|
||
raise ForbiddenError("仅客户账号可调整余额")
|
||
|
||
repo = FinAccountRepo(db)
|
||
account = await repo.get_by_customer_id(user.id)
|
||
|
||
if direction == "add":
|
||
if account is None:
|
||
account = await repo.create(user.id, amount)
|
||
else:
|
||
account = await repo.add_balance(user.id, amount)
|
||
else: # sub
|
||
if account is None:
|
||
raise NotFoundError("资金账户不存在")
|
||
updated = await repo.subtract_balance(user.id, amount)
|
||
if updated is None:
|
||
raise ParamError("可用余额不足")
|
||
account = updated
|
||
|
||
return _build_balance_resp(account)
|