Server-py/3rdparty/xapi/mccbts/api.py

# update: 2024.7.14
"""
中冶宝钢智慧研究院接口
"""
from urllib import parse
import requests
import time
import traceback

import sys
import importlib

# sys.path.append(r'C:\mccbts-ar-py\module-method-python')  # for 10.10.61.22
sys.path.append(r'C:\mccbts-ar-py\module-method-python')  # for 10.10.61.22
# sys.path.append(r'D:\mccbts-ar-py\module-method-python')  # for 192.168.103.34
# methods = importlib.import_module(f"libraries.base_original")
methods = importlib.import_module(f"xlib")


class API(object):

    # def __init__(self, username='System001', password='Swdx@143',
    def __init__(self, username='', password=''):
        """
        接口地址: https://eaiops.mccbts.com.cn/cigpf-gateway/cigpf-auth/login
        接口说明: 用户认证接口，mcctest/Mcc@1234

        接口地址: http://ycpx04.mccbts.com.cn/cnki-gw/expert/expert/getExpertList
        接口说明：获取专家列表

        接口地址: http://10.8.0.151:8000/datachange/release/interface
        接口说明：该接口可以获取用户数据、部门数据
        """

        # --- define ---
        self.cigpf_api_service = f"https://eaiops.mccbts.com.cn"
        self.cnki_api_service = f"http://ycpx04.mccbts.com.cn"
        self.datachange_api_service = f"http://10.8.0.151:8000"

        self.LIST_USER_TOKEN = 'FA460AD97660475BBD8E1CF31BC7A49B'
        self.LIST_DEPT_TOKEN = '4412B9724D254E38B57014CF99EB0312'

    def get_user_info(self, username, password):
        """
        获取用户信息
        return {
            'success': True,
            'message': '操作成功！',
            'code': 200,
            'result': {
                'id': '1399259233832144898',
                'username': 'mcctest',
                'realname': '测试账号',
                'avatar': '/cigpfupload/05a5b1ab84a446c4a2e126da89b0c0af.png',
                'birthday': None,
                'sex': None,
                'thirdId': '',
                'thirdType': 'qt',
                'email': None,
                'phone': '18210635895',
                'orgCode': '1',
                'orgCodeTxt': None,
                'status': 1,
                'delFlag': 0,
                'workNo': '99999',
                'post': None,
                'telephone': None,
                'createBy': '',
                'createTime': None,
                'updateBy': '张宇',
                'updateTime': '2023-11-07 06:00:09',
                'activitiSync': 1,
                'userIdentity': 2,
                'departIds': '1000051C',
                'relTenantIds': None,
                'clientId': None,
                'userType': '2'
            },
            'timestamp': 1699666081555
        }
        """
        url = f"{self.cigpf_api_service}/cigpf-gateway/cigpf-auth/login"
        # data = {
        #     'userName': 'mcctest',  # 用户
        #     'passWord': 'Mcc@1234',  # 密码
        # }
        data = {
            'userName': username,  # 用户
            'passWord': password,  # 密码
        }
        # print(f"API.get_user_info:data: {data}")
        print('API.get_user_info.url:', url)
        response = requests.post(url, json=data)

        if response.status_code > 300:
            print('API.get_user_info.result:', response.status_code)
            print('API.get_user_info.detail:', response.text)
            return {}
        else:
            print('API.get_user_info.result:', response.status_code)
            return response.json()

    def get_expert_type_tree_list(self, q=''):
        """
        获取专家结构树
        """
        url = f"https://ycpx04.mccbts.com.cn/cnki-gw/expert/expertType/getExpertTypeTreeList"
        data = {
            'keyword': q
        }
        response = requests.post(url, json=data)
        # print('API.test.url:', url)
        # print('API.test.data:', data)
        # print('API.test.code:', response.status_code)
        # print('API.test.text:', response.text)
        return response.json()

    def get_expert_list(self, page=1, rows=10):
        """
        获取专家列表
        return {
            'msg': '操作成功',
            'code': 200,
            'data': {
                'total': 320,
                'list': [
                    {
                        'id': 407880669122629,
                        'expertId': 'b46f7f7e-2883-4281-ba92-0c72fb30a668',
                        'expertCode': '40886',
                        'expertName': '冯浩川',
                        'expertLogo': '',
                        'sex': 1,
                        'birthday': '1974-08-12 00:00:00',
                        'education': None,
                        'degree': None,
                        'colleges': '',
                        'politicalOutlook': '',
                        'academicTitle': '',
                        'expertTypeId': 'a0846382-dc33-44f5-990d-ea042ab90cf4',
                        'expertType': '工程机械组',
                        'workUnitOrCurrentPosition': '',
                        'professionalAndTechnicalFields': '',
                        'mainSocialPositions': '',
                        'personalHonors': '',
                        'project': [],
                        'brief': '',
                        'createTime': '2023-04-17 15:30:43',
                        'createUserId': '6867b1b3-024f-4065-b991-3ebd48df7b27',
                        'type': '',
                        'companyId': '45b9801f-d2ff-4886-8076-7f7d8b403e37',
                        'company': '协力生产分公司',
                        'departmentId': '301e6551-dfd6-4e3e-b7f5-be36e14f84a1',
                        'department': '协力_机械维修中心_管理组'
                    }
                ]
            }
        }
        """
        url = f"{self.cnki_api_service}/cnki-gw/expert/expert/getExpertList"
        data = {
            'pageDto': {
                'page': page,
                'rows': rows
            }
        }
        # print(f"API.get_user_info:data: {data}")
        print('API.get_expert_list.url:', url)
        response = requests.post(url, json=data)

        # --- check ---
        if response.status_code > 300:
            print('API.get_expert_list.result:', response.status_code)
            print('API.get_expert_list.detail:', response.text)
            return []

        # --- check ---
        data = response.json()
        if data.get('code') != 200:
            return []

        # --- check ---
        if not data:
            return []

        # print(data.get('data').get('total'))
        return data.get('data').get('list')

    def get_staff_list(self):
        """
        获取员工列表
        return [
            {
                'STATUS': '1',
                'ORG_CODE': '1603605',
                'UPDATE_BY': 'HR',
                'CREATE_BY': 'HR',
                'DEPART_NAME': '山西_天车二区_硅钢班',
                'DEL_FLAG': '0',
                'CREATE_TIME': 1659117651000,
                'UPDATE_TIME': 1659117651000,
                'ID': '1001G1AR',
                'DEPART_NAME_ABBR': '山西_天车二区_硅钢班',
                'C1': '1',
                'PARENT_ID': '1001G1AM'
            }
        ]
        """
        url = f"{self.datachange_api_service}/datachange/release/interface"
        headers = {'Chinasofti-Access-Token': self.LIST_USER_TOKEN}
        print('API.get_staff_list.url:', url)
        print('API.get_staff_list.headers:', headers)
        response = requests.get(url, headers=headers)

        # --- check ---
        if response.status_code > 300:
            print('API.get_staff_list.result:', response.status_code)
            print('API.get_staff_list.detail:', response.text)
            return '请求请求失败', f"code: {response.status_code}, response: {response.text}"

        # --- check ---
        data = response.json()
        if data.get('code') != 0:
            # print('API.get_staff_list.detail:', response.text)
            return '接口返回异常', f"请求地址: {url}, 请求方式: GET, 请求参数: {headers}, 返回结果: {response.text}"

        return data.get('data')

    def get_department_list(self):
        """
        获取部门列表
        return [
            {
                'ORG_ID': '10012B4J',
                'REALNAME': '张天勤',
                'POST': '电焊工',
                'PHONE': '15921009690',
                'SEX': 1,
                'UPDATE_BY': 'HR',
                'DEL_FLAG': 0,
                'UPDATE_TIME': 1661492148000,
                'WORK_NO2': '983260',
                'STATUS': '1',
                'PKID': 'C1924F31EE850472E050080A5C00BCBA',
                'ORG_CODE': '153420410',
                'CREATE_BY': 'HR',
                'CREATE_TIME': 1620205014000,
                'ID': '10000RBZ',
                'WORK_NO': '983260',
                'BIRTHDAY': '1967-02-09'
            }
        ]
        """
        url = f"{self.datachange_api_service}/datachange/release/interface"
        headers = {'Chinasofti-Access-Token': self.LIST_DEPT_TOKEN}
        print('API.get_department_list.url:', url)
        print('API.get_department_list.headers:', headers)
        response = requests.get(url, headers=headers)
        print('API.get_department_list.status_code:', response.status_code)
        print('API.get_department_list.text:', response.text)

        # --- check ---
        if response.status_code > 300:
            # print('API.get_department_list.result:', response.status_code)
            # print('API.get_department_list.detail:', response.text)
            return {'code': -1, 'data': response.text}

        # --- check ---
        data = response.json()
        if data.get('code') != 0:
            # print('API.get_department_list.code:', data.get('code'))
            # print('API.get_department_list.data:', response.text)
            return {'code': -1, 'data': response.text}

        return {'code': 0, 'data': data.get('data')}

    def test(self):
        """
        """
        url = f"http://ycpx04.mccbts.com.cn/cnki-gw/expert/expertType/getExpertTypeList"
        data = {
            'keyword': 'xxx'
        }
        response = requests.post(url, json=data)
        print('API.test.url:', url)
        print('API.test.data:', data)
        print('API.test.code:', response.status_code)
        print('API.test.text:', response.text)
        # res = res.json()
        # data = {
        #     'count': len(res['data']),
        #     'data': res['data']
        # }


if __name__ == '__main__':
    # --- init ---
    api = API()

    # --- test ---
    items = api.get_staff_list()
    # items = api.get_department_list()
    print(items)

    # --- test ---
    # items = api.get_expert_list(page=1, rows=100000000)
    # for item in items:
    #     if 'c0dbc699-1640-43c2-8459-ffadbaa2d966' not in item.get('expertTypeId'):
    #         continue
    #     print(item)

    # --- test ---
    # items = api.get_expert_type_tree_list()
    # print(items)

    # --- test --- 获取一级公司
    # d1 = list()
    # items = api.get_department_list().get('data')
    # for item in items:
    #     # if item.get('ORG_CODE') == '191':
    #     #     print(item)
    #     # if '总部' in item.get('DEPART_NAME'):
    #     # if '总部' in item.get('DEPART_NAME'):
    #     #     print(item)
    #     if len(item.get('ORG_CODE')) == 3 and item.get('ORG_CODE')[0] == '1':
    #     # if item.get('ORG_CODE') in ['3', '4', '6']:
    #         # print(item)
    #         # d1.append(item.get('DEPART_NAME'))
    #         print(item.get('ORG_CODE'), item.get('DEPART_NAME'))
    # print(d1, len(d1))

    # --- test ---
    # items = api.get_expert_type_tree_list().get('data')
    # print(items[0]['children'])
    # for i in items[0]['children']:
    #     print(len(i.get('children')), i.get('count'))
    #     if i.get('children'):
    #         for j in i.get('children'):
    #             print(len(j.get('children')), j.get('count'))