Use fixed values for unchanging dict to replace split and for loops

This commit is contained in:
pagedown 2022-01-22 11:36:04 +08:00
parent a47aca6fdb
commit 7ef73babdd
No known key found for this signature in database
GPG key ID: E921CF18AC8FF6EB
2 changed files with 6 additions and 13 deletions

View file

@ -166,11 +166,8 @@ def scroll_screen(amt: int = 1) -> str:
return '\033[' + str(abs(amt)) + ('T' if amt < 0 else 'S')
STANDARD_COLORS = {name: i for i, name in enumerate(
'black red green yellow blue magenta cyan gray'.split())}
STANDARD_COLORS['white'] = STANDARD_COLORS['gray']
UNDERLINE_STYLES = {name: i + 1 for i, name in enumerate(
'straight double curly dotted dashed'.split())}
STANDARD_COLORS = {'black': 0, 'red': 1, 'green': 2, 'yellow': 3, 'blue': 4, 'magenta': 5, 'cyan': 6, 'gray': 7, 'white': 7}
UNDERLINE_STYLES = {'straight': 1, 'double': 2, 'curly': 3, 'dotted': 4, 'dashed': 5}
ColorSpec = Union[int, str, Color]

View file

@ -500,18 +500,14 @@ def scrollback_pager_history_size(x: str) -> int:
return min(ans, 4096 * 1024 * 1024 - 1)
# "single" for backwards compat
url_style_map = {'none': 0, 'single': 1, 'straight': 1, 'double': 2, 'curly': 3, 'dotted': 4, 'dashed': 5}
def url_style(x: str) -> int:
# for backwards compat
if x == 'single':
x = 'straight'
return url_style_map.get(x, url_style_map['curly'])
url_style_map = {
v: i for i, v in enumerate('none straight double curly dotted dashed'.split())
}
def url_prefixes(x: str) -> Tuple[str, ...]:
return tuple(a.lower() for a in x.replace(',', ' ').split())