AiOpsWechatiPad/app.py

from flask import Flask, send_from_directory, request,jsonify
from flask_restful import Api,got_request_exception
from resources.user_resource import UserResource
from resources.messages_resource import MessagesResource
from resources.contacts_resources import DeleteFriendResource,GetFriendsInfoResource
from resources.config_reources import GetWxchatConfigResource ,SaveWxchatConfigResource
from resources.groups_resources import GetGroupsInfoResource
from common.log import logger, log_exception 
from common.interceptors import before_request, after_request, handle_exception
import threading
from common import kafka_helper, redis_helper,utils

import logging 
from config import load_config
  
from wechat.biz import start_kafka_consumer_thread

from channel import channel_factory 

from wechat import gewe_chat

import os,time,json

from voice.ali.ali_voice import AliVoice



# 自定义错误消息
errors = {
    'UserAlreadyExistsError': {
        'message': "A user with that username already exists.",
        'status': 409,
    },
    'ResourceDoesNotExist': {
        'message': "A resource with that ID no longer exists.",
        'status': 410,
        'extra': "Any extra information you want.",
    },
}



def worker():
    kafka_helper.start()
    redis_helper.start()

    start_wxchat_thread()
    start_kafka_consumer_thread()



def login_or_reconnect(wxchat:gewe_chat.GeWeChatCom, token_id, app_id, hash_key, is_reconnect=False):
    """
    封装微信登录或重连的逻辑
    """
    while True:
        if is_reconnect:
            logger.info("尝试重连...")
        else:
            logger.info("获取二维码进行登录...")
        
        qr_code = wxchat.get_login_qr_code(token_id, app_id)
        base64_string = qr_code.get('qrImgBase64')
        uuid = qr_code.get('uuid')

        app_id = app_id or qr_code.get('appId')
        start_time = time.time()
        wxchat.qrCallback(uuid, base64_string)
        
        while True:
            # 如果登录超时，重新获取二维码
            if time.time() - start_time > 150:  
                break
            res = wxchat.check_login(token_id, app_id, uuid)
            flag = res.get('status')
            if flag == 2:
                logger.info(f"登录成功: {res}")
                login_info = res.get('loginInfo', {})
                login_info.update({'appId': app_id, 'uuid': uuid, 'tokenId': token_id})
                
                cleaned_login_info = {k: (v if v is not None else '') for k, v in login_info.items()}
                redis_helper.redis_helper.set_hash(hash_key, cleaned_login_info)
                return login_info
            time.sleep(5)
 





def fetch_and_save_contacts2():
    """
    获取联系人列表并保存到缓存
    """
    wxchat=gewe_chat.wxchat 
    while True:

        login_keys = list(redis_helper.redis_helper.client.scan_iter(match='__AI_OPS_WX__:LOGININFO:*'))
        logger.info(f"Fetching login keys: {login_keys}")

        # 遍历每一个获取到的登录键
        for k in login_keys:
            r= redis_helper.redis_helper.get_hash(k)
            print(r)
            token_id = r.get('tokenId')
            app_id = r.get('appId')
           
            wxid = r.get('wxid')
            status=r.get('status')
            if status=='1':   
                contacts_list = wxchat.fetch_contacts_list(token_id, app_id)
                friend_wxids = contacts_list['friends'][3:]  # 可以调整截取范围
                #friend_wxids.remove('weixin')
                wxchat.save_contacts_brief_to_cache(token_id, app_id, wxid, friend_wxids)
                logger.info(f'微信ID {wxid} 登录APPID {app_id} 成功，联系人已定时保存')
                chatrooms=contacts_list['chatrooms']
                wxchat.save_groups_info_to_cache(token_id, app_id, wxid, chatrooms)
                logger.info(f'微信ID {wxid} 登录APPID {app_id} 成功，群信息已定时保存')

            else:
                logger.info(f'微信ID {wxid} 未登录 {app_id} ，联系人不能定时保存')
            
            time.sleep(3)

        time.sleep(60*10)


def start_wxchat_thread():
    # gewe_chat.start()
    scan_wx_login_info()
    # 启动同步联系人线程
    threading.Thread(target=fetch_and_save_contacts2).start()
   



def scan_wx_login_info():
    gewe_chat.start()
    wxchat = gewe_chat.wxchat
    cursor = 0
    while True:
        cursor, login_keys = redis_helper.redis_helper.client.scan(cursor, match='__AI_OPS_WX__:LOGININFO:*')
        
        for k in login_keys:
            r = redis_helper.redis_helper.get_hash(k)
            app_id=r.get("appId")
            #tel=r.get("mobile")
            token_id=r.get("tokenId")
            wxid=r.get("wxid")

            is_online = wxchat.check_online(token_id, app_id)
            if is_online:
                logger.info(f'微信ID {wxid} 在APPID {app_id} 已经在线')
            else:
                # 尝试重连
                res = wxchat.reconnection(token_id, app_id)
                if res.get('ret') == 200:
                    logger.info(f'微信ID {wxid} 在APPID {app_id} 重连成功')
                    # 同步联系人
                    #fetch_and_save_contacts(wxchat, token_id, app_id, k)  
                else:
                    print("重连失败，重新登录...")
                    print("发送离线消息到kafka")
                    redis_helper.redis_helper.update_hash_field(k,'status',0)  


            time.sleep(3)
            
        
        # 如果游标为 0，则表示扫描完成
        if cursor == 0:
            break
    

def app_run():
    
    flask_app = Flask(__name__)

    # api = Api(app)
    flask_api = Api(flask_app,errors=errors, catch_all_404s=True)

    # 设置日志（logger 已在 log.py 中配置）
    flask_app.logger.handlers.clear()  # 清除 Flask 默认的日志处理器
    flask_app.logger.addHandler(logger.handlers[1])  # 使用文件日志处理器
    flask_app.logger.setLevel(logging.DEBUG)  # 设置日志级别

    # 添加拦截器
    flask_app.before_request(before_request)
    flask_app.after_request(after_request)
    flask_app.register_error_handler(Exception, handle_exception)

    # 定义路由
    flask_api.add_resource(UserResource, '/api/user', '/api/user/<int:user_id>')
    

    flask_api.add_resource(MessagesResource, '/messages')
    
    flask_api.add_resource(DeleteFriendResource, '/api/contacts/deletefriend')
    flask_api.add_resource(GetFriendsInfoResource, '/api/contacts/getfriends')

    flask_api.add_resource(GetWxchatConfigResource, '/api/wxchat/getconfig') 
    flask_api.add_resource(SaveWxchatConfigResource, '/api/wxchat/saveconfig') 

    # 
    flask_api.add_resource(GetGroupsInfoResource, '/api/groups/getchatroominfo') 


    load_config()
    worker()

    # channel = channel_factory.create_channel('wx')
    # channel.startup()
    environment = os.environ.get('environment', 'default')  
    port=5000
    if environment== 'default':
        port=80
    flask_app.run(debug=False,host='0.0.0.0',port=port)

if __name__ == '__main__':
    app_run()