kenzuya/unshackle
1
0
Fork 1
mirror of https://github.com/unshackle-dl/unshackle.git synced 2026-03-10 16:39:01 +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: if IS_WINDOWS:
import msvcrt import msvcrt
class Selector: class Selector:
""" """
A custom interactive selector class using the Rich library. A custom interactive selector class using the Rich library.
@@ -25,7 +26,7 @@ class Selector:
page_size: int = 8, page_size: int = 8,
minimal_count: int = 0, minimal_count: int = 0,
dependencies: dict[int, list[int]] = None, dependencies: dict[int, list[int]] = None,
collapse_on_start: bool = False collapse_on_start: bool = False,
): ):
""" """
Initialize the Selector. Initialize the Selector.
@@ -259,13 +260,13 @@ class Selector:
if key == b"\xe0" or key == b"\x00": if key == b"\xe0" or key == b"\x00":
try: try:
key = msvcrt.getch() key = msvcrt.getch()
if key == b"H": # Arrow Up if key == b"H": # Arrow Up
return "UP" return "UP"
if key == b"P": # Arrow Down if key == b"P": # Arrow Down
return "DOWN" return "DOWN"
if key == b"K": # Arrow Left if key == b"K": # Arrow Left
return "LEFT" return "LEFT"
if key == b"M": # Arrow Right if key == b"M": # Arrow Right
return "RIGHT" return "RIGHT"
except Exception: except Exception:
pass pass
@@ -309,27 +310,27 @@ class Selector:
# ANSI Escape Sequences for Arrow Keys # ANSI Escape Sequences for Arrow Keys
mapping = { mapping = {
"\x1b[A": "UP", # Escape + [ + A "\x1b[A": "UP", # Escape + [ + A
"\x1b[B": "DOWN", # Escape + [ + B "\x1b[B": "DOWN", # Escape + [ + B
"\x1b[C": "RIGHT", # Escape + [ + C "\x1b[C": "RIGHT", # Escape + [ + C
"\x1b[D": "LEFT", # Escape + [ + D "\x1b[D": "LEFT", # Escape + [ + D
} }
if char in mapping: if char in mapping:
return mapping[char] return mapping[char]
# Handling manual Escape sequences # Handling manual Escape sequences
if char == "\x1b": # ESC if char == "\x1b": # ESC
try: try:
next1 = click.getchar() next1 = click.getchar()
if next1 in ("[", "O"): # Sequence indicators if next1 in ("[", "O"): # Sequence indicators
next2 = click.getchar() next2 = click.getchar()
if next2 == "A": # Arrow Up if next2 == "A": # Arrow Up
return "UP" return "UP"
if next2 == "B": # Arrow Down if next2 == "B": # Arrow Down
return "DOWN" return "DOWN"
if next2 == "C": # Arrow Right if next2 == "C": # Arrow Right
return "RIGHT" return "RIGHT"
if next2 == "D": # Arrow Left if next2 == "D": # Arrow Left
return "LEFT" return "LEFT"
return "CANCEL" return "CANCEL"
except Exception: except Exception:
@@ -398,6 +399,7 @@ class Selector:
except KeyboardInterrupt: except KeyboardInterrupt:
return [] return []
def select_multiple( def select_multiple(
options: list[str], options: list[str],
minimal_count: int = 1, minimal_count: int = 1,