# @description: 
# @author: licanglong
# @date: 2025/11/20 14:22

from flask import request, jsonify

from app.blueprints.invoice import invoice_bp
from app.db.dbsession import SessionLocal
from app.models.Result import SysResult
from app.repositories.invoice_repo import InvoiceRepository


@invoice_bp.route('/list/by_tax_id', methods=['POST'])
def list_by_taxid():
    data = request.json or {}
    session = SessionLocal()
    if not data or not data.get('tax_id', None):
        return jsonify(SysResult.fail(msg="参数错误"))
    try:
        repo = InvoiceRepository(session)
        invoices = repo.list_by_taxid(data['tax_id'])
        return jsonify(SysResult.success(data=invoices.dict()['invoice_purchase_details']))
    finally:
        session.close()


@invoice_bp.route('/list/by_params', methods=['POST'])
def list_by_params():
    data = request.json or {}
    session = SessionLocal()
    try:
        repo = InvoiceRepository(session)
        invoices = repo.list_by_params(data.get("tax_id", None), data.get("hwmc", None))
        return jsonify(SysResult.success(data=invoices.dict()['invoice_purchase_details']))
    finally:
        session.close()