QQ.py

# coding=utf-8

import json
import requests
import html
import re
import os
import time
import logging
import traceback
import importlib
import signal
import platform
from requests.packages import urllib3
from flask import Flask,request
from datetime import datetime
from multiprocessing import Process, Manager

local_dir = str((os.path.split(os.path.realpath(__file__))[0]).replace('\\', '/'))
errorlog_clean = open(local_dir + '/error.log', 'w').close()
app = Flask(__name__)
ppid = os.getppid()
pid = os.getpid()


def prt(mes):
    print(str(datetime.now().strftime('[%Y.%m.%d %H:%M:%S] ')) + str(mes))


def error_log(local):
    with open(local + '/error.log', 'a', encoding='utf-8') as f:
        f.write(str(datetime.now().strftime('[%Y.%m.%d %H:%M:%S] ')) + str(traceback.format_exc()) + '\n')
    prt('程序运行出现错误,错误信息已保存至程序目录下的error.log文件中')


def msgFormat(msg, groupid='0'):
    if '[CQ:reply' in msg:
        reply_cqcode = re.findall('\[CQ:reply[^\]]*\]', msg)
        for cqcode in list(set(reply_cqcode)):
            replymsg_id = re.findall('.*,id=([^(\]|,|\s)]*).*', cqcode)[0]
            reply_format = replymsg(replymsg_id)
            msg = msg.replace(cqcode, reply_format)
    if '[CQ:image' in msg:
        img_cqcode = re.findall('\[CQ:image[^\]]*\]', msg)
        for cqcode in list(set(img_cqcode)):
            imgurl =  re.findall('.*,url=([^(\]|,|\s)]*).*', cqcode)[0]
            msg = msg.replace(cqcode, '[图片] ' + imgurl + '\n') if str(value.get('TG')) == "True" else msg.replace(cqcode, '[图片]')
    if '[CQ:video' in msg:
        video_cqcode = re.findall('\[CQ:video[^\]]*\]', msg)
        for cqcode in list(set(video_cqcode)):
            videourl = re.findall('.*,url=([^(\]|,|\s)]*).*', cqcode)[0]
            msg = msg.replace(cqcode, '[视频] ' + videourl + '\n') if str(value.get('TG')) == "True" else msg.replace(cqcode, '[视频]')
    if '[CQ:at' in msg:
        at_cqcode = re.findall('\[CQ:at[^\]]*\]', msg)
        for cqcode in list(set(at_cqcode)):
            uid = re.findall('.*,qq=([^(\]|,|\s)]*).*', cqcode)[0]
            if str(uid) == 'all':
                msg = msg.replace(cqcode, '@全体成员')
                continue
            at_info_api = 'http://localhost:5700/get_group_member_info?group_id=' + str(groupid) + "&user_id=" + str(uid)
            at_info = json.loads(requests.get(at_info_api).content)
            if str(at_info.get("data")) != 'None':
                at_name = "@" + str(at_info.get("data").get("nickname")) if at_info.get("data").get("card") == "" else "@" + str(at_info.get("data").get("card"))
                at_cqcode = '[CQ:at,qq=' + str(uid) + ']'
                msg = msg.replace(at_cqcode, at_name)
            else:
                msg = 'None'
                break
    if "[CQ:face" in msg:
        face_cqcode = re.findall('\[CQ:face[^\]]*\]', msg)
        for cqcode in list(set(face_cqcode)):
            face_id = re.findall('.*,id=([^(\]|,|\s)]*).*', cqcode)[0]
            emoji_name = getEmojiName(face_id)
            msg = msg.replace(cqcode, emoji_name)
    if "[CQ:json" in msg:
        try:
            data = json.loads(html.unescape(re.findall('\[CQ:json,data=([^\]]*?)\]', msg)[0]))
            view = list(data.get('meta').keys())[0]
            if 'com.tencent.miniapp' in data.get('app'):
                mini_title = data.get('meta').get(view).get('title')
                mini_url = data.get('meta').get(view).get('qqdocurl').replace('\\', '/')
                msg = '[小程序]' + mini_title + '\n' + mini_url if str(value.get('TG')) == "True" else '[小程序]' + mini_title
            elif 'com.tencent.structmsg' in data.get('app'):
                jumpurl = data.get('meta').get(view).get('jumpUrl').replace('\\', '/')
                title = data.get('meta').get(view).get('title')
                msg = '[分享]' + title + '\n' + jumpurl if str(value.get('TG')) == 'True' else '[分享]' + title
            else:
                msg = '[卡片消息]'
        except:
            msg = '[卡片消息]'
    if "[CQ:record" in msg:
        msg = "[语音]"
    if "[CQ:share" in msg:
        msg = "[链接]"
    if "[CQ:music" in msg:
        msg = "[音乐分享]"
    if "[CQ:redbag" in msg:
        msg = "[红包]"
    if "[CQ:forward" in msg:
        msg = "[合并转发]"
    if "[CQ:xml" in msg:
        msg = '[卡片消息]'
    if "[戳一戳]请使用最新版手机QQ体验新功能。" in msg:
        msg = "[戳一戳]"
    msg = html.unescape(msg)
    return msg


def error(pid, ppid, errorlog_dir):
    error_log(errorlog_dir)
    prt('程序终止运行')
    system = str(platform.system())
    if system == 'Linux':
        os.killpg(os.getpgid(int(pid)), signal.SIGTERM)
    elif system == 'Windows':
        os.system('taskkill /F /T /PID ' + str(pid))
    else:
        os.kill(int(ppid), signal.SIGTERM)


def config_update(value):
    try:
        config_path = value.get('local_dir') + '/config.py'
        stat = os.stat(config_path)
        old_st_mtime_ns = stat.st_mtime_ns
        while 1:
            stat = os.stat(config_path)
            if stat.st_mtime_ns != old_st_mtime_ns:
                old_st_mtime_ns = stat.st_mtime_ns
                try:
                    importlib.reload(config)
                    newcfg = {'MiPush': str(config.MiPush), 'FCM': str(config.FCM), 'TG': str(config.TG), 'MiPush_KEY': str(config.MiPush_KEY),
                                'FCM_KEY': str(config.FCM_KEY), 'TG_KEY': str(config.TG_KEY), 'WhiteList': list(config.WhiteList),
                                'TG_UID': str(config.TG_UID), 'TG_GroupLink': dict(config.TG_GroupLink), 'MiPush_API': str(config.MiPush_API),
                                'FCM_API': str(config.FCM_API), 'TG_API': str(config.TG_API)}
                except:
                    prt('读取配置文件异常,请检查配置文件的语法是否有误或所需变量是否存在，程序使用最后一次正确的的配置')
                    error_log(value.get('local_dir'))
                    continue
                for i in list(newcfg.keys()):
                    if str(value.get(i)) != str(newcfg.get(i)):
                        prt(str(i) + '更改,新' + str(i) + '值为' + str(newcfg.get(i)))
                value.update(newcfg)
            time.sleep(1)
    except KeyboardInterrupt:
        prt('由于触发了KeyboardInterrupt(同时按下了Ctrl+C等情况)，程序强制停止运行')
    except FileNotFoundError:
        prt('配置文件丢失，程序强制停止运行')
        error(value.get('pid'), value.get('ppid'), value.get('local_dir'))
    except:
        error(value.get('pid'), value.get('ppid'), value.get('local_dir'))


def getGroupName(groupId):
    groupInfo = json.loads(requests.get("http://localhost:5700/get_group_list").text)
    group = list(groupInfo.get("data"))
    for i in group:
        if str(groupId) == str(dict(i).get('group_id')):
            return str(dict(i).get('group_name'))
    return "未知"


def getnickname(id):
    url = 'http://localhost:5700/get_stranger_info?user_id=' + str(id)
    userInfo = json.loads(requests.get(url).text)
    return str(userInfo.get("data").get("nickname"))


def styletime(now):
    timeArray = time.localtime(now)
    otherStyleTime = time.strftime("%Y-%m-%d %H:%M:%S", timeArray)
    return otherStyleTime


def getfriendmark(UID):
    friendInfo = json.loads(requests.get("http://localhost:5700/get_friend_list").text)
    friend_list = list(friendInfo.get('data'))
    for i in friend_list:
        if UID == dict(i).get("user_id"):
            name = dict(i).get("remark") if dict(i).get("remark") != '' else dict(i).get("nickname")
            return str(name)
    return getnickname(UID)


def data_send(url, **kwargs):
    for i in range(1, 5):
        try:
            response = requests.post(url, data=kwargs, timeout=5, verify=False)
            if response.status_code > 299:
                raise RuntimeError
        except:
            if str(i) == '4':
                prt('连续三次向接口发送数据超时/失败，可能是网络问题或接口失效，终止发送')
                break
            prt('向接口发送数据超时/失败，第' + str(i) + '次重试')
        else:
            prt('成功向接口发送数据↑')
            break


def replymsg(msgid):
    replymsg_api= f'http://localhost:5700/get_msg?message_id={msgid}'
    replymsg_json = json.loads(requests.get(replymsg_api).text)
    if str(replymsg_json.get("data")) != 'None':
        replymsg = replymsg_json.get("data").get("message")
        replymsg_sender = replymsg_json.get("data").get("sender").get("nickname")
        replymsg_timestamp = replymsg_json.get("data").get("time")
        replymsg_styletime = styletime(replymsg_timestamp)
        reply_msg = "[回复" + replymsg_sender + "(" + replymsg_styletime + "): " + replymsg + "]\n" if str(value.get('TG')) == "True" else f"[回复{replymsg_sender}的消息]"
    else:
        reply_msg = '[消息回复]'
    return reply_msg


def getEmojiName(face_id):
    face_name = '[表情]'
    for i in range(0, len_face):
        QSid = face_data['sysface'][i]['QSid']
        QDes = face_data['sysface'][i]['QDes']
        if QSid == face_id:            
            face_name = '[' + QDes.replace('/','') + ']'
            break
    return face_name


def data_handling(value, json_data):
    try:
        urllib3.disable_warnings()
        groupId = '0'
        TG_ID = '0'
        if json_data.get("post_type") == "request":
            if json_data.get("request_type") == "friend":
                friendId = json_data.get("user_id")
                nickname = getnickname(friendId)
                prt("新的好友添加请求：%s" % friendId)
                if str(value.get('MiPush')) == "True":
                    data_send(value.get('MiPush_API'), title="新的好友添加请求", content='%s想要添加您为好友' % friendId, alias=value.get('MiPush_KEY'))
                if str(value.get('FCM')) == "True":
                    data_send(value.get('FCM_API'), id=value.get('FCM_KEY'), title="新的好友添加请求", message='%s想要添加您为好友' % friendId, type='QQ_FriendAdd', action='mqq://')
                if str(value.get('TG')) == "True":
                    msg = friendId + ' 请求添加您为好友'
                    url = f"{str(value.get('TG_API'))}/bot{str(value.get('TG_KEY'))}/sendMessage"
                    TG_ID = str(value.get('TG_UID'))
                    data_send(str(url), chat_id=str(TG_ID), text=str(msg), disable_web_page_preview="true")
        elif json_data.get("post_type") == "notice":
            if json_data.get("notice_type") == "group_upload":
                groupId = json_data.get("group_id")
                if int(groupId) in list(value.get('WhiteList')):
                    groupName = getGroupName(groupId)
                    filename = json_data.get("file").get("name")
                    userid = json_data.get("user_id")
                    cardurl = 'http://localhost:5700/get_group_member_info?group_id=' + str(groupId) + "&user_id=" + str(userid)
                    card = json.loads(requests.get(cardurl).content)
                    if card.get("data").get("card") != "":
                        card = card.get("data").get("card") + " "
                    else:
                        card = card.get("data").get("nickname") + " "
                    msg = card + '上传了 ' + filename
                    prt(str(groupName) + ': ' + str(msg))
                    if str(value.get('MiPush')) == "True":
                        data_send(value.get('MiPush_API'), title="QQ通知", content='%s' % (msg), alias=value.get('MiPush_KEY'))
                    if str(value.get('FCM')) == "True":
                        data_send(value.get('FCM_API'), id=value.get('FCM_KEY'), title="QQ通知", message=str(msg), type='QQ_groupMsg', action='mqq://')
                    if str(value.get('TG')) == "True":
                        if str(groupId) in dict(value.get('TG_GroupLink')):
                            TG_ID = dict(value.get('TG_GroupLink')).get(str(groupId))
                        else:
                            TG_ID = str(value.get('TG_UID'))
                        url = f"{str(value.get('TG_API'))}/bot{str(value.get('TG_KEY'))}/sendMessage"
                        data_send(str(url), chat_id=str(TG_ID), text=str(msg), disable_web_page_preview="true")
        elif json_data.get("message_type") == "private":
            msg = msgFormat(json_data.get("message"))
            uid = json_data.get("sender").get("user_id")
            nickname = getfriendmark(uid)
            prt("%s: %s" % (nickname, msg))
            if str(value.get('MiPush')) == "True":
                data_send(value.get('MiPush_API'), title=str(nickname), content=str(msg), alias=value.get('MiPush_KEY'))
            if str(value.get('FCM')) == "True":
                data_send(value.get('FCM_API'), id=value.get('FCM_KEY'), title=str(nickname), message=str(msg), type='QQ_privateMsg', action='mqq://')
            if str(value.get('TG')) == "True":
                if str(uid) in dict(value.get('TG_GroupLink')):
                    TG_ID = dict(value.get('TG_GroupLink')).get(str(uid))
                else:
                    TG_ID = str(value.get('TG_UID'))
                msg = nickname + ":\n" + msg
                url = f"{str(value.get('TG_API'))}/bot{str(value.get('TG_KEY'))}/sendMessage"
                data_send(str(url), chat_id=str(TG_ID), text=str(msg), disable_web_page_preview="true")
        elif json_data.get("message_type") == "group":
            groupId = json_data.get("group_id")
            if int(groupId) in list(value.get('WhiteList')):
                uid = json_data.get("sender").get("user_id")
                nickName = json_data.get("sender").get("nickname")
                card = json_data.get("sender").get("card")
                msg = msgFormat(json_data.get("message"), groupid=str(groupId))
                groupName = getGroupName(groupId)
                nickName = str(card) if str(card) != "" else str(nickName)
                if str(msg) != 'None':
                    prt("%s: %s: %s" % (groupName, nickName, msg))
                    if str(value.get('MiPush')) == "True":
                        data_send(value.get('MiPush_API'), title='%s' % groupName, content='%s:%s' % (nickName, msg), alias=value.get('MiPush_KEY'))
                    if str(value.get('FCM')) == "True":
                        data_send(value.get('FCM_API'), id='%s' % value.get('FCM_KEY'), title=str(groupName), message='%s:%s' % (nickName, msg), type='QQ_groupMsg', action='mqq://')
                    if str(value.get('TG')) == "True":
                        if str(groupId) in dict(value.get('TG_GroupLink')):
                            TG_ID = dict(value.get('TG_GroupLink')).get(str(groupId))
                        else:
                            TG_ID = str(value.get('TG_UID'))
                        text = nickName + "[" + groupName + "]" + ":\n" + msg
                        url = f"{str(value.get('TG_API'))}/bot{str(value.get('TG_KEY'))}/sendMessage"
                        data_send(str(url), chat_id=str(TG_ID), text=str(text), disable_web_page_preview="true")
    except KeyboardInterrupt:
        prt('由于触发了KeyboardInterrupt(同时按下了Ctrl+C等情况)，程序强制停止运行')
    except:
        error_log(value.get('local_dir'))


@app.route("/", methods=['GET', 'POST'])
def recvMsg():
    try:
        data = request.get_data()
        json_data = json.loads(data.decode("utf-8"))
        dat_handling = Process(target=data_handling, args=(value, json_data, ))
        dat_handling.daemon = True
        dat_handling.start()
        return "200 OK"
    except KeyboardInterrupt:
        prt('由于触发了KeyboardInterrupt(同时按下了Ctrl+C等情况)，程序强制停止运行')
    except:
        error_log(local_dir)


if __name__ == '__main__':
    try:
        import config
    except:
        prt("读取配置文件异常,程序终止运行,请检查配置文件是否存在或语法是否有问题")
        error_log(local_dir)
        os._exit(0)
    try:
        with open(local_dir + '/face_config.json', 'r', encoding='utf-8') as f:
            face_data = json.load(f)
        len_face = len(face_data.get("sysface"))
    except:
        prt("读取表情包配置文件异常,程序终止运行,请检查配置文件是否存在或语法是否有问题")
        error_log(local_dir)
        os._exit(0)
    try:
        user_nickname = json.loads(requests.get("http://localhost:5700/get_login_info").text).get("data").get("nickname")
        prt('欢迎回来，' + str(user_nickname))
        del user_nickname
    except:
        prt("无法从go-cqhttp获取信息,程序终止运行,请检查go-cqhttp是否运行或端口配置是否正确")
        error_log(local_dir)
        os._exit(0)
    try:
        log = logging.getLogger('werkzeug')
        log.setLevel(logging.ERROR)
        urllib3.disable_warnings()
        prt('程序开始运行')
        value = Manager().dict()
        value.update({'pid': str(pid), 'ppid': str(ppid),'local_dir': str(local_dir), 'MiPush': str(config.MiPush),
                        'FCM': str(config.FCM), 'TG': str(config.TG), 'MiPush_KEY': str(config.MiPush_KEY),
                        'FCM_KEY': str(config.FCM_KEY), 'TG_KEY': str(config.TG_KEY),
                        'WhiteList': list(config.WhiteList), 'TG_UID': str(config.TG_UID),
                        'TG_GroupLink': dict(config.TG_GroupLink), 'MiPush_API': str(config.MiPush_API),
                        'FCM_API': str(config.FCM_API), 'TG_API': str(config.TG_API)})
        conf_update = Process(target=config_update, args=(value, ))
        conf_update.daemon = True
        conf_update.start()
        app.run(host="127.0.0.1", port=5000)
    except KeyboardInterrupt:
        prt('由于触发了KeyboardInterrupt(同时按下了Ctrl+C等情况)，程序强制停止运行')
    except:
        error(pid, ppid, local_dir)