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

152 lines
4.1 KiB
Python

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'])
@verify_token
def guarantee_info_route(**kwargs):
"""担保信息"""
try:
RouteParamsCheck(request.args, ["cid"]).required()
cid = request.args["cid"]
impl = CompanyManageImpl()
result = impl.guarantee_info(cid=cid)
return {"info": "担保信息", "result": result}, 200
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