# -*- coding: utf-8 -*-
# @Time : 2024/11/12 16:37
# @Author : 黄子寒
# @Email : 1064071566@qq.com
# @File : script.py
# @Project : EmoLLM
import requests


def query_sensor_data(area_code: str, parameter_type: str) -> dict:
    """
    查询指定区域和数据类型的传感器数据。
    :param area_code: str, 区域代码 (如 "A" 或 "B")
    :param parameter_type: str, 数据类型 (如 "moisture"、"temperature"、"conductivity")
    :return: dict, 包含查询结果的字典
    """
    url = f"http://127.0.0.1:8000/sensors/api/{area_code}/{parameter_type}/"
    try:
        response = requests.get(url)
        response.raise_for_status()  # 检查请求是否成功
        return response.json()  # 返回 JSON 格式的数据
    except requests.exceptions.RequestException as e:
        return {"error": str(e)}


# 示例用法：可供 LLM function calling 调用
if __name__ == "__main__":
    import sys

    # 从命令行获取参数
    if len(sys.argv) < 3:
        print("请提供区域代码和数据类型参数")
    else:
        area_code = sys.argv[1]
        parameter_type = sys.argv[2]
        result = query_sensor_data(area_code, parameter_type)
        print(result)