nonebot-plugin-weather-fighter 0.4.0__py3-none-any.whl

nonebot_plugin_weather_fighter/__init__.py

@@ -0,0 +1,87 @@
+from __future__ import annotations
+
+import os
+import platform
+from pathlib import Path
+
+from nonebot import get_driver, get_plugin_config, logger, require
+from nonebot.plugin import PluginMetadata
+
+require("nonebot_plugin_localstore")
+
+_driver = get_driver()
+
+
+def _playwright_cache_dir() -> str:
+    configured = os.environ.get("PLAYWRIGHT_BROWSERS_PATH")
+    if configured:
+        return configured
+    if platform.system() == "Windows":
+        base = Path(os.environ.get("LOCALAPPDATA", Path.home() / "AppData" / "Local"))
+        return str(base / "ms-playwright")
+    if platform.system() == "Darwin":
+        return str(Path.home() / "Library" / "Caches" / "ms-playwright")
+    return str(Path.home() / ".cache" / "ms-playwright")
+
+
+if getattr(_driver.config, "render", None) is None:
+    render_config: dict[str, object] = {"provider": "playwright"}
+    executable = os.environ.get("PLAYWRIGHT_CHROMIUM_EXECUTABLE")
+    if executable:
+        render_config["provider_config"] = {
+            "executable_path": executable,
+            "launch_args": os.environ.get(
+                "PLAYWRIGHT_CHROMIUM_ARGS", "--no-sandbox --disable-crashpad"
+            ),
+        }
+    else:
+        render_config["provider_config"] = {"storage_path": _playwright_cache_dir()}
+    _driver.config.render = render_config  # type: ignore[attr-defined]
+
+require("nonebot_plugin_htmlrender")
+
+from . import consts  # noqa: E402
+from . import matcher as matcher  # noqa: E402
+from .config import Config  # noqa: E402
+from .game.engine import get_engine  # noqa: E402
+from .services.http import close_http  # noqa: E402
+from .services.openmeteo import OpenMeteoConfig, configure_client  # noqa: E402
+from .services.provinces import resolve_province  # noqa: E402
+from .services.store import close_store, init_store  # noqa: E402
+
+__plugin_meta__ = PluginMetadata(
+    name="天气战士",
+    description="老实看天气,很枯燥吧",
+    usage=consts.HELP_TEXT,
+    type="application",
+    config=Config,
+    homepage="https://github.com/qwq12738qwq/nonebot-plugin-beatsaberscore",
+    supported_adapters={"~onebot.v11"},
+)
+
+
+@_driver.on_startup
+async def _startup() -> None:
+    config = get_plugin_config(Config)
+    configure_client(
+        OpenMeteoConfig(
+            bucket_seconds=config.weather_fighter_bucket_bars,
+            stale_seconds=config.weather_fighter_stale_seconds,
+            timeout=config.weather_fighter_http_timeout,
+        )
+    )
+    await init_store()
+    province = resolve_province(config.weather_fighter_default_province)
+    if province is None:
+        logger.warning("默认省份 {} 无效，跳过启动开盘", config.weather_fighter_default_province)
+        return
+    try:
+        await get_engine().open_board(province)
+    except Exception as exc:  # noqa: BLE001 - 启动开盘失败时等首次命令重试
+        logger.opt(exception=exc).warning("启动开盘失败，首次命令时会重试")
+
+
+@_driver.on_shutdown
+async def _shutdown() -> None:
+    await close_store()
+    await close_http()

nonebot_plugin_weather_fighter/config.py

@@ -0,0 +1,25 @@
+from __future__ import annotations
+
+from pydantic import BaseModel, Field
+
+
+class Config(BaseModel):
+    """插件配置；环境变量前缀见 .env.example。"""
+
+    weather_fighter_bucket_bars: int = Field(default=1800, ge=60)
+    weather_fighter_bucket_realtime: int = Field(default=300, ge=30)
+    weather_fighter_bucket_hourly: int = Field(default=3600, ge=60)
+    weather_fighter_bucket_daily: int = Field(default=10800, ge=300)
+    weather_fighter_stale_seconds: int = Field(default=6 * 3600, ge=60)
+    weather_fighter_http_timeout: float = Field(default=30.0, gt=0)
+    weather_fighter_preload_season_series: bool = True
+    weather_fighter_caiyun_token: str = ""
+    weather_fighter_command_cooldown: float = Field(default=5.0, ge=0)
+    weather_fighter_principal: float = Field(default=10_000.0, ge=1_000)
+    weather_fighter_leverage: float = Field(default=100.0, ge=1, le=100)
+    weather_fighter_safety_entry_pct: float = Field(default=20.0, ge=1, le=100)
+    weather_fighter_safety_grant_pct: float = Field(default=5.0, gt=0, le=100)
+    weather_fighter_safety_target_pct: float = Field(default=35.0, gt=0, le=100)
+    weather_fighter_chart_bars: int = Field(default=64, ge=8, le=1000)
+    weather_fighter_enable_region: bool = True
+    weather_fighter_default_province: str = "河北"

nonebot_plugin_weather_fighter/consts.py

@@ -0,0 +1,127 @@
+from __future__ import annotations
+
+PLUGIN_NAME = "weather_fighter"
+PLUGIN_VERSION = "0.4.0"
+
+CMD_WEATHER = "天气"
+CMD_LONG = "买入"
+CMD_SHORT = "卖出"
+CMD_CLOSE = "平仓"
+CMD_ORDER = "到价"
+CMD_CANCEL = "撤单"
+CMD_LEVERAGE = "倍数"
+CMD_HELP = "帮助"
+
+DB_FILE = "game.db"
+PROVINCES_FILE = "provinces.json"
+REGIONS_FILE = "regions.json"
+
+BASE_PRICE = 1000.0
+BARS_PER_DAY = 16
+OPENMETEO_PAST_DAYS = 92
+
+WEATHER_NAMES = {
+    0: "晴",
+    1: "基本晴",
+    2: "多云",
+    3: "阴",
+    45: "雾",
+    48: "雾凇",
+    51: "毛毛雨",
+    53: "毛毛雨",
+    55: "毛毛雨",
+    56: "冻毛雨",
+    57: "冻毛雨",
+    61: "小雨",
+    63: "中雨",
+    65: "大雨",
+    66: "冻雨",
+    67: "冻雨",
+    71: "小雪",
+    73: "中雪",
+    75: "大雪",
+    77: "米雪",
+    80: "阵雨",
+    81: "阵雨",
+    82: "强阵雨",
+    85: "阵雪",
+    86: "强阵雪",
+    95: "雷暴",
+    96: "雷暴夹雹",
+    99: "强雷暴",
+}
+
+WEATHER_ICONS = {
+    0: "☀",
+    1: "🌤",
+    2: "⛅",
+    3: "☁",
+    45: "🌫",
+    48: "🌫",
+    51: "🌦",
+    53: "🌦",
+    55: "🌦",
+    56: "🌧",
+    57: "🌧",
+    61: "🌧",
+    63: "🌧",
+    65: "🌧",
+    66: "🌧",
+    67: "🌧",
+    71: "🌨",
+    73: "🌨",
+    75: "❄",
+    77: "🌨",
+    80: "🌦",
+    81: "🌧",
+    82: "⛈",
+    85: "🌨",
+    86: "❄",
+    95: "⛈",
+    96: "⛈",
+    99: "⛈",
+}
+
+_HELP_TEXT = """天气战士 · 所有群共用一个盘
+
+主图 + 天气（天气 / 行情 / 大盘 / K线 同一命令，不用开盘）：
+  天气 [省份] / 天气 [市级地]          ——  主图 + 现在天气 + 今日高低温
+
+交易（按真实时间自动运行，不需要开盘）：
+  买入 [省份] 1000 / 卖出 河北 1万    ——  顾名思义
+  买入 [省份] 1000 杠杆 5             ——  下单时指定倍数
+  卖出 [省份]                         ——  不写金额默认全卖出
+  金额：1000 / 1.5万 / 20% / 全；不填默认 20%
+  到价 [省份] 16.5 买入 1000         ——  价格碰到 16.5℃ 自动买入
+  到价 [省份] 16.5 卖出 1000         ——  价格碰到 16.5℃ 自动卖出
+  到价 [省份] 18.0 平                ——  价格碰到 18.0℃ 自动平掉河北
+  平仓 [省]                        ——  平掉该省；不带省则平掉全部持仓
+  撤单                              ——  取消自己的到价单
+  倍数 10                           ——  设置杠杆倍数（1~上限，有持仓时不能改）
+
+规则：
+  · 能吃保底
+  · 每个玩家自己的余额、持仓、到价单互相独立。
+  · 倍数默认 1 倍，可下单时带「杠杆 5」或先用「倍数 5」设置；
+  · 按照 A 股时间：09:30-11:30、13:00-15:00；午休和周末停盘。
+  · 默认余额 10,000。低于 20%（2,000）时进入保底：每个真实自然日补 5%（500），
+"""
+
+HELP_TEXT = _HELP_TEXT
+
+
+def weather_name(code: int) -> str:
+    return WEATHER_NAMES.get(code, f"码 {code}")
+
+
+def weather_icon(code: int) -> str:
+    return WEATHER_ICONS.get(code, "🌡")
+
+
+def wind_level(gust: float) -> int:
+    """把阵风 m/s 粗换成风力等级，给玩家一个直观数字。"""
+    thresholds = (0.3, 1.6, 3.4, 5.5, 8.0, 10.8, 13.9, 17.2, 20.8, 24.5, 28.5, 32.7)
+    for level, limit in enumerate(thresholds):
+        if gust < limit:
+            return level
+    return 12

nonebot_plugin_weather_fighter/game/__init__.py

@@ -0,0 +1 @@
+"""游戏引擎（会话推进、结算）在 M1 接入。"""

nonebot_plugin_weather_fighter/game/amount.py

@@ -0,0 +1,47 @@
+from __future__ import annotations
+
+UNITS: tuple[tuple[str, float], ...] = (
+    ("亿", 100_000_000.0),
+    ("万", 10_000.0),
+    ("w", 10_000.0),
+    ("W", 10_000.0),
+    ("千", 1_000.0),
+    ("k", 1_000.0),
+    ("K", 1_000.0),
+)
+ALL_IN = {"全", "全仓", "梭哈", "all", "ALL"}
+
+
+class AmountError(ValueError):
+    pass
+
+
+def parse_amount(text: str | None, cash: float, *, default_pct: float = 20.0) -> float:
+    """把“1000 / 1.5万 / 30% / 全”解析成投入金额，并校验不超过余额。"""
+    raw = (text or "").strip().replace(",", "").replace("，", "").replace("元", "")
+    if not raw:
+        amount = cash * default_pct / 100.0
+    elif raw in ALL_IN:
+        amount = cash
+    elif raw.endswith("%"):
+        try:
+            percent = float(raw[:-1])
+        except ValueError as exc:
+            raise AmountError("金额格式不对，例如：1000 / 1万 / 30% / 全") from exc
+        amount = cash * percent / 100.0
+    else:
+        scale = 1.0
+        for unit, multiplier in UNITS:
+            if raw.endswith(unit):
+                raw = raw[: -len(unit)]
+                scale = multiplier
+                break
+        try:
+            amount = float(raw) * scale
+        except ValueError as exc:
+            raise AmountError("金额格式不对，例如：1000 / 1万 / 30% / 全") from exc
+    if amount <= 0:
+        raise AmountError("还有反向消费?")
+    if amount > cash + 1e-6:
+        raise AmountError(f"穷了,只有 {cash:,.0f}")
+    return amount