import json

from flask import Blueprint, request

from common import verify_token
from tyc import save_full_data, company_baseinfo, corporate_shareholders

tyc = Blueprint('tyc', __name__)


# 保存天眼查全量数据
@tyc.route('/drag_data', methods=['POST'])
@verify_token
def upload_tyc_data():
    name = json.loads(request.data)['企业名称']
    res = save_full_data(name)
    if res == '保存成功':
        return {"info": res}, 200
    else:
        return {"info": res}, 210


@tyc.route('/basic_info', methods=['POST'])
@verify_token
def upload_basic_data():
    """保存企业基本信息"""
    name = json.loads(request.data)['企业名称']
    res = company_baseinfo(name)
    if res == '保存成功':
        return {"info": res}, 200
    else:
        return {"info": res}, 210


@tyc.route('/shareholders_info', methods=['POST'])
@verify_token
def upload_shareholders_data():
    """保存企业股东"""
    name = json.loads(request.data)['企业名称']
    res = corporate_shareholders(name)
    if res == '保存成功':
        return {"info": res}, 200
    else:
        return {"info": res}, 210