import requests

API_KEY = "sk_xxxxxxxxxxxxxxxx"   # 到 https://data.epmap.org/api_key 创建
BASE = "https://mcp.epmap.org/v1/province_air/api"
# 下面两种写法等价，跑其中一种即可（两段都跑会各扣一次费）

# 方式一：GET，参数走 query string
resp = requests.get(
    f"{BASE}/get_station_realtime",
    headers={"Authorization": f"Bearer {API_KEY}"},
    params={
        "province": "河北省",
        "city": "保定市",
        "station": "华电二区",
        "start_time": "2026-10-01 10:00:00",
        "end_time": "2026-10-31 10:00:00",
        "page": "1",
    },
    timeout=30,
)
print(resp.status_code)
print(resp.json())

# 方式二：POST，参数走 JSON body（写 json=，不要用 data= 传表单）
resp = requests.post(
    f"{BASE}/get_station_realtime",
    headers={"Authorization": f"Bearer {API_KEY}"},
    json={
        "province": "河北省",
        "city": "保定市",
        "station": "华电二区",
        "start_time": "2026-10-01 10:00:00",
        "end_time": "2026-10-31 10:00:00",
        "page": "1",
    },
    timeout=30,
)
print(resp.status_code)
print(resp.json())

import asyncio
from mcp import ClientSession
from mcp.client.streamable_http import streamablehttp_client

API_KEY = "sk_xxxxxxxxxxxxxxxx"
URL = "https://mcp.epmap.org/v1/province_air/mcp"

async def main():
    async with streamablehttp_client(
        URL, headers={"Authorization": f"Bearer {API_KEY}"}
    ) as (read, write, _):
        async with ClientSession(read, write) as session:
            await session.initialize()

            # 先看服务端提供了哪些工具
            listed = await session.list_tools()
            print([t.name for t in listed.tools])

            result = await session.call_tool(
                "get_station_realtime",
                {
                    "province": "河北省",
                    "city": "保定市",
                    "station": "华电二区",
                    "start_time": "2026-10-01 10:00:00",
                    "end_time": "2026-10-31 10:00:00",
                    "page": "1",
                },
            )
            print(result.structuredContent or result.content[0].text)

asyncio.run(main())

curl -G "https://mcp.epmap.org/v1/province_air/api/get_station_realtime" \
  -H "Authorization: Bearer sk_xxxxxxxxxxxxxxxx" \
  --data-urlencode 'province=河北省' \
  --data-urlencode 'city=保定市' \
  --data-urlencode 'station=华电二区' \
  --data-urlencode 'start_time=2026-10-01 10:00:00' \
  --data-urlencode 'end_time=2026-10-31 10:00:00' \
  --data-urlencode 'page=1'

{
  "mcpServers": {
    "epmap-province_air": {
      "type": "streamable-http",
      "url": "https://mcp.epmap.org/v1/province_air/mcp",
      "headers": {
        "Authorization": "Bearer sk_xxxxxxxxxxxxxxxx"
      }
    }
  }
}