chat_history_collection/user_api.py

import requests
from logger_config import logger
import os
import dotenv
dotenv.load_dotenv()
class UserCollector:

    def __init__(self):
        self.base_url = os.getenv('USER_URL', '')
        token = os.getenv('JWT_TOKEN', '')
        self.headers = {
            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/146.0.0.0 Safari/537.36',
            'JWT-TOKEN': f'{token}'
        }

    def get_user_info(self):
        """访问用户API并获取返回值"""
        try:
            response = requests.get(self.base_url, headers=self.headers)
            response.raise_for_status()  # 检查请求是否成功
            data = response.json()
            if data.get('code') == 200:
                user_names = [user.get('UserName', 'N/A') for user in data.get('data', {}).get('list', [])]
                logger.info(f"获取用户信息成功, 共计{len(user_names)}条")
                return user_names
            else:
                logger.Error(f"获取用户信息失败, 状态码：{data.get('code')}")
                return []
        except requests.exceptions.RequestException as e:
            logger.error(f"获取用户发生错误: {e}")
            return []


if __name__ == "__main__":
    user_collector = UserCollector()
    user_info = user_collector.get_user_info()
    pass