odoo_wechat_connector/models/ewi_wechat_config.py

# -*- coding: utf-8 -*-
"""
@Time    : 2022/12/22 08:44
@Author  : Jason Zou
@Email   : zou.jason@qq.com
"""
from odoo import models, fields, exceptions
import os
import json
import logging
import requests
_logger = logging.getLogger(__name__)
os.environ['NLS_LANG'] = 'SIMPLIFIED CHINESE_CHINA.UTF8'
headers = {'content-type': 'application/json'}


class EwiWechatConfig(models.Model):
    _name = 'ewi.wechat.config'
    _description = '接口认证信息'

    # 定义接口基本信息
    name = fields.Char(string='Auth接口名称', default='获取企业微信接口调用token', help='API名称')
    corp_id = fields.Char(string='企业ID',
                          default='ww5a789309ca91ec7e',
                          help='路径：我的企业-企业信息-企业ID')

    # 定义“审批”应用对接信息
    sp_AgentId = fields.Char(string='审批AgentId',
                             default='3010040',
                             help='路径：审批-AgentId')
    sp_Secret = fields.Char(string='审批Secret',
                            default='ub3ELV5X9d1JGkwmC9If_qr0szG78ogYBnGKZnCQPEg',
                            help='路径：审批-Secret-查看')
    sp_URL = fields.Char(string='审批URL',
                         default='http://bpm-test.dingyang.com:8069/corp_handler',
                         help='路径：审批-接收消息服务器配置-URL')
    sp_access_token = fields.Char(string='审批Token',
                                  default='TKkL0wVvklvIXD',
                                  help='路径：审批-接收消息服务器配置-Token')
    sp_EncodingAESKey = fields.Char(string='审批EncodingAESKey',
                                    default='vzpWA1cL9D8vPwgFOqTjFaysLhNbV36kWcXODL6knxf',
                                    help='路径：审批-接收消息服务器配置-EncodingAESKey')



    def gen_approval_access_token(self):
        """授权信息，获取企微“审批”应用Access Token"""
        access_obj = self.env['ewi.wechat.config']
        access_record = access_obj.search([('name', '=', '获取企业微信接口调用token')])
        corp_id = access_record.corp_id
        approval_secret = access_record.approval_secret
        token_url = f"https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid={corp_id}&corpsecret={approval_secret}"
        try:
            ret = requests.get(token_url, headers=headers)
            ret.raise_for_status()
            result = ret.json()
            if result.get('errcode') == 0:
                access_token = result.get('access_token')
                return access_token
            else:
                _logger.error(f"获取企微“审批”应用Access Token失败: {result.get('errmsg')}")
                return None
        except requests.RequestException as e:
            _logger.error(f"请求获取企微“审批”应用Access Token时出错: {str(e)}")
            return None