from typing import Any, Dict
from core.document_processor.templates.base_template import DocumentTemplate


class SaleTaskTemplate(DocumentTemplate):
    """销售订单模板"""

    @property
    def template_name(self) -> str:
        return "saletask"

    @property
    def description(self) -> str:
        return "销售订单识别模板"

    def get_hardcoded_guidance(self) -> Dict[str, Any]:
        """硬编码的字段指导信息（项目初期为空，验收后转移内容到这里）"""
        return {
            "field_guidance": {
                # 项目初期保持为空，验收后将template_config.json的内容转移到这里
            },
            "additional_rules": "",  # 硬编码的额外规则
        }

    def output_schema(self) -> Dict[str, Any]:
        return {
            "cusname": "客户名称",
            "taskdate": "订货日期",
            "requiredate": "交货日期",
            "banktype": "结算方式",
            "relcode": "相关号码",
            "otheramt": "优惠金额",
            "damt": "订金",
            "cus_tele": "客户联系电话",
            "rel_rep": "客户联系人",
            "dscrp": "单据备注描述",
            "freight": "货运部",
            "items": [
                {
                    "mtrlname": "货物\产品名称",
                    "unit": "计量单位(例如张,台,件等)",
                    "saleqty": "数量",
                    "enprice": "单价",
                    "rebate": "折扣",
                }
            ],
        }

    def extraction_rules(self) -> str:
        return """
        """