58 lines
2.2 KiB
Python
58 lines
2.2 KiB
Python
|
"""Идентичность: acting (X-Acting-Emp-Id) и target (emp_id)."""
|
|||
|
|
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
from typing import Annotated
|
|||
|
|
|
|||
|
|
from fastapi import Header, HTTPException
|
|||
|
|
|
|||
|
|
from app.emp_schema import EMP_TABLE_CANONICAL
|
|||
|
|
from app.main import _column_lookup, _quote_ident, _table_columns, resolve_emp_table
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _biz_error(status: int, code: str, message: str) -> HTTPException:
|
|||
|
|
return HTTPException(status_code=status, detail={"code": code, "message": message})
|
|||
|
|
|
|||
|
|
|
|||
|
|
def parse_acting_emp_id(
|
|||
|
|
x_acting_emp_id: Annotated[str | None, Header(alias="X-Acting-Emp-Id")] = None,
|
|||
|
|
) -> int:
|
|||
|
|
if not x_acting_emp_id or not str(x_acting_emp_id).strip():
|
|||
|
|
raise _biz_error(400, "missing_acting_emp", "Требуется заголовок X-Acting-Emp-Id")
|
|||
|
|
try:
|
|||
|
|
v = int(str(x_acting_emp_id).strip())
|
|||
|
|
except ValueError as e:
|
|||
|
|
raise _biz_error(
|
|||
|
|
400, "invalid_acting_emp", "X-Acting-Emp-Id должен быть целым числом"
|
|||
|
|
) from e
|
|||
|
|
if v <= 0:
|
|||
|
|
raise _biz_error(400, "invalid_acting_emp", "X-Acting-Emp-Id должен быть > 0")
|
|||
|
|
return v
|
|||
|
|
|
|||
|
|
|
|||
|
|
def resolve_target_emp_id(acting_emp_id: int, emp_id: int | None) -> int:
|
|||
|
|
return int(emp_id) if emp_id is not None else acting_emp_id
|
|||
|
|
|
|||
|
|
|
|||
|
|
def ensure_emp_exists(cur, db: str, emp_id: int, *, kind: str = "target") -> None:
|
|||
|
|
table = resolve_emp_table(cur, db)
|
|||
|
|
cols = _table_columns(cur, db, table)
|
|||
|
|
lut = _column_lookup(cols)
|
|||
|
|
prefix = EMP_TABLE_CANONICAL.lower()
|
|||
|
|
id_col = lut.get(f"{prefix}_id") or lut.get("id")
|
|||
|
|
removed_col = lut.get(f"{prefix}_removed") or lut.get("removed")
|
|||
|
|
if not id_col:
|
|||
|
|
raise _biz_error(500, "db_error", "Не найдена колонка id в таблице сотрудников")
|
|||
|
|
|
|||
|
|
cond = f"{_quote_ident(id_col)} = %s"
|
|||
|
|
params: list = [emp_id]
|
|||
|
|
if removed_col:
|
|||
|
|
cond += f" AND {_quote_ident(removed_col)} = 0"
|
|||
|
|
cur.execute(
|
|||
|
|
f"SELECT 1 AS ok FROM {_quote_ident(table)} WHERE {cond} LIMIT 1",
|
|||
|
|
params,
|
|||
|
|
)
|
|||
|
|
if not cur.fetchone():
|
|||
|
|
code = "acting_emp_not_found" if kind == "acting" else "target_emp_not_found"
|
|||
|
|
raise _biz_error(404, code, f"Сотрудник {emp_id} не найден")
|