from flask import Blueprint, request

from Company.Basic.Shareholder.ShareholderImpl import ShareholderImpl
from Utils.ErrorUtil import APIReturnError
from Utils.RouteUtil import RouteParamsCheck

shareholder_route = Blueprint('shareholder', __name__)


@shareholder_route.route('/auto_update', methods=['GET'])
def auto_update_route():
    """新增/更新（自动）"""
    try:
        RouteParamsCheck(req=request.args, params=["cname", "cid"]).required()
        impl = ShareholderImpl()
        result = impl.auto_update(condition=request.args)
        return {"info": result}
    except APIReturnError as e:
        return {"info": e.__str__()}, e.status_code


@shareholder_route.route('/delete', methods=['GET'])
def delete_route():
    """删除"""
    try:
        RouteParamsCheck(req=request.args, params=["cid", "shareholder"]).required()
        impl = ShareholderImpl()
        result = impl.delete(param=request.args)
        return result
    except APIReturnError as e:
        return {"info": e.__str__()}, e.status_code


@shareholder_route.route('/edit', methods=['POST'])
def edit_route():
    """编辑"""
    try:
        RouteParamsCheck(req=request.json, params=["cid", "shareholder", "body"]).required()
        impl = ShareholderImpl()
        result = impl.edit(param=request.json)
        return {"info": result}
    except APIReturnError as e:
        return {"info": e.__str__()}, e.status_code


@shareholder_route.route('/view', methods=['GET'])
def view_route():
    """查看"""
    try:
        RouteParamsCheck(req=request.args, params=["cid", "page_num", "page_size"]).required()
        impl = ShareholderImpl()
        result = impl.view(query=request.args)
        return result
    except APIReturnError as e:
        return {"info": e.__str__()}, e.status_code