snowy-yang
fcd4cb4649
配置见 pyproject.toml：
- line-length=100：贴着现有代码定。若用默认的 88 会重排 122 行，
噪音淹没真正有价值的修复。
- 启用 E/W/F/I/UP/B/C4/SIM/RET/PERF/RUF；**不**启用 PTH。因为本项目的路径
多是"字符串身份"——要塞进 subprocess 命令行、写进 JSON 存档、打印给用户，
全换 pathlib 只是包一层再拆一层，换来 70 处纯风格改动。
- 忽略 RUF001/002/003：中文全角标点是正确写法，不是"歧义字符"。
- 忽略 PERF203：剩下两处都是"逐项容错"语义（逐个牧场存档要立刻上报失败、
收获全部要跳过失败地块）。把 try 提到循环外会毁掉"一项失败不影响其它项"
这个行为，是错的优化。
自动修复 358 项：Dict/List/Optional 换成 PEP 585/604 写法、导入排序、
失效的 noqa、空占位 f-string 等。
手工修复的（这些是真问题，不是风格偏好）：
- B904 x9：except 中 raise 补异常链。刻意区分——网络/IO 失败用 from exc 保留
根因；参数解析失败用 from None，因为底层 ValueError 对玩家只是噪音。
- RUF012 x2：StateView 的 plots/animals 原本是**类级共享可变列表**，
改成在 __init__ 里初始化（apply({}) 会提前返回，所以必须显式摆好）。
- F841：TUI 里一个赋值后再没用过的死变量。
- SIM105 x11：try/except/pass 统一改成 contextlib.suppress。
- SIM115：日志句柄改用 with。Popen 已把 fd 复制给子进程，父进程可安全关闭。
- E501 x7：拆超长行；main.py 那处顺手提取变量，避免超长嵌套表达式。
- E741：把 l/r 改成 left_line/right_line。
- 其余：RUF046 / PERF401 / RET504 / RUF005 / UP031 / B007。
ruff format 重排 24 个文件。
验证：97 项测试全绿。
142 lines
3.9 KiB
Python
142 lines
3.9 KiB
Python
"""跨平台非阻塞键盘读取。
|
||
|
||
Windows 用 ``msvcrt``，POSIX 用 ``termios`` + ``select``，
|
||
两端统一返回规范化的按键名，TUI 层无需关心平台差异。
|
||
|
||
在非交互式环境（管道、重定向）下 ``available`` 为 False，
|
||
TUI 会退化为无输入的只读模式。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import os
|
||
import sys
|
||
|
||
IS_WINDOWS = os.name == "nt"
|
||
|
||
# Windows 下的扩展键（以 \x00 / \xe0 开头）
|
||
_WIN_EXTENDED = {
|
||
"H": "UP",
|
||
"P": "DOWN",
|
||
"K": "LEFT",
|
||
"M": "RIGHT",
|
||
"G": "HOME",
|
||
"O": "END",
|
||
"S": "DELETE",
|
||
}
|
||
|
||
_WIN_CTRL = {
|
||
"\x03": "CTRL_C",
|
||
"\r": "ENTER",
|
||
"\n": "ENTER",
|
||
"\t": "TAB",
|
||
"\x08": "BACKSPACE",
|
||
"\x1b": "ESC",
|
||
" ": "SPACE",
|
||
}
|
||
|
||
|
||
class KeyReader:
|
||
"""把终端的原始字节流转换成按键名。"""
|
||
|
||
def __init__(self) -> None:
|
||
self.available = sys.stdin is not None and sys.stdin.isatty()
|
||
self._fd: int | None = None
|
||
self._saved_term = None
|
||
|
||
if self.available and not IS_WINDOWS:
|
||
try:
|
||
import termios
|
||
import tty
|
||
|
||
self._fd = sys.stdin.fileno()
|
||
self._saved_term = termios.tcgetattr(self._fd)
|
||
# cbreak: 立即返回按键，但保留 Ctrl-C 等信号
|
||
tty.setcbreak(self._fd)
|
||
except Exception:
|
||
self.available = False
|
||
self._saved_term = None
|
||
|
||
if self.available and IS_WINDOWS:
|
||
try:
|
||
import msvcrt # noqa: F401
|
||
except Exception:
|
||
self.available = False
|
||
|
||
# ------------------------------------------------------------------ 读取
|
||
def poll(self) -> str | None:
|
||
"""非阻塞读取一个按键；没有按键时返回 None。"""
|
||
if not self.available:
|
||
return None
|
||
return self._poll_windows() if IS_WINDOWS else self._poll_posix()
|
||
|
||
def _poll_windows(self) -> str | None:
|
||
import msvcrt
|
||
|
||
if not msvcrt.kbhit():
|
||
return None
|
||
char = msvcrt.getwch()
|
||
|
||
if char in ("\x00", "\xe0"):
|
||
if not msvcrt.kbhit():
|
||
return None
|
||
return _WIN_EXTENDED.get(msvcrt.getwch().upper())
|
||
if char in _WIN_CTRL:
|
||
return _WIN_CTRL[char]
|
||
return char.lower()
|
||
|
||
def _poll_posix(self) -> str | None:
|
||
import select
|
||
|
||
if self._fd is None:
|
||
return None
|
||
try:
|
||
ready, _, _ = select.select([self._fd], [], [], 0)
|
||
except (OSError, ValueError):
|
||
return None
|
||
if not ready:
|
||
return None
|
||
try:
|
||
data = os.read(self._fd, 8)
|
||
except OSError:
|
||
return None
|
||
if not data:
|
||
return None
|
||
|
||
text = data.decode("utf-8", errors="ignore")
|
||
if not text:
|
||
return None
|
||
|
||
# 方向键：ESC [ A/B/C/D，或 ESC O A/B/C/D
|
||
if text.startswith("\x1b["):
|
||
return {"A": "UP", "B": "DOWN", "C": "RIGHT", "D": "LEFT"}.get(text[2:3])
|
||
if text.startswith("\x1bO"):
|
||
return {"A": "UP", "B": "DOWN", "C": "RIGHT", "D": "LEFT"}.get(text[2:3])
|
||
|
||
char = text[0]
|
||
mapping = {
|
||
"\x03": "CTRL_C",
|
||
"\r": "ENTER",
|
||
"\n": "ENTER",
|
||
"\t": "TAB",
|
||
"\x7f": "BACKSPACE",
|
||
"\x08": "BACKSPACE",
|
||
"\x1b": "ESC",
|
||
" ": "SPACE",
|
||
}
|
||
if char in mapping:
|
||
return mapping[char]
|
||
return char.lower()
|
||
|
||
# ------------------------------------------------------------------ 清理
|
||
def close(self) -> None:
|
||
if self._fd is not None and self._saved_term is not None:
|
||
try:
|
||
import termios
|
||
|
||
termios.tcsetattr(self._fd, termios.TCSADRAIN, self._saved_term)
|
||
except Exception:
|
||
pass
|
||
self._fd = None
|
||
self._saved_term = None
|