main.py

from wxpy import *
from meme_maker import *
from emotional import *
from pymongo import *
import requests
import time
import json
import re


# Init the env, login
bot = Bot()
print("========INIT SUCCESSFULLY=====")
frs = bot.friends()
dummy = ''
yuze = bot.friends().search('Yuze Ma 鱼哲')[0]
apiUrl = 'http://www.tuling123.com/openapi/api'
data = {
	"key": "c306b319ca4541d78a6413ccc16c8119",
	"info": "你好呀",
	"userid":"123"
}
dummy_dict = dict()



# connect to mongodb
client = MongoClient("mongodb://localhost:27017/")
db = client.codemao_data_analysis_dbs
collection = db.msg_records


@bot.register(Group, TEXT)
def auto_reply(msg):
    # 如果是群聊，但没有被 @，则不回复
    if isinstance(msg.chat, Group) and not msg.is_at:
        # print(msg.raw)
        # data["info"] = msg.text
        # try:
        #     time.sleep(1200)
        #     res = requests.post(apiUrl, data = data).json()
        #     return res["text"]
        # except:    
        #     return "你说什么我听不见！"
        print("==EXECUTRED==")
        get_meme()
        msg.sender.send_image("out.png")
        return
    else:
        # 回复消息内容和类型
        try:
            print("==EXECUTRED==")
            get_meme()
            msg.sender.send_image("out.png")
            return "lolol"
            # data["info"] = msg.text
            # res = requests.post(apiUrl, data = data).json()
            # return res["text"]
        except:
            return "Connection lost"

@bot.register(Group, PICTURE)
def auto_reply(msg):
    # 如果是群聊，但没有被 @，则不回复
    if isinstance(msg.chat, Group) and not msg.is_at:
        try:
            print("==Execute by pic in a group chat==")
            get_meme()
            msg.sender.send_image("out.png")
            return 
        except:
            return "Connection lost"
        # return
    else:
        # 回复消息内容和类型
        try:
            print("==Execute by pic in a group chat==")
            get_meme()
            msg.sender.send_image("out.png")
            return "lolol"
        except:
            return "Connection lost"

@bot.register(bot.friends(), PICTURE)
def auto_reply(msg):
    raw_content = msg.raw['Content']
    print(raw_content)
    url = re.findall("cdnurl=(.*) des",raw_content, re.S)
    real_url = re.findall("\"(.*)\"",url[0], re.S)[0]
    print("=================================")
    emotional_data = get_emotion_from_img(real_url)
    print(emotional_data)
    get_meme_by_emotion(emotional_data)
    msg.sender.send_image("out.png")
    # Send emotion data to front-end server
    try:
        # r = requests.post("http://10.2.12.201:9090/getData", data=emotional_data)
        r = requests.post("http://localhost:9090/getData", data=emotional_data)
        print("req suceessfully made")
    except:
        return "你说什么我没听清"

    
    # return emotional_data

@bot.register(bot.friends(), TEXT)
def auto_r(msg):
    print(">>>>>>>>>>>>>>>GET: \n",msg.text)
    try:
        print("==EXECUTRED==")
        text_content = msg.text
        collection.insert_one(msg.raw)

        get_meme()
        msg.sender.send_image("out.png")
        data["info"] = msg.text
        try:
            res = requests.post(apiUrl, data = data).json()
        except:
            print("Tuling is down...")
        return res["text"]        # return 
    except:
        print("Tuling is down...")
        return "23333"