9a70264b51
纯排版无行为变化（ruff format 重排 7 个此前批次的文件）， 涉及模块集成回归全绿（67 passed）。 Co-Authored-By: Claude Code <noreply@anthropic.com>
126 lines
4.4 KiB
Python
126 lines
4.4 KiB
Python
"""回款数据访问：回款行与关联引用（业务号/发货单号/发货日）。
|
||
|
||
只读写数据并 flush；commit/rollback 与业务异常一律在 service 层。
|
||
|
||
- 列表与详情统一 join 出 ``business_no / shipment_no / ship_date``，
|
||
ship_date 供 service 按 ``formula.payment_days`` 派生回款天数 —— 关联
|
||
引用经本层查询一次供给，service 不再对发货单二次查询
|
||
- shipment_id 是手动指定（可空），发货单关联一律 outerjoin
|
||
"""
|
||
|
||
from datetime import date
|
||
|
||
from sqlalchemy import Select, func, select
|
||
from sqlalchemy.orm import Session
|
||
|
||
from app.models import Business, Receipt, Shipment
|
||
|
||
|
||
def get_by_id(session: Session, receipt_id: int) -> Receipt | None:
|
||
"""按主键取回款行（供摘要预取与状态判定；不锁行，写保护锚在业务行）。"""
|
||
return session.get(Receipt, receipt_id)
|
||
|
||
|
||
def get_with_refs(
|
||
session: Session, receipt_id: int
|
||
) -> tuple[Receipt, str, str | None, date | None] | None:
|
||
"""回款行 + 业务号 + 发货单号 + 发货日（DTO 与 payment_days 派生共用）。
|
||
|
||
business 主表为 inner join（FK NOT NULL，业务与回款同存亡）；发货单
|
||
可未指定，outerjoin 后 shipment_no / ship_date 可能为 None。
|
||
"""
|
||
row = session.execute(
|
||
select(Receipt, Business.business_no, Shipment.shipment_no, Shipment.ship_date)
|
||
.join(Business, Business.id == Receipt.business_id)
|
||
.outerjoin(Shipment, Shipment.id == Receipt.shipment_id)
|
||
.where(Receipt.id == receipt_id)
|
||
).first()
|
||
return row
|
||
|
||
|
||
def get_shipment(session: Session, shipment_id: int) -> Shipment | None:
|
||
"""按主键取发货单（同业务强校验的只读预查；FK RESTRICT 为最终防线）。"""
|
||
return session.get(Shipment, shipment_id)
|
||
|
||
|
||
def _apply_filters(
|
||
stmt: Select,
|
||
*,
|
||
business_id: int | None,
|
||
date_from: date | None,
|
||
date_to: date | None,
|
||
is_barter: int | None,
|
||
settlement_status: int | None,
|
||
) -> Select:
|
||
"""回款列表的筛选条件（全部落在 Receipt 自身列上）。"""
|
||
if business_id is not None:
|
||
stmt = stmt.where(Receipt.business_id == business_id)
|
||
if date_from is not None:
|
||
stmt = stmt.where(Receipt.receipt_date >= date_from)
|
||
if date_to is not None:
|
||
stmt = stmt.where(Receipt.receipt_date <= date_to)
|
||
if is_barter is not None:
|
||
stmt = stmt.where(Receipt.is_barter == is_barter)
|
||
if settlement_status is not None:
|
||
stmt = stmt.where(Receipt.settlement_status == settlement_status)
|
||
return stmt
|
||
|
||
|
||
def list_receipts(
|
||
session: Session,
|
||
*,
|
||
offset: int,
|
||
limit: int,
|
||
business_id: int | None,
|
||
date_from: date | None,
|
||
date_to: date | None,
|
||
is_barter: int | None,
|
||
settlement_status: int | None,
|
||
) -> list[tuple[Receipt, str, str | None, date | None]]:
|
||
"""分页查询回款（附业务号/发货单号/发货日），按 id 倒序。"""
|
||
stmt = _apply_filters(
|
||
select(Receipt, Business.business_no, Shipment.shipment_no, Shipment.ship_date)
|
||
.join(Business, Business.id == Receipt.business_id)
|
||
.outerjoin(Shipment, Shipment.id == Receipt.shipment_id),
|
||
business_id=business_id,
|
||
date_from=date_from,
|
||
date_to=date_to,
|
||
is_barter=is_barter,
|
||
settlement_status=settlement_status,
|
||
)
|
||
return list(session.execute(stmt.order_by(Receipt.id.desc()).offset(offset).limit(limit)).all())
|
||
|
||
|
||
def count_receipts(
|
||
session: Session,
|
||
*,
|
||
business_id: int | None,
|
||
date_from: date | None,
|
||
date_to: date | None,
|
||
is_barter: int | None,
|
||
settlement_status: int | None,
|
||
) -> int:
|
||
"""回款总数（筛选条件与列表共用同一 apply_filters，防两侧口径漂移）。"""
|
||
stmt = _apply_filters(
|
||
select(func.count()).select_from(Receipt),
|
||
business_id=business_id,
|
||
date_from=date_from,
|
||
date_to=date_to,
|
||
is_barter=is_barter,
|
||
settlement_status=settlement_status,
|
||
)
|
||
return int(session.execute(stmt).scalar_one())
|
||
|
||
|
||
def create(session: Session, receipt: Receipt) -> Receipt:
|
||
"""新增回款行。只 flush 以取得自增 id，编号唯一键冲突由 service 兜底。"""
|
||
session.add(receipt)
|
||
session.flush()
|
||
return receipt
|
||
|
||
|
||
def delete(session: Session, receipt: Receipt) -> None:
|
||
"""物理删除回款行（已核算与否的准入判定在 service 层）。"""
|
||
session.delete(receipt)
|
||
session.flush()
|