guarantee-admin-api-v0.2/Modules/Company/CompanyRoutes.py

from flask import Blueprint, request

from Modules.AdminUser.UserAuthUtils import verify_token
from Modules.Company.CompanyImpl import BasicBusinessInfoImpl
from Modules.Company.CompanyManageImpl import CompanyManageImpl
from Utils.ErrorUtil import AttrCheckError, APIReturnError
from Utils.RouteUtil import RouteParamsCheck

company_route = Blueprint('company', __name__)


@company_route.route('/create', methods=['POST'])
@verify_token
def create_route(**kwargs):
    """新增公司"""
    try:
        file = request.files['file']
        impl = BasicBusinessInfoImpl()
        impl.parse_excel_and_create_company(file)
        return "success"
    except AssertionError as e:
        return {"info": e.__str__()}, 202
    except AttrCheckError as e:
        return {"info": e.__str__()}, 202


@company_route.route('/search', methods=['GET'])
@verify_token
def search_route(**kwargs):
    """公司搜索"""
    try:
        RouteParamsCheck(request.args, ["company_name", "page_size", "page_no"]).required()

        company_name = request.args["company_name"]
        page_size = request.args["page_size"]
        page_no = request.args["page_no"]

        impl = CompanyManageImpl()

        result = impl.search(
            company_name=company_name,
            page_size=page_size,
            page_no=page_no
        )

        return {"info": "公司搜索", "result": result}, 200

    except APIReturnError as e:
        return {"info": e.__str__()}, e.status_code


@company_route.route('/basic_data', methods=['GET'])
@verify_token
def basic_data_route(**kwargs):
    """基本信息"""

    try:
        RouteParamsCheck(request.args, ["cid"]).required()

        cid = request.args["cid"]

        impl = CompanyManageImpl()

        result = impl.basic_data(cid=cid)

        return {"info": "基本信息", "result": result}, 200

    except APIReturnError as e:
        return {"info": e.__str__()}, e.status_code


@company_route.route('/financial_data', methods=['GET'])
@verify_token
def financial_data_route(**kwargs):
    """财务数据"""

    try:
        RouteParamsCheck(request.args, ["cid"]).required()

        cid = request.args["cid"]

        impl = CompanyManageImpl()

        result = impl.financial_data(cid=cid)

        return {"info": "财务数据", "result": result}, 200

    except APIReturnError as e:
        return {"info": e.__str__()}, e.status_code


@company_route.route('/guarantee_info', methods=['GET', 'POST'])
@verify_token
def guarantee_info_route(**kwargs):
    """担保信息"""

    try:

        if request.method == "GET":
            RouteParamsCheck(request.args, ["cid"]).required()
            cid = request.args["cid"]
            impl = CompanyManageImpl()
            result = impl.guarantee_info(cid=cid)
            return {"info": "担保信息", "result": result}, 200

        if request.method == "POST":

            RouteParamsCheck(
                request.json,
                [
                    "项目名称", "被担保人", "担保类型", "担保方式", "担保余额(万元)", "解除金额(万元)",
                    "责任担保余额(万元)", "担保开始日期", "担保结束日期", "省份", "地级市", "行业"
                ]
            ).required()

    except APIReturnError as e:
        return {"info": e.__str__()}, e.status_code


@company_route.route('/repay', methods=['GET', 'POST'])
@verify_token
def repay_route(**kwargs):
    """代偿信息"""

    try:

        if request.method == "GET":
            RouteParamsCheck(request.args, ["gid"]).required()
            gid = request.args["gid"]
            impl = CompanyManageImpl()
            result = impl.get_repay(gid=gid)
            return {"info": "担保信息", "result": result}, 200

        if request.method == "POST":
            RouteParamsCheck(
                request.json,
                [
                    "project_name", "repay_amount", "repay_recycle_amount", "repay_date"
                ]
            ).required()

            project_name = request.json["project_name"]
            repay_amount = request.json["repay_amount"]
            repay_recycle_amount = request.json["repay_recycle_amount"]
            repay_date = request.json["repay_date"]

            impl = CompanyManageImpl()
            impl.edit_repay(
                project_name=project_name,
                repay_amount=repay_amount,
                repay_recycle_amount=repay_recycle_amount,
                repay_date=repay_date,
            )
            return {"info": "担保信息", "result": "操作成功"}, 200

    except APIReturnError as e:
        return {"info": e.__str__()}, e.status_code


@company_route.route('/rating_records', methods=['GET'])
@verify_token
def rating_records_route(**kwargs):
    """评级记录"""

    try:
        if request.method == "GET":
            result = CompanyManageImpl.get_rating_records()
            return {"info": "评级记录", "result": result}

    except APIReturnError as e:
        return {"info": e.__str__()}, e.status_code