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



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")
        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))


        try:
            # 使用 Pydantic 严格校验数据类型和结构
            validated_config = 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})