Python怎么實現(xiàn)釘釘發(fā)送報警消息

小編給大家分享一下Python怎么實現(xiàn)釘釘發(fā)送報警消息，希望大家閱讀完這篇文章之后都有所收獲，下面讓我們一起去探討吧！

下面介紹如何用Python實現(xiàn)釘釘發(fā)送報警消息。

獲取access token

要使用釘釘發(fā)送消息，首先需要獲取access token，代碼如下：

def get_access_token():
 url = 'https://oapi.dingtalk.com/gettoken?corpid=%s&corpsecret=%s' % (corp_id, corp_secret)
 request = urllib2.Request(url)
 response = urllib2.urlopen(request)
 response_str = response.read()
 response_dict = json.loads(response_str)
 error_code_key = "errcode"
 access_token_key = "access_token"
 if response_dict.has_key(error_code_key) and response_dict[error_code_key] == 0 and response_dict.has_key(access_token_key):
  return response_dict[access_token_key]
 else:
  return ''

access token在2小時內(nèi)有效，有效期內(nèi)重復(fù)獲取返回相同結(jié)果，有效期會自動延長。corp_id和corp_secret是企業(yè)的id和secret，在釘釘?shù)墓芾砗笈_可以找到。另外，上面用到了urllib2和json，需要import：

import urllib2
import json

消息類型

釘釘?shù)南㈩愋头譃椋簍ext，image，voice，file，link和OA，具體消息格式參見：https://open-doc.dingtalk.com/docs/doc.htm?treeId=172&articleId=104972&docType=1 。

下面以發(fā)送文本，鏈接和文件消息為例進(jìn)行說明。

給用戶發(fā)送消息

發(fā)送文本

def send_text_to_users(access_token, users, text):
 msg_type, msg = _gen_text_msg(text)
 return _send_msg_to_users(access_token, users, msg_type, msg)

def _gen_text_msg(text):
 msg_type = 'text'
 msg = { "content": text }
 return msg_type, msg

def _send_msg_to_users(access_token, users, msg_type, msg):
 to_users = '|'.join(users)
 body_dict = {
  "touser": to_users,
  "agentid": agent_id,
  "msgtype": msg_type
 }
 body_dict[msg_type] = msg
 body = json.dumps(body_dict)
 return _send_msg("https://oapi.dingtalk.com/message/send?access_token=", access_token, body)

其中agent_id是一個釘釘應(yīng)用的id，以釘釘應(yīng)用的名義給用戶發(fā)送消息。users是用戶id列表，每個用戶id是一個字符串。

發(fā)送鏈接

def send_link_to_users(access_token, users, url, title, text):
 msg_type, msg = _gen_link_msg(url, title, text)
 return _send_msg_to_users(access_token, users, msg_type, msg)

def _gen_link_msg(url, title, text):
 msg_type = 'link'
 msg = { 
  "messageUrl": url,
  "picUrl": "https://cache.yisu.com/upload/information/20200622/113/38812.png",
  "title": title,
  "text": text 
 }
 return msg_type, msg

其中_send_msg_to_users方法參見前面的代碼，picUrl字段設(shè)置的是釘釘官方的圖片，這里用于測試。

發(fā)送文件

def send_file_to_users(access_token, users, file_name):
 media_id = upload_file(access_token, file_name)
 if media_id == '':
  return
 msg_type, msg = _gen_file_msg(media_id)
 return _send_msg_to_users(access_token, users, msg_type, msg)

def upload_file(access_token, file_name):
 register_openers()
 datagen, headers = multipart_encode({'media': open(file_name, 'rb')})
 requst_url = 'https://oapi.dingtalk.com/media/upload?access_token=' + access_token + '&type=file'
 request = urllib2.Request(requst_url, datagen, headers)
 response = urllib2.urlopen(request)
 response_str = response.read()
 response_dict = json.loads(response_str)
 media_id_key = 'media_id'
 error_code_key = 'errcode'
 if response_dict.has_key(error_code_key) and response_dict[error_code_key] == 0 and response_dict.has_key(media_id_key):
  return response_dict[media_id_key]
 else:
  return ''

需要先上傳文件獲得media_id，然后使用media_id將文件發(fā)送給用戶。另外，這里用到了poster，可使用pip安裝：

pip install poster

之后引入multipart_encode和register_openers函數(shù):

from poster.encode import multipart_encode
from poster.streaminghttp import register_openers

給群會話發(fā)送消息

與給用戶發(fā)送信息類似，區(qū)別是需要群會話id，而不是用戶列表，以發(fā)送文本消息為例，代碼如下：

def send_text_to_chat(access_token, chat_id, text):
 msg_type, msg = _gen_text_msg(text)
 return _send_msg_to_chat(access_token, chat_id, msg_type, msg)

def _send_msg_to_chat(access_token, chat_id, msg_type, msg):
 body_dict = {
  "chatid": chat_id,
  "msgtype": msg_type
 }
 body_dict[msg_type] = msg
 body = json.dumps(body_dict)
 return _send_msg("https://oapi.dingtalk.com/chat/send?access_token=", access_token, body)

其中_gen_text_msg方法參見前面的代碼。

看完了這篇文章，相信你對“Python怎么實現(xiàn)釘釘發(fā)送報警消息”有了一定的了解，如果想了解更多相關(guān)知識，歡迎關(guān)注億速云行業(yè)資訊頻道，感謝各位的閱讀！