mirror of
https://github.com/sharkdp/bat
synced 2024-11-24 04:43:07 +00:00
59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
import subprocess
|
||
|
|
import glob
|
||
|
|
import sys
|
||
|
|
import os.path as path
|
||
|
|
import os
|
||
|
|
|
||
|
|
|
||
|
|
BAT_OPTIONS = [
|
||
|
|
"--no-config",
|
||
|
|
"--style=plain",
|
||
|
|
"--color=always",
|
||
|
|
"--theme='1337'",
|
||
|
|
"--italic-text=always",
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
def create_highlighted_versions():
|
||
|
|
root = os.path.dirname(os.path.abspath(__file__))
|
||
|
|
|
||
|
|
for source in glob.glob(path.join(root, "source", "*", "*")):
|
||
|
|
try:
|
||
|
|
bat_output = subprocess.check_output(
|
||
|
|
["bat"] + BAT_OPTIONS + [source], stderr=subprocess.PIPE,
|
||
|
|
)
|
||
|
|
|
||
|
|
source_dirname = path.basename(path.dirname(source))
|
||
|
|
source_filename = path.basename(source)
|
||
|
|
|
||
|
|
output_dir = path.join(root, "highlighted", source_dirname)
|
||
|
|
output_path = path.join(output_dir, source_filename)
|
||
|
|
|
||
|
|
os.makedirs(output_dir, exist_ok=True)
|
||
|
|
|
||
|
|
with open(output_path, "wb") as output_file:
|
||
|
|
output_file.write(bat_output)
|
||
|
|
|
||
|
|
relative_path = path.relpath(output_path, root)
|
||
|
|
print("Created '{}'".format(relative_path))
|
||
|
|
except subprocess.CalledProcessError as err:
|
||
|
|
print(
|
||
|
|
"=== Error: Could not highlight source file '{}".format(source),
|
||
|
|
file=sys.stderr,
|
||
|
|
)
|
||
|
|
print(
|
||
|
|
"=== bat stdout:\n{}".format(err.stdout.decode("utf-8")),
|
||
|
|
file=sys.stderr,
|
||
|
|
)
|
||
|
|
print(
|
||
|
|
"=== bat stderr:\n{}".format(err.stderr.decode("utf-8")),
|
||
|
|
file=sys.stderr,
|
||
|
|
)
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
create_highlighted_versions()
|