from flask import Blueprint, request

from Modules.AdminUser.UserAuthUtils import verify_token
from Modules.Company.CompanyImpl import BasicBusinessInfoImpl
from Utils.ErrorUtil import AttrCheckError

company_route = Blueprint('company', __name__)


@company_route.route('/create', methods=['POST'])
@verify_token
def create_route(**kwargs):
    """新增公司"""
    try:
        company_name = request.form['company_name']
        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