"""数据文件各环节共用的异常、上限和警告文案的写法"""

from collections.abc import Sequence

# 每张表的数据行数上限
MAX_TABLE_ROWS = 1_000_000

UNSUPPORTED_FORMAT = "不支持的文件格式，支持 CSV、TSV、Excel、JSON、JSONL、Parquet"


class DataFileError(Exception):
    """文件无法接收或解析；status 是返回给调用方的 HTTP 状态码，message 可以直接展示

    解析在子进程里执行，异常要能跨进程传递，所以参数原样放进 args
    """

    def __init__(self, status: int, message: str):
        super().__init__(status, message)
        self.status = status
        self.message = message

    def __str__(self) -> str:
        return self.message


def where(sheet: str | None) -> str:
    """警告和错误信息的位置前缀：Excel 写 sheet 名，其他格式不写"""
    return f"{sheet}：" if sheet else ""


def brief(items: Sequence[str], sep: str = "、", limit: int = 5) -> str:
    """警告里列举的内容最多写 limit 个，多出的用“等”概括"""
    return sep.join(items[:limit]) + ("等" if len(items) > limit else "")


def rows_exceeded(sheet: str | None) -> DataFileError:
    return DataFileError(
        422, f"{where(sheet)}超过 {MAX_TABLE_ROWS // 10_000} 万行，请拆分后上传"
    )