Python實現釘釘發送報警消息

釘釘開放平台傳送門：https://open.dingtalk.com 我司使用釘釘作為內部通訊工具，基本上大家在電腦和手機上都開着，消息可以第一時間查看，報警消息的即時性要求比較高，所以適合用釘釘通知。 下面介紹如何用Python實現釘釘發送報警消息。 獲取access token 要使用釘釘發送消息，首先需要獲取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 '' 1 2 3 4 5 6 7 8 9 10 11 12 access token在2小時內有效，有效期內重復獲取返回相同結果，有效期會自動延長。corp_id和corp_secret是企業的id和secret，在釘釘的管理后台可以找到。另外，上面用到了urllib2和json，需要import： import urllib2 import json 1 2 消息類型 釘釘的消息類型分為：text，image，voice，file，link和OA，具體消息格式參見：https://open-doc.dingtalk.com/docs/doc.htm?treeId=172&articleId=104972&docType=1 。 下面以發送文本，鏈接和文件消息為例進行說明。 給用戶發送消息 發送文本 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) 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 其中agent_id是一個釘釘應用的id，以釘釘應用的名義給用戶發送消息。users是用戶id列表，每個用戶id是一個字符串。 發送鏈接 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://gw.alicdn.com/tps/TB1FN16LFXXXXXJXpXXXXXXXXXX-256-130.png", "title": title, "text": text } return msg_type, msg 1 2 3 4 5 6 7 8 9 10 11 12 13 其中_send_msg_to_users方法參見前面的代碼，picUrl字段設置的是釘釘官方的圖片，這里用於測試。 發送文件 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 '' 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 需要先上傳文件獲得media_id，然后使用media_id將文件發送給用戶。另外，這里用到了poster，可使用pip安裝： pip install poster 1 之后引入multipart_encode和register_openers函數: from poster.encode import multipart_encode from poster.streaminghttp import register_openers 1 2 給群會話發送消息 與給用戶發送信息類似，區別是需要群會話id，而不是用戶列表，以發送文本消息為例，代碼如下： 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) 1 2 3 4 5 6 7 8 9 10 11 12 其中_gen_text_msg方法參見前面的代碼。 群會話可以自行創建，參見https://open-doc.dingtalk.com/docs/doc.htm?treeId=172&articleId=104977&docType=1 。