dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-20 12:24:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: Optimize terminal emulator initialization in redact-cast.py

Browse source
This commit is contained in:
Paul Gauthier 2025-03-13 16:30:41 -07:00 • committed by Paul Gauthier (aider)
parent 4ca229fd42
commit 953391d9d0
1 changed files with 7 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

17
scripts/redact-cast.py
View file

@ -30,14 +30,12 @@ def main():
height = header_data.get("height", 24) height = header_data.get("height", 24)
print(f"Terminal dimensions: {width}x{height}") print(f"Terminal dimensions: {width}x{height}")
# Initialize terminal emulator but don't use it unless necessary
screen = pyte.Screen(width, height)
stream = pyte.Stream(screen)
# Track if we need to check the terminal (if "Atuin" might be on screen) # Track if we need to check the terminal (if "Atuin" might be on screen)
check_terminal = False
atuin_pattern = re.compile(r"A.*t.*u.*i.*n") atuin_pattern = re.compile(r"A.*t.*u.*i.*n")
screen = stream = None
# Process events line by line # Process events line by line
for line in tqdm(fin, desc="Processing events", total=total_lines - 1): for line in tqdm(fin, desc="Processing events", total=total_lines - 1):
if not line.strip(): if not line.strip():
@ -47,10 +45,11 @@ def main():
raw_line_has_atuin = bool(atuin_pattern.search(line)) raw_line_has_atuin = bool(atuin_pattern.search(line))
if raw_line_has_atuin: if raw_line_has_atuin:
check_terminal = True screen = pyte.Screen(width, height)
stream = pyte.Stream(screen)
# Only parse JSON if we're checking terminal or need to check # Only parse JSON if we're checking terminal or need to check
if not check_terminal: if not screen:
fout.write(line) fout.write(line)
continue continue
@ -71,10 +70,8 @@ def main():
atuin_visible = True atuin_visible = True
break break
# Reset flag if Atuin is no longer visible
check_terminal = atuin_visible
if not atuin_visible: if not atuin_visible:
screen = stream = None
fout.write(line) fout.write(line)