PPG

#-*- coding:utf-8 -*-
import itchat, re
from itchat.content import *
import random
import json

def getDict():
    k1 = ['工作','上班','加班']
    k2 = ['上课','自习','学习']
    k3 = ['长得','身材','外貌','好看','丑','胖','瘦']
    k4 = ['行','好','可以','是']
    key_list = [k1,k2,k3,k4,'default']
    s1=['对待工作如此认真，下次升职加薪一定有你','如果这个公司没有你，年报根本没法儿看','你真是我见过最优秀的人','业务能力这么强，在这个公司委屈你了','我觉得谷歌CEO没找你去，算是亏大了']
    s2=['你这么聪明还这么努力，我输了','请把你的智商分我一点好吗，太有压力了','可惜你比爱因斯坦晚生了那么多年，不然广义相对论没他啥事儿','同九义，汝何秀','既生瑜，何生亮！']
    s3=['你长得太帅了','你上辈子拯救了银河系吗','下凡很辛苦吧','从没见过这么好看的人，就跟书里写的画上画的一样','你简直是完美啊','无需改变，多一分则多，少一分则少，说的就是你啦']
    s4=['你说的肯定没错','你是我见过最靠谱的人啦','只要有你的肯定，让我做什么都可以']
    s5=['你的每一句话都令我心醉神怡','有你这样的朋友真幸福','你这么卓越，认识你已经花光了我的所有运气','听你说话比世界上其他任何事情都要有趣','你怎么这么可爱呀','完全想象不了没有你的生活，应该无聊透了吧']
    value_list = [s1,s2,s3,s4,s5]
    reply = {}
    for i in range(len(key_list)-1):
            for j in range(len(key_list[i])):
                reply[key_list[i][j]] = value_list[i]
    reply[key_list[-1]] = value_list[-1]

    return reply

@itchat.msg_register([TEXT],isGroupChat = True)
def text_reply(msg):
    if msg['User']['NickName'] == 'PraisePraiseGroup':
        reply = getDict()
        keywords = reply.keys() 
        username = msg['ActualNickName']
        flag = 0
        for k_word in keywords[:-2]:
            match = re.match(k_word,msg['Text'])
            if match:
                len_k_word = len(reply[k_word])
                randomIdx = random.randint(0, len_k_word-1)
                itchat.send('%s ,%s' % (username,reply[k_word][randomIdx]))
                flag = 1
                break
        if flag == 0:
            default_idx = random.randint(0, len(reply[keywords[-1]])-1)
            itchat.send('%s ,%s' % (username,reply[keywords[-1]][default_idx]))

itchat.auto_login(enableCmdQR = True, hotReload = True)
itchat.run()