wechatbot.py

#!/usr/bin/env python
# coding: utf-8
# 郑州GDG小助手微信机器人，功能：
# 1.自动接受好友，并根据消息自动将朋友加群
# 2.为避免超过消息限额，采用了队列，并随机延迟消息处理的机制
# @Author 脉脉不得语, email:inferjay@gmail.com

from wxpy import *
import os
import time
import random

import threading
import Queue

#小助手为群主，负责管理的群
OWN_GROUP = u'郑州GDG'
#接受管理指令的群
ADMIN_GROUP = u'管理群名称'
#转发的微信公众号
FW_MPS = u'谷歌'
#加好友欢迎消息
WELCOME_MSG = u'哈哈，我接受了你的好友请求,我是郑州 GDG 小助手机器人，回复"I love GDG"可以帮你加入郑州 GDG 微信群'
#默认回复消息
DEFAULT_MSG = u'Hello {} ，回复"I love GDG"可以帮你加入郑州 GDG 微信群'
#加群关键字(小写)
VALID_KEY="i love gdg"
#命令
ADMIN_CMD =['remove_member','forward_msg','list_cmd']
#

try:
    bot = Bot(os.getcwd() + '/bot.pkl', qr_path=os.getcwd() + '/qr.png')
except IOError: 
    print "Sorry,the file cannot be accessed."    

mps=bot.mps()
i=0;

msg_q=Queue.Queue()

#使用前先调试，确保能找到这个群
group =  bot.groups().search(OWN_GROUP)[0]

admin_g =  bot.groups().search(ADMIN_GROUP)[0]

google_mps = bot.mps().search(FW_MPS)

def send_user_msg(user,msg):
    user.send(msg.format(user.name))


class Process_thread(threading.Thread):
    def run(self):
	while True:
	    msg=msg_q.get()
            print(msg)
            if msg.type==FRIENDS:
                # 接受好友请求
	        time.sleep(random.randint(1,60))
                new_friend = msg.card.accept()
                # 向新的好友发送消息
                new_friend.send(WELCOME_MSG)
	    elif msg.chat==admin_g and msg.is_at:
		cmd = msg.text.split('-:-')
		try:
		    if cmd[1]==ADMIN_CMD[0]:
		        group.remove_members(group.members.search(cmd[2]))
			group.send(u'{} 因违反群规已被移出本群'.format(cmd[2]))
		        admin_g.send(u'@{} 已从{} 移除 {}'.format(msg.member.name, group.name, cmd[2]))
		    elif cmd[1]==ADMIN_CMD[1]:
			group.send(cmd[2])
		        admin_g.send(u'@{} 已转消息到{}'.format(msg.member.name,group.name))
		    else:
			admin_g.send(u'小助手识别的命令列表:{}\n格式：在本群AT小助手-:-CMD-:-参数'.format(ADMIN_CMD))
		except:
		    print(u'命令执行出错')
	    elif valid_msg(msg):
                invite(msg)
	    elif (msg.member is None) and (msg.chat not in mps) and (msg.chat is not None) and (msg.chat.user_name!=u'weixin'):
	        time.sleep(random.randint(5,30))
                send_user_msg(msg.chat,DEFAULT_MSG)
            elif msg.chat in google_mps:
	        group.send('{}\n{}'.format(msg.raw['Text'],msg.raw['Url']))
	    if msg.chat is not None:
	    	msg.chat.mark_as_read()


def valid_msg(msg):
    if msg.text is None:
	return False
    print(msg.text.lower())
    if VALID_KEY in msg.text.lower():
	return True
    else:
	return False

def invite(msg):
    #群用户或是朋友
    if (msg.member==None):
	user=msg.chat
    else:
	user=None
    #用户为空或是公众号则不处理
    if (user is None) or (user in mps):
	print("No user")
	return False

    time.sleep(random.randint(5,30))
    try:
        if user in group:
	    user.send('hi, 你已在{}中!'.format(group.name))
        elif len(group) < 40:
	    group.add_members(user, use_invitation=False)
        else:
	    group.add_members(user, use_invitation=True)
    except ResponseError as e:
            print(e.err_code, e.err_msg)
            #如果超限，随机休眠10-17分钟,共重试5次
            if e.err_code==1205:
                time.sleep(random.randint(600,1000))
		if i<5:  
		    i=i+1  
		    invite(msg)
                else:
	            return False    
    #正常退出，计数器重置为0
    i=0
    return True

@bot.register()
def new_msg(msg):
    if msg is not None:
	msg_q.put(msg)

msg_process=Process_thread()
msg_process.start()



embed()