git.iko.soy/bat
on master help sign in
Add a way to directly search for a pattern
edd2764sharkdp
tests/scripts/find-slow-to-highlight-files.py+19−9
compare with master
unifiedsplit
@@ -12,6 +12,7 @@
12# - fd (https://github.com/sharkdp/fd)
12# - fd (https://github.com/sharkdp/fd)
13# - wc
13# - wc
14
14
15+import sys
15import time
16import time
16import os
17import os
17import subprocess as sp
18import subprocess as sp
@@ -30,7 +31,7 @@ MAX_NUM_FILES = 100
30SEARCH_ROOT = os.getenv("HOME")
31SEARCH_ROOT = os.getenv("HOME")
31
32
32
33
33−def find_slow_files(startup_time, language, glob_pattern):
34+def find_slow_files(startup_time, glob_pattern, language=None):
34 out = sp.check_output(
35 out = sp.check_output(
35 [
36 [
36 "fd",
37 "fd",
@@ -46,7 +47,8 @@ def find_slow_files(startup_time, language, glob_pattern):
46 )
47 )
47
48
48 paths = out.split(b"\n")[:-1]
49 paths = out.split(b"\n")[:-1]
50+ language_text = f"Language {language}, " if language else ""
49− print(f"Language: {language}, glob pattern: {glob_pattern} ({len(paths)} matches)")
51+ print(f"{language_text}glob pattern: {glob_pattern} ({len(paths)} matches)")
50
52
51 for path in paths:
53 for path in paths:
52 num_chars = int(sp.check_output(["wc", "-c", path]).split(b" ")[0].decode())
54 num_chars = int(sp.check_output(["wc", "-c", path]).split(b" ")[0].decode())
@@ -96,19 +98,27 @@ def measure_bat_startup_speed():
96 return min_duration
98 return min_duration
97
99
98
100
99−def main():
101+def traverse_all_languages(startup_time):
100− print("Measuring 'bat' startup speed ... ", flush=True, end="")
101− startup_time = measure_bat_startup_speed()
102− print(f"{startup_time * 1000:.1f} ms")
103−
104 output = sp.check_output(["bat", "--list-languages"]).decode()
102 output = sp.check_output(["bat", "--list-languages"]).decode()
105
103
106 for line in output.strip().split("\n"):
104 for line in output.strip().split("\n"):
107 language, extensions = line.split(":")
105 language, extensions = line.split(":")
108 for ext in extensions.split(","):
106 for ext in extensions.split(","):
109− find_slow_files(startup_time, language, ext)
107+ find_slow_files(startup_time, ext, language)
110 if not ext.startswith("."):
108 if not ext.startswith("."):
111− find_slow_files(startup_time, language, f"*.{ext}")
109+ find_slow_files(startup_time, f"*.{ext}", language)
110+
111+
112+def main():
113+ print("Measuring 'bat' startup speed ... ", flush=True, end="")
114+ startup_time = measure_bat_startup_speed()
115+ print(f"{startup_time * 1000:.1f} ms")
116+
117+ if len(sys.argv) == 1:
118+ traverse_all_languages(startup_time)
119+ else:
120+ pattern = sys.argv[1]
121+ find_slow_files(startup_time, pattern)
112
122
113
123
114if __name__ == "__main__":
124if __name__ == "__main__":