AiOpsWechatiPad/resources/config_reources.py

from flask_restful import Resource, reqparse
from pydantic import BaseModel, ValidationError
from flask import jsonify,request,json
from common import redis_helper,utils
from wechat import gewe_chat
from dataclasses import dataclass, asdict
from typing import List
import json
from model import Models



class GetWxchatConfigResource(Resource):
    def __init__(self):
        self.parser = reqparse.RequestParser()
        self.wxchat = gewe_chat.wxchat

    def post(self):
        req = request.get_json()
        wxid = req.get("wxid")

        k,loginfo=utils.get_login_info_by_wxid(wxid)  
        if not k:
            response=jsonify({"code": 404, "message": f"{wxid} 没有对应的登录信息"})
            response.status_code = 404
            return response
        
        config=self.wxchat.get_wxchat_config_from_cache(wxid)
        return jsonify(config)
    
# @dataclass
# class AgentConfig(BaseModel):
#     chatroomIdWhiteList: List[str] = []
#     agentTokenId: str
#     agentEnabled: bool
#     addContactsFromChatroomIdWhiteList: List[str] = []
#     chatWaitingMsgEnabled: bool

class SaveWxchatConfigResource(Resource):
    def __init__(self):
        self.parser = reqparse.RequestParser()
        self.wxchat = gewe_chat.wxchat

    def post(self):
        req = request.get_json()
        wxid = req.get("wxid")
        data = req.get("data")
        # hash_key="__AI_OPS_WX__:WXCHAT_CONFIG"
        # redis_helper.redis_helper.update_hash_field(hash_key, wxid, json.dumps(data,ensure_ascii=False))
        k,loginfo=utils.get_login_info_by_wxid(wxid)  
        if not k:
            response=jsonify({"code": 404, "message": f"{wxid} 没有对应的登录信息"})
            response.status_code = 404
            return response

        try:
            # 使用 Pydantic 严格校验数据类型和结构
            validated_config = Models.AgentConfig.model_validate(data)
        except ValidationError as e:

            response=jsonify({'code': 407, 'message': e.errors().__str__()})
            response.status_code = 407
            return response
        
        


        self.wxchat.save_wxchat_config(wxid, data)    
        return jsonify({'wxid': wxid, 'config': data})