76 lines
3.4 KiB
Python
76 lines
3.4 KiB
Python
"""Чтение версионированных правил элиминации и счетов их сторон."""
|
||
|
||
import uuid
|
||
from collections.abc import Collection
|
||
|
||
from sqlalchemy import select
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
||
from src.models.accounting import ChartOfAccount, EliminationRule, EliminationRuleAccount
|
||
|
||
|
||
async def list_rules_with_accounts(
|
||
session: AsyncSession, version_no: int | None
|
||
) -> list[tuple[EliminationRule, EliminationRuleAccount | None]]:
|
||
"""Правила заданной версии либо все правила без `valid_to`."""
|
||
condition = (
|
||
EliminationRule.version_no == version_no if version_no is not None else EliminationRule.valid_to.is_(None)
|
||
)
|
||
stmt = (
|
||
select(EliminationRule, EliminationRuleAccount)
|
||
.outerjoin(EliminationRuleAccount, EliminationRuleAccount.rule_id == EliminationRule.id)
|
||
.where(condition)
|
||
.order_by(
|
||
EliminationRule.statement_type,
|
||
EliminationRule.rule_no,
|
||
EliminationRuleAccount.company_no,
|
||
)
|
||
)
|
||
return [(rule, account) for rule, account in (await session.execute(stmt)).all()]
|
||
|
||
|
||
def is_valid_for_version(rule: EliminationRule | None, version_no: int | None) -> bool:
|
||
"""Проверить номер версии либо отсутствие `valid_to`."""
|
||
return rule is not None and (rule.version_no == version_no if version_no is not None else rule.valid_to is None)
|
||
|
||
|
||
async def list_rules(session: AsyncSession, version_no: int) -> list[EliminationRule]:
|
||
"""Все правила заданной версии, включая ББ и ОФР, в порядке типа отчётности и номера."""
|
||
stmt = (
|
||
select(EliminationRule)
|
||
.where(EliminationRule.version_no == version_no)
|
||
.order_by(EliminationRule.statement_type, EliminationRule.rule_no)
|
||
)
|
||
return list((await session.scalars(stmt)).all())
|
||
|
||
|
||
async def list_rule_accounts(
|
||
session: AsyncSession, rule_ids: Collection[uuid.UUID]
|
||
) -> list[tuple[EliminationRuleAccount, str]]:
|
||
"""Счета правил с кодами из плана счетов; `company_no` задаёт сторону, `is_excluded` — исключение."""
|
||
if not rule_ids:
|
||
return []
|
||
stmt = (
|
||
select(EliminationRuleAccount, ChartOfAccount.code)
|
||
.join(ChartOfAccount, ChartOfAccount.id == EliminationRuleAccount.account_id)
|
||
.where(EliminationRuleAccount.rule_id.in_(rule_ids))
|
||
.order_by(EliminationRuleAccount.rule_id, EliminationRuleAccount.company_no, ChartOfAccount.code)
|
||
)
|
||
return [(account, code) for account, code in (await session.execute(stmt)).all()]
|
||
|
||
|
||
async def rule_with_account(
|
||
session: AsyncSession, rule_id: uuid.UUID
|
||
) -> tuple[EliminationRule | None, list[EliminationRuleAccount]]:
|
||
"""Правило и его счета; `company_no` задаёт сторону, `is_excluded` — исключение."""
|
||
rule = await session.get(EliminationRule, rule_id)
|
||
if rule is None:
|
||
return None, []
|
||
stmt = (
|
||
select(EliminationRuleAccount)
|
||
.join(ChartOfAccount, ChartOfAccount.id == EliminationRuleAccount.account_id)
|
||
.where(EliminationRuleAccount.rule_id == rule_id)
|
||
.order_by(EliminationRuleAccount.company_no, ChartOfAccount.code)
|
||
)
|
||
return rule, list((await session.scalars(stmt)).all())
|