python 企业微信发送脚本

GIT方式：git clone https://github.com/huangantai/QywxPython.git 1、调用方式： python qywx.py -t text -m 测试消息发送 python qywx.py -t image -m /root/test.png python qywx.py -t voice -m /root/test.amr python qywx.py -t video -m /root/test.mp4 python qywx.py -t file -m /root/test.zip 可以直接脚本调用，也可以在其他python中导入引用。 2、qywx.py import urllib.request import json import requests import logging import os import sys import getopt

class Qywx(): '''

-----------发送企业微信的消息格式------------

图片（image）:1MB，支持JPG,PNG格式

语音（voice）：2MB，播放长度不超过60s，支持AMR格式

视频（video）：10MB，支持MP4格式

普通文件（file）：20MB

''' def init(self,corpid='yourcorpid',corpsecret='yoursecret',is_log=True,log_path="qywx.log",log_level=0): """初始化，需要传入企业ID和密钥，在企业微信的页面上有显示""" url="https://qyapi.weixin.qq.com" self.corpid=corpid self.corpsecret=corpsecret self.is_log=is_log self.log_path=log_path self.log_level=log_level if is_log==True: LOG_FORMAT = "%(asctime)s - %(levelname)s: %(message)s" DATE_FORMAT = "%m/%d/%Y %H:%M:%S %p" if log_level==0: logging.basicConfig(level=logging.INFO,filename=log_path,format=LOG_FORMAT, datefmt=DATE_FORMAT) else: logging.basicConfig(level=logging.ERROR,filename=log_path,format=LOG_FORMAT, datefmt=DATE_FORMAT)

def send_message(self,msg,msgtype,agid): upload_token=self.get_upload_token(self.corpid,self.corpsecret) if msgtype=="text": data=self.msg_messages(msg,agid,msgtype='text',msgid="content") elif msgtype=="image": media_id=self.get_media_ID(msg,upload_token,msgtype="image") data=self.msg_messages(media_id,agid,msgtype='image',msgid="media_id") elif msgtype=="voice": media_id=self.get_media_ID(msg,upload_token,msgtype="voice") data=self.msg_messages(media_id,agid,msgtype='voice',msgid="media_id") elif msgtype=="video": media_id=self.get_media_ID(msg,upload_token,msgtype="video") data=self.msg_messages(media_id,agid,msgtype='video',msgid="media_id") elif msgtype=="file": media_id=self.get_media_ID(msg,upload_token,msgtype="file") data=self.msg_messages(media_id,agid,msgtype='file',msgid="media_id") else: raise Exception("msgtype参数错误，参数只能是image或text或voice或video或file") url="https://qyapi.weixin.qq.com" token=self.get_token(url,self.corpid,self.corpsecret) send_url = '%s/cgi-bin/message/send?access_token=%s' % (url,token) respone=urllib.request.urlopen(urllib.request.Request(url=send_url, data=data)).read() x = json.loads(respone.decode())['errcode'] if x == 0: print ("{} 发送成功".format(msg)) if self.is_log==True: logging.info("{} 发送成功".format(msg)) else: print ("{} 发送失败".format(msg)) if self.is_log==True: logging.info("{} 发送失败".format(msg))

def send_msg_message(self,msg,agid=1000002): try: self.send_message(msg,'text',agid) except Exception as e: if self.is_log==True: logging.error(e) def send_image_message(self,path,agid=1000002): if path.endswith("jpg")==False and path.endswith("png")==False: raise Exception("图片只能为jpg或png格式") if os.path.getsize(path)>1048576: raise Exception("图片大小不能超过1MB") try: self.send_message(path,'image',agid) except Exception as e: if self.is_log==True: logging.error(e) def send_voice_message(self,path,agid=1000002): if path.endswith("amr")==False: raise Exception("语音文件只能为amr格式，并且不能大于2MB，不能超过60s") if os.path.getsize(path)>2097152: raise Exception("语音文件大小不能超过2MB，并且不能超过60s，只能为amr格式") try: self.send_message(path,'voice',agid) except Exception as e: if self.is_log==True: logging.error(e) def send_video_message(self,path,agid=1000002): if path.endswith("mp4")==False: raise Exception("视频文件只能为mp4格式，并且不能大于10MB") if os.path.getsize(path)>10485760: raise Exception("视频文件大小不能超过10MB,只能为mp4格式") try: self.send_message(path,'video',agid) except Exception as e: if self.is_log==True: logging.error(e) def send_file_message(self,path,agid=1000002): if os.path.getsize(path)>20971520: raise Exception("文件大小不能超过20MB") try: self.send_message(path,'file',agid) except Exception as e: if self.is_log==True: logging.error(e)

def get_token(self,url, corpid, corpsecret): token_url = '%s/cgi-bin/gettoken?corpid=%s&corpsecret=%s' % (url, corpid, corpsecret) token = json.loads(urllib.request.urlopen(token_url).read().decode())['access_token'] return token def get_upload_token(self,corid,corsec): gurl="https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid={}&corpsecret={}".format(corid,corsec) r=requests.get(gurl) dict_result=(r.json()) return dict_result['access_token']

def get_media_ID(self,path,token,msgtype="image"): """上传资源到企业微信的存储上,msgtype有image,voice,video,file""" media_url = "https://qyapi.weixin.qq.com/cgi-bin/media/upload?access_token={}&type={}".format(token,msgtype) with open(path, 'rb') as f: files = {msgtype: f} r = requests.post(media_url, files=files) re = json.loads(r.text) return re['media_id']

def msg_messages(self,msg,agid,msgtype='text',msgid="content"): """ msgtype有text,image,voice,video,file；如果msgytpe为text,msgid为content，如果是其他，msgid为media_id。 msg为消息的实际内容，如果是文本消息，则为字符串，如果是其他类型，则传递media_id的值。

""" values = { "touser": '@all', "msgtype": msgtype, "agentid": agid, msgtype: {msgid: msg}, "safe": 0 } msges=(bytes(json.dumps(values), 'utf-8')) return msges

def usage(): str=""" ********************调用格式如下********************** 发送文本消息： python qywx.py -t text -m 你要发送的消息 发送文本消息，指定要发送的应用的ID： python qywx.py -t text -m 你要发送的消息 -a 1000005 发送图片消息： python qywx.py -t image -m 图片的全路径 发送图片消息，指定要发送的应用的ID： python qywx.py -t image -m 图片的全路径 -a 1000005 发送语音消息： python qywx.py -t voice -m 语音的路径 发送视频消息： python qywx.py -t video -m 视频的路径 发送文件消息： python qywx.py -t file -m 文件的路径 *********************注意事项*************************** 图片只能是png或jpg图片，大小不能超过1MB 语音只能是amr格式，播放长度不能超过60s,大小不能超过2MB 视频只能是mp4格式，大小不能超过10MB 普通文件大小不能超过20MB

*********************例子******************************* python qywx.py -t image -m /root/test.png python qywx.py -t text -m 测试消息

""" print(str) if name=="main": qywx=Qywx() #Qywx(corpid='yourid',corpsecret='yoursecret',is_log=True,log_path='yourpath') try: options,args = getopt.getopt(sys.argv[1:],"ht:m:a:",["help","type=","message=","agentid="]) except Except as e: print(e) a=1000002 t=None m=None for name,value in options: if name in ("-h","--help"): usage() elif name in ("-t","--type"): t=value elif name in ("-m","--message"): m=value elif name in ("-a","--agentid"): a=value else: usage() if t and m: if t=="text": qywx.send_msg_message(m,agid=a) elif t=="image": qywx.send_image_message(m,agid=a) elif t=="voice": qywx.send_voice_message(m,agid=a) elif t=="video": qywx.send_video_message(m,agid=a) elif t=="file": qywx.send_file_message(m,agid=a) else: usage()

-----------代码中直接调用---------------------

qywx=Qywx() #Qywx(corpid='yourid',corpsecret='yoursecret',is_log=True,log_path='yourpath')

qywx.send_msg_message("test")

qywx.send_image_message("/root/test.png")

qywx.send_voice_message("/root/test.amr")

qywx.send_video_message("/root/test.mp4")

qywx.send_file_message("/root/test.mp3")

qywx.send_msg_message("test",agid=1000003)

-----------代码中直接调用---------------------