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

def getDict():
    k1 = ['工作','上班','加班']
    k2 = ['上課','自習(xí)','學(xué)習(xí)']
    k3 = ['長得','身材','外貌','好看','丑','胖','瘦']
    k4 = ['行','好','可以','是']
    key_list = [k1,k2,k3,k4,'default']
    s1=['對待工作如此認(rèn)真，下次升職加薪一定有你','如果這個公司沒有你，年報根本沒法兒看','你真是我見過最優(yōu)秀的人','業(yè)務(wù)能力這么強(qiáng)，在這個公司委屈你了','我覺得谷歌CEO沒找你去，算是虧大了']
    s2=['你這么聰明還這么努力，我輸了','請把你的智商分我一點(diǎn)好嗎，太有壓力了','可惜你比愛因斯坦晚生了那么多年，不然廣義相對論沒他啥事兒','同九義，汝何秀','既生瑜，何生亮！']
    s3=['你長得太帥了','你上輩子拯救了銀河系嗎','下凡很辛苦吧','從沒見過這么好看的人，就跟書里寫的畫上畫的一樣','你簡直是完美啊','無需改變，多一分則多，少一分則少，說的就是你啦']
    s4=['你說的肯定沒錯','你是我見過最靠譜的人啦','只要有你的肯定，讓我做什么都可以']
    s5=['你的每一句話都令我心醉神怡','有你這樣的朋友真幸福','你這么卓越，認(rèn)識你已經(jīng)花光了我的所有運(yùn)氣','聽你說話比世界上其他任何事情都要有趣','你怎么這么可愛呀','完全想象不了沒有你的生活，應(yīng)該無聊透了吧']
    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()