jwf-common-python 1.0.0__py3-none-any.whl

jwf_common_python/weather/amap.py

@@ -0,0 +1,127 @@
+import datetime
+
+import requests
+
+
+class WeatherAmap:
+    def __init__(self, key: str):
+        """
+        city: 城市的 adcode，adcode
+        extensions: base 返回实时天气, all 返回预报天气
+        """
+        self.key = key
+        self.base_url = (
+            f"https://restapi.amap.com/v3/weather/weatherInfo?key={self.key}"
+        )
+
+    def _send_request(self, url: str) -> dict:
+        response = requests.get(url)
+        if response.status_code != 200:
+            raise Exception("请求高德地图API失败")
+        res = response.json()
+
+        # 1. 校验响应状态
+        status = res["status"]
+        assert status == "1", f"请求高德地图API失败，status: {status}"
+        info = res.get("info")
+        assert info == "OK", f"请求高德地图API失败，info: {info}"
+        infocode = res.get("infocode")
+        assert infocode == "10000", f"请求高德地图API失败，infocode: {infocode}"
+
+        return res
+
+    def get_current_weather(self, city_name: str, city_adcode: str):
+        """获取实时/当前天气
+        实况天气每小时更新多次，预报天气每天更新3次，分别在8、11、18点左右更新。
+
+        返回结构:
+         {
+            "province": "河南",
+            "city": "新安县",
+            "adcode": "410323",
+            "weather": "晴",
+            "temperature": "19",
+            "winddirection": "东南",
+            "windpower": "≤3",
+            "humidity": "36",
+            "reporttime": "2024-10-23 14:00:05",
+            "temperature_float": "19.0",
+            "humidity_float": "36.0"
+        }
+        """
+        url = f"{self.base_url}&city={city_name}&extensions=base"
+        res = self._send_request(url)
+        lives_data = res["lives"]
+
+        # check1 结构检查
+        assert len(lives_data) == 1
+        live_data = lives_data[0]
+
+        assert live_data["city"] == city_name
+        assert live_data["adcode"] == city_adcode
+
+        return live_data
+
+    def get_forecast_weather(
+        self, city_name: str, city_adcode: str, isTodayForecast: bool = True
+    ) -> list[dict]:
+        """获取预报/未来天气
+        city_name: 城市名
+        adcode: 城市的 adcode
+        isTodayForecast: 是否获取今天的天气预报
+            - True 表示只获取今天的预报天气。list长度为1,仅有今天的数据. [默认值]
+            - False 表示获取未来3天的预报天气。list长度为4
+
+        返回的数据结构:
+        {
+            "date": "2024-10-23",
+            "week": "3",
+            "dayweather": "晴",
+            "nightweather": "晴",
+            "daytemp": "20",
+            "nighttemp": "9",
+            "daywind": "东",
+            "nightwind": "东",
+            "daypower": "1-3",
+            "nightpower": "1-3",
+            "daytemp_float": "20.0",
+            "nighttemp_float": "9.0"
+        }
+        """
+        url = f"{self.base_url}&city={city_name}&extensions=all"
+        res = self._send_request(url)
+        forecasts_data = res["forecasts"]
+
+        # check1 结构检查
+        assert len(forecasts_data) == 1
+        forecast_data = forecasts_data[0]
+
+        # check2 数据准确性检查
+        city_res = forecast_data["city"]
+        assert city_res == city_name, f"请求的城市名与返回的城市名不一致: {city_res} != {city_name}"
+        adcode_res = forecast_data["adcode"]
+        assert (
+            adcode_res == city_adcode
+        ), f"请求的城市adcode与返回的城市adcode不一致: {adcode_res} != {city_adcode}"
+
+        # 解析核心数据
+        forecast_core_res = forecast_data["casts"]
+
+        if isTodayForecast:
+            # 只获取今天的天气预报
+
+            # 1. 获取今天的日期字符串
+            today_datetime: datetime.datetime = datetime.datetime.now()
+            today_date: str = today_datetime.strftime("%Y-%m-%d")
+
+            # 2. 获取今天的预报天气数据
+            today_forecast_res = forecast_core_res[0]
+
+            # 断言，校验数据获取是否准确
+            today_forecast_date = today_forecast_res["date"]
+            _msg = f"获取到的预报数据的日期不是今天: 预测日期:'{today_forecast_date}' 实际日期:'{today_date}'"
+            assert today_date == today_forecast_date, _msg
+            return [today_forecast_res]
+
+        # 获取未来3天的预报天气
+        return forecast_core_res

jwf_common_python/weather/wanwei.py

@@ -0,0 +1,58 @@
+# 【万维易源服务商】
+import requests
+
+
+class WeatherWanwei:
+    """万维易源历史天气数据查询
+    https://market.aliyun.com/apimarket/detail/cmapi010812
+    """
+
+    def __init__(self, key: str):
+        self.key = key
+        self.base_url = "https://ali-weather.showapi.com/weatherhistory"
+
+    def get_history_weather(
+        self, city_name: str, city_adcode: str, year_month: str
+    ) -> list[dict]:
+        """获取指定地区、月份的历史天气数据
+        :param area: 地区名称，如新安县
+        :param month: 年月，格式为YYYYMM， 201601
+
+        :return: 历史天气数据
+         {
+                "aqiLevel": "6",
+                "wind_direction": "西南风",
+                "aqi": "406",
+                "wind_power": "微风",
+                "area": "新安",
+                "time": "20160101",
+                "min_temperature": "1",
+                "max_temperature": "12",
+                "aqiInfo": "严重",
+                "weather": "晴"
+        }
+        """
+        url = f"{self.base_url}?area={city_name}&month={year_month}"
+        headers = {"Authorization": f"APPCODE {self.key}"}
+        params = {"area": city_name, "month": year_month}
+        response = requests.get(url, headers=headers, params=params)
+
+        # check 1
+        if response.status_code != 200:
+            raise Exception(f"请求历史天气数据失败，状态码：{response.status_code}")
+        res = response.json()
+
+        # check2
+        showapi_res_error = res["showapi_res_error"]
+        if showapi_res_error != "":
+            raise Exception(f"请求历史天气数据失败，错误码：{showapi_res_error}")
+
+        # check3
+        showapi_res_body = res["showapi_res_body"]
+        remark = showapi_res_body["remark"]
+        areaCode_res = showapi_res_body["areaCode"]
+        if remark != "查询成功！" or areaCode_res != city_adcode:
+            raise Exception(f"请求历史天气数据失败，错误信息：{remark} {areaCode_res}")
+
+        data = showapi_res_body["list"]
+        return data

jwf_common_python/youdao/__init__.py

@@ -0,0 +1,13 @@
+"""有道单词本导出和单词过滤工具。"""
+
+from .wordbook import (
+    fetch_all_wordbook_data,
+    fetch_wordbook_data,
+    filter_new_words,
+)
+
+__all__ = [
+    "fetch_wordbook_data",
+    "fetch_all_wordbook_data",
+    "filter_new_words",
+]

jwf_common_python/youdao/wordbook.py

@@ -0,0 +1,88 @@
+"""有道单词本工具；不读写文件，不打印，不在导入时发起请求。"""
+
+from __future__ import annotations
+
+from collections.abc import Iterable
+from typing import Any
+
+import requests
+
+WORDBOOK_URL = "https://youdao.com/wordbook/webapi/words"
+
+
+def fetch_wordbook_data(
+    cookie: str,
+    limit: int = 15,
+    offset: int = 0,
+    timeout: float = 30,
+) -> dict[str, Any]:
+    """获取一页单词本，返回完整 JSON，保留所有响应字段。
+
+    cookie 为浏览器请求头中的完整 Cookie 字符串。
+    HTTP 错误、超时、JSON 解析失败直接抛出 requests 对应异常；
+    接口返回失败状态时抛出 ValueError。
+    """
+    if not cookie.strip():
+        raise ValueError("cookie 不能为空")
+    if limit <= 0:
+        raise ValueError("limit 必须大于 0")
+    if offset < 0:
+        raise ValueError("offset 不能小于 0")
+
+    response = requests.get(
+        WORDBOOK_URL,
+        params={"limit": limit, "offset": offset},
+        headers={"Cookie": cookie},
+        timeout=timeout,
+    )
+    response.raise_for_status()
+    data = response.json()
+    if not isinstance(data, dict) or data.get("msg") != "SUCCESS":
+        raise ValueError("获取有道单词本失败，请检查登录状态和 Cookie")
+    return data
+
+
+def fetch_all_wordbook_data(
+    cookie: str, limit: int = 15, timeout: float = 30
+) -> list[dict[str, Any]]:
+    """按页导出整个单词本，返回每页完整 JSON 响应组成的列表。
+
+    单词及释义等原始字段位于每页的 data.itemList 中，不提取或删减字段。
+    中途请求失败时抛出异常，不将部分结果当作完整导出返回。
+    """
+    pages = []
+    offset = 0
+    while True:
+        page = fetch_wordbook_data(cookie, limit=limit, offset=offset, timeout=timeout)
+        data = page.get("data")
+        if not isinstance(data, dict) or not isinstance(data.get("itemList"), list):
+            raise ValueError("有道单词本响应缺少 data.itemList")
+        total = data.get("total")
+        if not isinstance(total, int) or total < 0:
+            raise ValueError("有道单词本响应中的 total 无效")
+
+        pages.append(page)
+        items = data["itemList"]
+        offset += len(items)
+        if not items and offset < total:
+            raise ValueError("有道单词本尚未导出完毕，但接口返回了空页")
+        if offset >= total:
+            return pages
+
+
+def filter_new_words(
+    existing_words: Iterable[str], current_words: Iterable[str]
+) -> list[str]:
+    """返回当前词表中尚未存在的单词，按当前顺序去重。
+
+    两个参数均为单词字符串的可迭代对象；多行文本请先调用 splitlines()。
+    去除首尾空白、忽略空词，保留大小写差异，不修改传入的词表。
+    """
+    seen = {word.strip() for word in existing_words}
+    new_words = []
+    for word in current_words:
+        word = word.strip()
+        if word and word not in seen:
+            seen.add(word)
+            new_words.append(word)
+    return new_words