tfse-admin-api-v0.2/Test/TestImpl.py

import json
import requests
import pandas as pd
import numpy as np

from DBHelper.MongoHelper import MongoHelper
from Test.TestObj import CompanyUser, RatingRecord


class CompanyAuthImpl(CompanyUser):
    """企业相关类"""
    db = MongoHelper("tfse_v0.21")

    def company_register(self):
        """
        新注册企业
        1.调用发送注册邮箱，获取验证码
        2.使用邮箱、密码、验证码进行注册
        3.使用认证接口进行认证、初始化企业数据
        """

        def send_mail():
            """获取验证码"""
            url = "https://api.fecribd.com/app/test/verify_email"
            secret = 'EZgo9ykxrYuBMYnYmmKIh'
            headers = {'secret': secret, "Content-Type": "application/json;charset=UTF-8"}
            param = {"email": self.dict_to_save(columns=['邮箱'])['邮箱']}
            insert = json.dumps(param)
            res = requests.post(url, headers=headers, data=insert)
            result = json.loads(res.text)
            if result['info'] == '邮件已发送':
                code = self.db.find_single_data(
                    '应用端',
                    '邮箱验证码记录',
                    {'邮箱': self.email},
                    ['验证码']
                )
                return code['验证码'] if code else False
            else:
                return '邮箱已被注册'

        def register():
            """注册"""
            email_code = send_mail()
            if email_code and email_code != '邮箱已被注册':
                data = dict()
                data['email'] = self.dict_to_save(columns=['邮箱'])['邮箱']
                data['code'] = email_code
                url = "https://api.fecribd.com/app/test/register"
                secret = 'EZgo9ykxrYuBMYnYmmKIh'
                headers = {'secret': secret, "Content-Type": "application/json;charset=UTF-8"}
                insert = json.dumps(data)
                res = requests.post(url, headers=headers, data=insert)
                result = json.loads(res.text)
                if result['info'] == '注册成功':
                    cid = self.db.find_single_data(
                        '应用端',
                        '企业用户',
                        {'邮箱': self.email},
                        ['企业ID']
                    )
                    return cid['企业ID'] if cid else False
            elif email_code == '邮箱已被注册':
                return {"info": "邮箱已被注册"}, 210
            else:
                return {"info": "验证码发送失败"}, 210

        return register()

    def enterprise_certification(self):
        """企业认证（测试用）"""
        insert_data = self.dict_to_save(columns=['认证信息'])
        self.db.update_single_data(
            "应用端",
            "企业用户",
            {"企业ID": self.cid},
            insert_data
        )

        def certification_interface():
            """调用应用端认证接口"""
            url = "https://api.fecribd.com/app/test/company_verify?cid={}".format(self.cid)
            secret = 'EZgo9ykxrYuBMYnYmmKIh'
            headers = {'secret': secret, "Content-Type": "application/json;charset=UTF-8"}
            res = requests.get(url, headers=headers)
            return json.loads(res.text)

        return certification_interface()

    def change_verified(self):
        """企业认证状态修改"""
        verify_info = self.VerifyInfo()
        verify_info.name = None
        verify_info.code = None
        verify_info.legal_person = None
        insert_data = verify_info.dict_to_save()
        self.db.update_single_data(
            "应用端",
            "企业用户",
            {"企业ID": self.cid},
            {"已认证": "否", "认证信息": insert_data}
        )


class CompanyIndexImpl(RatingRecord):
    """企业主页"""
    db = MongoHelper("tfse_v0.21")

    def generate_homepage(self):
        """生成主页数据"""
        self.cid = self.db.find_single_data(
            '企业数据',
            '评价记录',
            {"评价ID": self.rid},
            ['企业ID']
        )['企业ID']
        data = self.dict_to_save(columns=['企业ID', '评价项目'])

        def generate_interface():
            """调用打分接口"""
            url = "https://api.fecribd.com/app/test/general_rating"
            secret = 'EZgo9ykxrYuBMYnYmmKIh'
            headers = {'secret': secret, "Content-Type": "application/json;charset=UTF-8"}
            insert = json.dumps(data)
            res = requests.post(url, headers=headers, data=insert)
            return json.loads(res.text)

        return generate_interface()


class DeleteCompanyDataImpl(RatingRecord):
    """删除企业相关数据"""
    db = MongoHelper("tfse_v0.21")

    def form_rid_delete(self, record, data):
        """根据rid删除相关记录"""

        def delete_all_records():
            """根据rid删除评价数据库相匹配数据"""
            if '财指结果' in data:
                database = '综信评价数据'
                database_01 = '综信报告'
                database_02 = '综信证书'
            else:
                database = 'ESG评价数据'
                database_01 = 'ESG报告'
                database_02 = 'ESG证书'
            for name in data:
                if name == '财指结果':
                    for i in range(3):
                        self.db.delete_single_data(database, name, {"企业ID": self.rid})
                else:
                    self.db.delete_single_data(database, name, {"企业ID": self.rid})
            # 文件相关
            report_fid = records['报告fid']
            certificate_fid = records['证书fid']
            self.db.delete_file('文件', database_01, report_fid)
            self.db.delete_file('文件', database_02, certificate_fid)
            return '删除成功'

        def delete_part_records():
            """根据rid删除评价数据库相匹配数据"""
            change = {'进行状态': '进行', '评价结果': None, '开始填报时间': None, '提交填报时间': None, '评价完成时间': None,
                      '评价完成日期': None, '报告fid': None, '证书fid': None}
            self.db.update_single_data(
                '企业数据',
                '评价记录',
                {'评价ID': self.rid},
                change
            )
            return '状态更改成功'

        records = self.db.find_single_data(
            '企业数据',
            '评价记录',
            {"评价ID": self.rid},
            ['进行状态', '报告fid', '证书fid']
        )
        if records:
            if records['进行状态'] == '完成':
                if record == '删除记录':
                    info = delete_all_records()
                else:
                    info = delete_all_records()
                    if info == '删除成功':
                        info = delete_part_records()
                        return info
                return info
            else:
                return '不能删除正在进行的评价记录'

        else:
            return '未查询到此rid相关记录'

    def form_cid_delete(self, data):
        """根据cid删除相关记录"""
        for name in data:
            self.db.delete_single_data('企业数据', name, {"企业ID": self.cid})
        return '企业相关数据删除成功'


class GenerateFillInData(RatingRecord):
    """生成填报数据"""
    db = MongoHelper('tfse_v0.21')

    def insert_esg_data(self):
        """自动生成esg问卷填报数据"""
        def search_data():
            """根据rid查询填报数据"""
            fill = self.db.find_single_data(
                'ESG评价数据',
                '填报数据',
                {'评价ID': self.rid},
                []
            )
            return fill

        data = search_data()

        def e_script():
            """环境问卷"""
            e_data = data['环境问卷']['近三年公司数据']
            name = list()
            for k in e_data[0].keys():
                if k != '年份':
                    name.append(k)
            year = [e_data[0]['年份'], e_data[1]['年份'], e_data[2]['年份']]
            df = pd.DataFrame(np.random.randint(200, 500, size=(3, 13)), columns=name)
            df.insert(loc=0, column='年份', value=year)
            num = df.shape[0]
            result = list()
            for n in range(num):
                result.append(df.iloc[n].to_dict())
            return result

        def s_script():
            """社会问卷"""
            s_data = data['社会问卷']['近三年公司数据']
            year = [s_data[0]['年份'], s_data[1]['年份'], s_data[2]['年份']]
            df = pd.DataFrame(year, columns=['年份'])
            df['员工总数（年底）'] = np.random.randint(150, 250, size=3)
            df['当年离职人数'] = np.random.randint(10, 25, size=3)
            df['人均薪酬水平（元/月）'] = np.random.randint(7500, 8500, size=3)
            df['劳动合同要求工作时长（每周小时数）'] = np.random.randint(40, 50, size=3)
            df['劳资纠纷次数'] = np.random.randint(0, 5, size=3)
            df['安全事故发生次数'] = [0, 0, 0]
            df['组织培训次数'] = np.random.randint(5, 15, size=3)
            df['公司从前3大供应商拿货占全部供应商比例（%）'] = [round(i * 100, 2) for i in np.random.rand(3)]
            df['公司前3大客户销售额占全部销售比例（%）'] = [round(i * 100, 2) for i in np.random.rand(3)]
            df['返修、退回、投诉产品对应销售额占全部销售比例（%）'] = [round(i * 100, 2) for i in np.random.rand(3)]
            df['扶贫+捐赠规模（万元）'] = np.random.randint(100, 500, size=3)
            num = df.shape[0]
            result = list()
            for n in range(num):
                result.append(df.iloc[n].to_dict())
            return result

        def save_data():
            """保存数据"""
            self.db.update_single_data(
                'ESG评价数据',
                '填报数据',
                {'评价ID': self.rid},
                {
                    '环境问卷.近三年公司数据': e_script(),
                    '社会问卷.近三年公司数据': s_script()
                }
            )
            return '生成成功'

        return save_data()