Code improvements, line numbering fix, reformatting (with Black)
This commit is contained in:
parent
ec63eb4e58
commit
d9c8ccabae
GPG key ID:
EF8E44AF715C28A0
1 changed files with 69 additions and 69 deletions
138
pat
138
pat
|
|
@ -1,98 +1,98 @@
|
|||
#!/usr/bin/env python3
|
||||
from enum import IntEnum
|
||||
from sys import argv, stdin, stderr
|
||||
from sys import argv, stdin
|
||||
from string import whitespace
|
||||
import argparse
|
||||
import os
|
||||
import string
|
||||
|
||||
|
||||
# If the program is using too much memory, try decreasing this value
|
||||
# If some text is getting cut off, try increasing this value
|
||||
bytes_to_read = 1000000
|
||||
|
||||
first_file = True
|
||||
|
||||
line_str = lambda line: str(line).rjust(8, ' ') + ' '
|
||||
line_str = lambda line: str(line).rjust(6, " ") + " "
|
||||
color_print = lambda red, green, blue, string: print(
|
||||
f"\x1b[38;2;{red};{green};{blue}m{string}",
|
||||
end='')
|
||||
f"\033[38;2;{red};{green};{blue}m{string}", end=""
|
||||
)
|
||||
|
||||
parser = argparse.ArgumentParser(description="Output text from files with color.")
|
||||
parser.add_argument("-n", "--number",
|
||||
action="store_true",
|
||||
help="number all output lines")
|
||||
parser.add_argument("-m", "--max_rgb_value",
|
||||
type=float, nargs='?', default=255,
|
||||
help="max value any r/g/b can have. 0 <= argument <= 255")
|
||||
parser.add_argument("-c", "--color_amount",
|
||||
type=int, nargs='?', default=300,
|
||||
help="amount of colors to output. arg is divisible by 3, 0 <= arg <= max * 3.")
|
||||
parser.add_argument("paths",
|
||||
nargs='*', default=['-'],
|
||||
help="a path to open ('-' for stdin)")
|
||||
parser.add_argument(
|
||||
"-n", "--number", action="store_true", help="number all output lines"
|
||||
)
|
||||
parser.add_argument(
|
||||
"-m",
|
||||
"--max_rgb_value",
|
||||
type=float,
|
||||
nargs="?",
|
||||
default=255,
|
||||
help="max value any r/g/b can have. 0 <= argument <= 255",
|
||||
)
|
||||
parser.add_argument(
|
||||
"-c",
|
||||
"--color_amount",
|
||||
type=int,
|
||||
nargs="?",
|
||||
default=300,
|
||||
help="amount of colors to output. arg is divisible by 3, 0 <= arg <= max * 3.",
|
||||
)
|
||||
parser.add_argument(
|
||||
"paths", nargs="*", default=["-"], help="a path to open ('-' for stdin)"
|
||||
)
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
# Checking arguments are valid
|
||||
assert (
|
||||
0 <= args.max_rgb_value <= 255 and
|
||||
args.color_amount <= args.max_rgb_value * 3 and
|
||||
not args.color_amount % 3), f"Invalid arguments. Run `{argv[0]} -h` for help."
|
||||
|
||||
class ColorInfo(IntEnum):
|
||||
AMOUNT = args.color_amount
|
||||
MAX_RGB_VALUE = args.max_rgb_value
|
||||
PER_STAGE = args.color_amount / 3
|
||||
MIN_RGB_VALUE = MAX_RGB_VALUE - PER_STAGE
|
||||
0 <= args.max_rgb_value <= 255
|
||||
and args.color_amount <= args.max_rgb_value * 3
|
||||
and not args.color_amount % 3
|
||||
), f"Invalid arguments. Run `{argv[0]} -h` for help."
|
||||
|
||||
colors_per_stage = args.color_amount // 3
|
||||
min_rgb_value = args.max_rgb_value - colors_per_stage
|
||||
line = 1 if args.number else False
|
||||
line_print_pending = False
|
||||
|
||||
# Windows command prompt / powershell support
|
||||
if os.name == "nt":
|
||||
os.system('')
|
||||
os.system("")
|
||||
|
||||
for path in args.paths:
|
||||
try:
|
||||
file = stdin if path == '-' else open(path, 'r')
|
||||
with stdin if path == "-" else open(path, "r") as file:
|
||||
read_result = file.read(bytes_to_read)
|
||||
except Exception as e:
|
||||
print(f"Error reading {path}: {e}", file=stderr)
|
||||
continue
|
||||
changes = len(list(filter(lambda c: not c in whitespace, read_result))) - 1
|
||||
step = 0 if changes == 0 else (args.color_amount - 1) / changes
|
||||
index = 0
|
||||
|
||||
changes = len(list(filter(lambda c: not c in string.whitespace, read_result))) - 1
|
||||
if line:
|
||||
color_print(
|
||||
args.max_rgb_value, min_rgb_value, min_rgb_value, line_str(line)
|
||||
)
|
||||
|
||||
step = 0 if changes == 0 else (ColorInfo.AMOUNT - 1) / changes
|
||||
index = 0
|
||||
for char in read_result:
|
||||
if line_print_pending:
|
||||
print(line_str(line), end="")
|
||||
line_print_pending = False
|
||||
|
||||
if line and first_file:
|
||||
color_print(
|
||||
ColorInfo.MAX_RGB_VALUE,
|
||||
ColorInfo.MIN_RGB_VALUE,
|
||||
ColorInfo.MIN_RGB_VALUE,
|
||||
line_str(line))
|
||||
if char in whitespace:
|
||||
print(char, end="")
|
||||
if line and char == "\n":
|
||||
line += 1
|
||||
line_print_pending = True
|
||||
continue
|
||||
|
||||
color_value = round(index)
|
||||
value_modifier = color_value % colors_per_stage
|
||||
rgb_values = [
|
||||
min_rgb_value + value_modifier,
|
||||
args.max_rgb_value - value_modifier,
|
||||
min_rgb_value,
|
||||
]
|
||||
[red, green, blue] = [
|
||||
[rgb_values[1], rgb_values[0], rgb_values[2]],
|
||||
rgb_values[::-1],
|
||||
[rgb_values[0], rgb_values[2], rgb_values[1]],
|
||||
][color_value // colors_per_stage]
|
||||
|
||||
for char in read_result:
|
||||
if char in string.whitespace:
|
||||
print(char, end='')
|
||||
if line and char == '\n':
|
||||
line += 1
|
||||
print(line_str(line), end='')
|
||||
continue
|
||||
|
||||
color_value = round(index)
|
||||
value_modifier = color_value % ColorInfo.PER_STAGE
|
||||
rgb_values = [
|
||||
ColorInfo.MIN_RGB_VALUE + value_modifier,
|
||||
ColorInfo.MAX_RGB_VALUE - value_modifier,
|
||||
ColorInfo.MIN_RGB_VALUE,
|
||||
]
|
||||
[red, green, blue] = [
|
||||
[rgb_values[1], rgb_values[0], rgb_values[2]],
|
||||
rgb_values[::-1],
|
||||
[rgb_values[0], rgb_values[2], rgb_values[1]],
|
||||
][color_value // ColorInfo.PER_STAGE]
|
||||
|
||||
color_print(red, green, blue, char)
|
||||
index += step
|
||||
print("\x1b[0m")
|
||||
color_print(red, green, blue, char)
|
||||
index += step
|
||||
line_print_pending = False
|
||||
print("\033[0m", end="")
|
||||
|
|
|
|||
Loading…
Reference in a new issue