suggest_stock
Search for Chinese stock symbols using keywords to identify relevant companies on the Xueqiu platform.
Instructions
关键词搜索股票代码
Args:
keyword: 关键词Input Schema
TableJSON Schema
| Name | Required | Description | Default |
|---|---|---|---|
| keyword | No | SZ000002 |
Implementation Reference
- main.py:497-505 (handler)The handler function for the 'suggest_stock' tool. It is decorated with @mcp.tool() for registration and executes the core logic by calling the underlying pysnowball library's suggest_stock method with the provided keyword, then processes the result using process_data.
@mcp.tool() def suggest_stock(keyword: str="SZ000002") -> dict: """关键词搜索股票代码 Args: keyword: 关键词 """ result = ball.suggest_stock(keyword) return process_data(result) - main.py:34-61 (helper)Helper function used by suggest_stock (and other tools) to process the raw data from pysnowball, including converting timestamps to readable datetime strings.
def process_data(data, process_config=None): """ 通用数据处理函数，可扩展添加各种数据处理操作 Args: data: 原始数据 process_config: 处理配置字典，用于指定要执行的处理操作 例如: {'convert_timestamps': True, 'other_process': params} Returns: 处理后的数据 """ if process_config is None: # 默认配置 process_config = { 'convert_timestamps': True } # 如果开启了时间戳转换 if process_config.get('convert_timestamps', True): data = convert_timestamps(data) # 在这里可以添加更多的数据处理逻辑 # 例如: # if 'format_numbers' in process_config: # data = format_numbers(data, **process_config['format_numbers']) return data - main.py:14-31 (helper)Supporting helper recursively called by process_data to convert timestamp fields in the data to human-readable datetime strings.
def convert_timestamps(data): """递归地将数据中的所有 timestamp 转换为 datetime 字符串""" if isinstance(data, dict): for key, value in list(data.items()): if key == 'timestamp' and isinstance(value, (int, float)) and value > 1000000000000: # 毫秒级时间戳 data[key] = datetime.datetime.fromtimestamp(value/1000).strftime('%Y-%m-%d %H:%M:%S') elif key == 'timestamp' and isinstance(value, (int, float)) and value > 1000000000: # 秒级时间戳 data[key] = datetime.datetime.fromtimestamp(value).strftime('%Y-%m-%d %H:%M:%S') elif key.endswith('_date') and isinstance(value, (int, float)) and value > 1000000000000: # 毫秒级时间戳 data[key] = datetime.datetime.fromtimestamp(value/1000).strftime('%Y-%m-%d %H:%M:%S') elif key.endswith('_date') and isinstance(value, (int, float)) and value > 1000000000: # 秒级时间戳 data[key] = datetime.datetime.fromtimestamp(value).strftime('%Y-%m-%d %H:%M:%S') elif isinstance(value, (dict, list)): data[key] = convert_timestamps(value) elif isinstance(data, list): for i, item in enumerate(data): data[i] = convert_timestamps(item) return data