kenzuya/unshackle
1
0
Fork 1
mirror of https://github.com/unshackle-dl/unshackle.git synced 2026-03-10 08:29:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update selector.py

Browse Source
This commit is contained in:
CodeName393
2026-02-26 15:25:09 +09:00
committed by GitHub
parent 1611fcc971
commit c5ef13df5d
1 changed files with 17 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
unshackle/core/utils/selector.py
View File

@@ -11,6 +11,7 @@ IS_WINDOWS = sys.platform == "win32"
if IS_WINDOWS:
import msvcrt
class Selector:
"""
A custom interactive selector class using the Rich library.
@@ -25,7 +26,7 @@ class Selector:
page_size: int = 8,
minimal_count: int = 0,
dependencies: dict[int, list[int]] = None,
collapse_on_start: bool = False
collapse_on_start: bool = False,
):
"""
Initialize the Selector.
@@ -259,13 +260,13 @@ class Selector:
if key == b"\xe0" or key == b"\x00":
try:
key = msvcrt.getch()
if key == b"H": # Arrow Up
if key == b"H": # Arrow Up
return "UP"
if key == b"P": # Arrow Down
if key == b"P": # Arrow Down
return "DOWN"
if key == b"K": # Arrow Left
if key == b"K": # Arrow Left
return "LEFT"
if key == b"M": # Arrow Right
if key == b"M": # Arrow Right
return "RIGHT"
except Exception:
pass
@@ -309,27 +310,27 @@ class Selector:
# ANSI Escape Sequences for Arrow Keys
mapping = {
"\x1b[A": "UP", # Escape + [ + A
"\x1b[B": "DOWN", # Escape + [ + B
"\x1b[C": "RIGHT", # Escape + [ + C
"\x1b[D": "LEFT", # Escape + [ + D
"\x1b[A": "UP", # Escape + [ + A
"\x1b[B": "DOWN", # Escape + [ + B
"\x1b[C": "RIGHT", # Escape + [ + C
"\x1b[D": "LEFT", # Escape + [ + D
}
if char in mapping:
return mapping[char]
# Handling manual Escape sequences
if char == "\x1b": # ESC
if char == "\x1b": # ESC
try:
next1 = click.getchar()
if next1 in ("[", "O"): # Sequence indicators
if next1 in ("[", "O"): # Sequence indicators
next2 = click.getchar()
if next2 == "A": # Arrow Up
if next2 == "A": # Arrow Up
return "UP"
if next2 == "B": # Arrow Down
if next2 == "B": # Arrow Down
return "DOWN"
if next2 == "C": # Arrow Right
if next2 == "C": # Arrow Right
return "RIGHT"
if next2 == "D": # Arrow Left
if next2 == "D": # Arrow Left
return "LEFT"
return "CANCEL"
except Exception:
@@ -398,6 +399,7 @@ class Selector:
except KeyboardInterrupt:
return []
def select_multiple(
options: list[str],
minimal_count: int = 1,