duodoo-ai
/
odoo_wechat_connector
Observar 1
Favorito 0
Wiki 捐赠
Odoo与企业微信对接（Odoo18对接企业微信）
Branch: master
odoo_wechat_connector/models/ewi_interface.py

332 linhas
17KB
Original Link permanente Anotar Histórico 

  1. # -*- coding: utf-8 -*-

  2. """

  3. @Time    : 2022/12/23 08:44

  4. @Author  : Jason Zou

  5. @Email   : zou.jason@qq.com

  6. @ 一级部门/公司跳过不处理，需手动维护ODOO与企业微信一级部门名称一致

  7. """

  8. from odoo import models, fields, exceptions

  9. import os

  10. import json

  11. import logging

  12. import requests

  13. _logger = logging.getLogger(__name__)

  14. os.environ['NLS_LANG'] = 'SIMPLIFIED CHINESE_CHINA.UTF8'

  15. headers = {'content-type': 'application/json'}

  16. 

  17. 

  18. class EWIInterface(models.Model):

  19.     _name = 'ewi.interface'

  20.     _description = '企业微信接口'

  21. 

  22.     name = fields.Char(string='接口名称', required=True, tracking=True)

  23.     description = fields.Char(string='接口说明', tracking=True)

  24.     AgentId = fields.Char(string='应用AgentId', help='本项在应用对接的时候应填入')

  25.     Secret = fields.Char(string='应用凭证密钥Secret', help='应用的凭证密钥，注意应用需要是启用状态，获取方式参考：术语说明-secret')

  26.     url = fields.Text(string='TokenUrl')

  27.     access_token = fields.Text(string='Token', help='获取到的凭证，最长为512字节，通过corp_id，Secret请求返回')

  28.     errcode = fields.Char(string='errcode', help='出错返回码，为0表示成功，非0表示调用失败')

  29.     errmsg = fields.Char(string='errmsg', help='返回码提示语')

  30.     expires_in = fields.Char(string='expires_in', help='凭证的有效时间（秒）')

  31. 

  32.     def btn_execute(self):

  33.         """

  34.         按钮执行函数，可用于触发一系列操作

  35.         """

  36.         if self.name == '通讯录同步授权':

  37.             self.gen_contacts_access_token()     # 获得连接Token，通讯录授权

  38.         if self.name == 'R2':

  39.             self.send_message()     # 发送各种应用消息

  40.         # # 示例：调用部门相关接口

  41.         if self.name == '创建部门':

  42.             self.new_department()

  43.         # self.update_department()

  44.         # self.delete_department()

  45.         # # 调用人员相关接口

  46.         # self.gen_employee_userid_list()

  47.         if self.name == '创建成员':

  48.             self.new_employee()

  49. 

  50.     def gen_contacts_access_token(self):

  51.         """授权信息，获取企微通讯录Access Token"""

  52.         access_obj = self.env['ewi.wechat.config']

  53.         access_record = access_obj.search([('name', '=', '企业微信接口')])

  54.         corp_id = access_record.corp_id

  55.         corp_secret = self.Secret

  56.         if not corp_id or not corp_secret:

  57.             _logger.info(f"通讯录授权信息{corp_id}----{corp_secret}为空，请填入！")

  58.             return

  59.         token_url = f"https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid={corp_id}&corpsecret={corp_secret}"

  60.         _logger.info(f"通讯录访问 Token_url -----  {token_url}")

  61.         for line in self:

  62.             try:

  63.                 ret = requests.get(token_url, headers=headers)

  64.                 ret.raise_for_status()

  65.                 result = ret.json()

  66.                 if result.get('errcode') == 0:

  67.                     line.write({'access_token': result['access_token'],

  68.                                          'errcode': result['errcode'],

  69.                                          'errmsg': result['errmsg'],

  70.                                          'expires_in': result['expires_in'],

  71.                                          })

  72.                     return result['access_token']

  73.                 else:

  74.                     _logger.error(f"获取企微通讯录Access Token失败: {result.get('errmsg')}")

  75.                     return None

  76.             except requests.RequestException as e:

  77.                 _logger.error(f"请求获取企微通讯录Access Token时出错: {str(e)}")

  78.                 return None

  79. 

  80.     def gen_application_access_token(self):

  81.         """授权信息，获取企微应用授权Access Token"""

  82.         access_obj = self.env['ewi.wechat.config']

  83.         access_record = access_obj.search([('name', '=', '企业微信接口')])

  84.         corp_id = access_record.corp_id

  85.         corp_secret = self.Secret       # 通过每个应用动态获得

  86.         _logger.info(f"应用授权信息{corp_id} --- {corp_secret}")

  87.         if not corp_id or not corp_secret:

  88.             _logger.info(f"应用授权信息{corp_secret}为空，请填入！")

  89.             return

  90.         token_url = f"https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid={corp_id}&corpsecret={corp_secret}"

  91.         for line in self:

  92.             try:

  93.                 ret = requests.get(token_url, headers=headers)

  94.                 ret.raise_for_status()

  95.                 result = ret.json()

  96.                 if result.get('errcode') == 0:

  97.                     line.write({'access_token': result['access_token'],

  98.                                          'errcode': result['errcode'],

  99.                                          'errmsg': result['errmsg'],

  100.                                          'expires_in': result['expires_in'],

  101.                                          })

  102.                     return result['access_token']

  103.                 else:

  104.                     _logger.error(f"获取企业应用Access Token失败: {result.get('errmsg')}")

  105.                     return None

  106.             except requests.RequestException as e:

  107.                 _logger.error(f"获取企业应用Access Token时出错: {str(e)}")

  108.                 return None

  109. 

  110.     def send_message(self):

  111.         """发送各种应用消息"""

  112.         access_token = self.gen_application_access_token()

  113.         if not access_token:

  114.             _logger.info(f"创建应用消息-----access_token------凭证为空 {access_token}")

  115.             return

  116.         token_url = f"https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token={access_token}"

  117.         data = {

  118.            "touser" : "18066043008",

  119.            # "toparty" : "PartyID1|PartyID2",

  120.            # "totag" : "TagID1 | TagID2",

  121.            "msgtype" : "text",

  122.            "agentid" : self.AgentId,

  123.            "text" : {

  124.                "content" : "你的快递已到，请携带工卡前往邮件中心领取。聪明避开排队。"

  125.            },

  126.            "safe":0,

  127.            "enable_id_trans": 0,

  128.            "enable_duplicate_check": 0,

  129.            "duplicate_check_interval": 1800

  130.         }

  131.         ret = requests.post(token_url, data=json.dumps(data), headers=headers)

  132.         if json.loads(ret.text)['errcode'] == 0:

  133.             _logger.info("发送应用消息成功{}".format(json.loads(ret.text)))

  134.         else:

  135.             _logger.error("发送应用消息失败{}".format(json.loads(ret.text)))

  136. 

  137.     def new_department(self):

  138.         """部门接口，创建部门"""

  139.         access_token = self.gen_contacts_access_token()

  140.         if not access_token:

  141.             _logger.info(f"创建部门-----access_token------凭证为空 {access_token}")

  142.             return

  143.         # 去拿Odoo本地的组织架构

  144.         depart_obj = self.with_user(2).env['hr.department']

  145.         depart_records = depart_obj.search([])

  146.         odoo_dept_list = [i['id'] for i in depart_records]  # odoo中正常使用的部门ID

  147.         # 去请求企业微信组织架构

  148.         search_token_url = f"https://qyapi.weixin.qq.com/cgi-bin/department/simplelist?access_token={access_token}"

  149.         response = requests.get(search_token_url, headers=headers)  # 请求企业微信正常使用的部门ID

  150.         _logger.info(f"企业微信请求部门状态 ----【{response.status_code}】 ----")

  151.         create_dept_list = []

  152.         try:

  153.             if response.status_code == 200:

  154.                 ret = json.loads(response.text)

  155.                 wechat_dept_list = [i['id'] for i in ret['department_id']]  # 企业微信正常使用的部门ID

  156.                 create_dept_list = [i for i in odoo_dept_list if i not in wechat_dept_list]  # 需要创建的部门ID

  157.         except Exception as e:

  158.             _logger.info(f"企业微信请求部门有误 ---- {e} ----")

  159. 

  160.         # 需要创建的部门ID

  161.         create_token_url = f"https://qyapi.weixin.qq.com/cgi-bin/department/create?access_token={access_token}"

  162.         print(create_dept_list)

  163.         for wid in create_dept_list:

  164.             # 一级部门/公司跳过不处理，需手动维护ODOO与企业微信一级部门 名称一致

  165.             depart_id = depart_obj.search([('id', '=', wid)])

  166.             if not depart_id.parent_id.id:

  167.                 continue

  168.             data = {

  169.                 "name": depart_id.name,

  170.                 "parentid": depart_id.parent_id.id or False,

  171.                 "id": depart_id.id,

  172.                 "order": depart_id.id,

  173.             }

  174.             ret = requests.post(create_token_url, data=json.dumps(data), headers=headers)

  175.             if ret.status_code == 200:

  176.                 _logger.info("创建部门成功{}".format(json.loads(ret.text)))

  177.             else:

  178.                 _logger.error("创建部门失败{}".format(json.loads(ret.text)))

  179. 

  180.     def update_department(self):

  181.         """部门接口，企微部门更新"""

  182.         token = self.gen_access_token()

  183.         if not token:

  184.             return

  185.         update_depart_url = self.with_user(2).env['ewi.interface'].search([('name', '=', '更新部门')]).url

  186. 

  187.         depart_records = self.with_user(2).env['hr.department'].search([])  # 换部门上下级、换部门负责人、部门更名

  188.         odoo_dept_list = [i['id'] for i in depart_records]  # odoo中正常使用的部门ID

  189. 

  190.         # odoo中正常使用的部门ID

  191.         for i in odoo_dept_list:  # 全量更新企业微信数据

  192.             record_id = self.with_user(2).env['hr.department'].search([('id', '=', i)])

  193.             if i == 1:

  194.                 continue

  195.             data = {

  196.                 "id": record_id.id,

  197.                 "name": record_id.name,  # 部门更名

  198.                 "parentid": record_id.parent_id.id or False,  # 换部门上下级

  199.                 "order": record_id.ewc_dept_order,

  200.             }

  201.             ret = requests.post(update_depart_url.format(token), data=json.dumps(data), headers=headers)

  202.             if json.loads(ret.text)['errcode'] == 0:

  203.                 _logger.info("更新部门成功{}".format(json.loads(ret.text)))

  204.             else:

  205.                 _logger.error("更新部门失败{}".format(json.loads(ret.text)))

  206. 

  207.     def delete_department(self):

  208.         """部门接口，企微部门删除"""

  209.         token = self.gen_access_token()

  210.         if not token:

  211.             return

  212.         delete_depart_url = self.with_user(2).env['ewi.interface'].search([('name', '=', '删除部门')]).url

  213. 

  214.         depart_records_is_leaf = self.with_user(2).env['hr.department'] \

  215.             .search([('active', '=', False), ('is_leaf', '=', True)])  # 取部门标识为归档的数据，层级为末级数据

  216.         odoo_dept_list_is_leaf = [i['id'] for i in depart_records_is_leaf]  # odoo中已停用的部门ID

  217. 

  218.         depart_records_not_leaf = self.with_user(2).env['shr.department'] \

  219.             .search([('active', '=', False), ('is_leaf', '=', False)])  # 取部门标识为归档的数据，层级为非末级数据

  220.         odoo_dept_list_not_leaf = [i['id'] for i in depart_records_not_leaf]  # odoo中已停用的部门ID

  221. 

  222.         # odoo中已停止使用的部门ID

  223.         for i in odoo_dept_list_is_leaf:  # 全取部门标识为归档的数据，层级为末级数据

  224.             if i == 1:

  225.                 continue

  226.             ret = requests.get(delete_depart_url.format(token, i), headers=headers)

  227.             if json.loads(ret.text)['errcode'] == 0:

  228.                 _logger.info("删除部门成功{}".format(json.loads(ret.text)))

  229.             else:

  230.                 _logger.error("删除部门失败{}".format(json.loads(ret.text)))

  231. 

  232.         # odoo中已停止使用的部门ID

  233.         for i in odoo_dept_list_not_leaf:  # 全取部门标识为归档的数据，层级为非末级数据

  234.             if i == 1:

  235.                 continue

  236.             ret = requests.get(delete_depart_url.format(token, i), headers=headers)

  237.             if json.loads(ret.text)['errcode'] == 0:

  238.                 _logger.info("删除部门成功{}".format(json.loads(ret.text)))

  239.             else:

  240.                 _logger.error("删除部门失败{}".format(json.loads(ret.text)))

  241. 

  242.     def gen_employee_userid_list(self):

  243.         """

  244.         人员接口，获取部门成员ID列表

  245.         """

  246.         token = self.gen_access_token()

  247.         if not token:

  248.             return

  249.         gen_employee_userid_url = self.with_user(2).env['ewi.interface'].search([('name', '=', '获取成员ID列表')]).url

  250.         userid_args = {

  251.             "limit": 100

  252.         }

  253.         try:

  254.             ret = requests.post(gen_employee_userid_url.format(token), json=userid_args, headers=headers)

  255.             if json.loads(ret.text)['errcode'] == 0:

  256.                 _logger.info("获取部门成员ID列表成功{}".format(json.loads(ret.text)))

  257.             else:

  258.                 _logger.error("获取部门成员ID列表失败{}".format(json.loads(ret.text)))

  259.         except Exception as e:

  260.             _logger.error('企业微信获取部门成员ID列表时连接失败：' + str(e))

  261.             return

  262. 

  263.     def new_employee(self):

  264.         """

  265.         人员接口，获取部门成员

  266.         """

  267.         access_token = self.gen_contacts_access_token()

  268.         if not access_token:

  269.             _logger.info(f"创建成员-----access_token------凭证为空 {access_token}")

  270.             return

  271.         # 去拿Odoo本地的内部职工

  272.         employee_obj = self.with_user(2).env['hr.employee']

  273.         employee_records = employee_obj.search([])

  274.         odoo_employee_list = [i['mobile_phone'] for i in employee_records]  # odoo中在职人员ID，原则上要求职工号企业内唯一

  275.         # 去请求企业微信部门职工

  276.         search_token_url = f"https://qyapi.weixin.qq.com/cgi-bin/user/list_id?access_token={access_token}"

  277.         response = requests.get(search_token_url, headers=headers)  # 请求企业微信正常使用的部门ID

  278.         _logger.info(f"企业微信请求职工状态 ----【{response.status_code}】 ----")

  279.         create_employee_list = []

  280.         try:

  281.             if response.status_code == 200:

  282.                 ret = json.loads(response.text)

  283.                 wechat_employee_list = [i['userid'] for i in ret['dept_user']]  # 企业微信在职人员ID

  284.                 create_employee_list = [i for i in odoo_employee_list if i not in wechat_employee_list]  # 需要创建的在职人员ID

  285.         except Exception as e:

  286.             _logger.info(f"企业微信请求职工有误 ---- {e} ----")

  287. 

  288.         # 需要创建的职工ID

  289.         create_token_url = f"https://qyapi.weixin.qq.com/cgi-bin/user/create?access_token={access_token}"

  290.         print(create_employee_list)

  291.         for employee in create_employee_list:

  292.             employee_id = employee_obj.search([('mobile_phone', '=', '18066043008')])

  293.             print(employee_id)

  294.             # """填充主部门"""

  295.             # depart_list = [employee_record.department_id.id]

  296.             # """判断是否部门负责人"""

  297.             # if employee_record.department_id.manager_id.id == employee_record.id:

  298.             #     is_leader_in_dept_list = [1]

  299.             # else:

  300.             #     is_leader_in_dept_list = [0]

  301.             # """设置”成员所属部门id列表“、是否”部门负责人“"""

  302.             # for depart in employee_record.department_ids:

  303.             #     depart_list.append(depart.id)

  304.             #     is_leader_in_dept_list.append(0)

  305.             data = {

  306.                 "userid": '18066043008',  # 是, 成员UserID

  307.                 "name": employee_id.name,  # 是, 成员名称

  308.                 # "alias": employee_record.work_no,  # 是, 成员别名

  309.                 "mobile": employee_id.mobile_phone,  # 手机号码

  310.                 "department": [22],  # 成员所属部门id列表

  311.                 # "position": employee_record.job_id.name,  # 职务信息

  312.                 # "gender": 1 if employee_record.gender == 'male' else 2,  # 性别。1表示男性，2表示女性

  313.                 # "email": employee_record.work_email,  # 邮箱

  314.                 # "is_leader_in_dept": is_leader_in_dept_list,  # 1表示为部门负责人，0表示非部门负责人

  315.                 # "direct_leader": ["%s" % employee_record.parent_id.work_no],  # 直属上级UserID

  316.                 # "enable": 0 if employee_record.ewc_enable else 1,  # 启用/禁用成员。1表示启用成员，0表示禁用成员

  317.                 # "telephone": employee_record.work_phone or False,  # 座机

  318.                 # "address": self.env['res.company'].search([('id', '=', '1')]).street or False,  # 地址

  319.                 # "main_department": employee_record.department_id.id,  # 主部门

  320.                 # "to_invite": to_invite,  # 是否邀请该成员使用企业微信

  321.             }

  322.             # if employee_record.depart_code == '1' or not employee_record.parent_id.id or not employee_record.job_id.name:

  323.             #     del employee_args['direct_leader']

  324.             # if not employee_record.job_id.name:

  325.             #     del employee_args['position']

  326.             # if not employee_record.work_email:

  327.             #     del employee_args['email']

  328.             ret = requests.post(create_token_url, data=json.dumps(data), headers=headers)

  329.             if ret.status_code == 200:

  330.                 _logger.info("创建职工成功{}".format(json.loads(ret.text)))

  331.             else:

  332.                 _logger.error("创建职工失败{}".format(json.loads(ret.text)))
上海开阖软件有限公司 沪ICP备12045867号-1