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: Reorganize redact script and improve code formatting

Browse source
This commit is contained in:
Paul Gauthier 2025-03-11 19:30:46 -07:00 • committed by Paul Gauthier (aider)
parent 533e5ec03f
commit 9513d307a1
1 changed files with 29 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

23
redact.py → scripts/redact-cast.py
View file

@ -1,12 +1,13 @@
#!/usr/bin/env python3
import json
import os
import re
import sys
import os
import json
# Speed up factor for the recording
SPEEDUP = 1.25
def process_file(input_path, output_path):
"""
Process an asciinema cast v2 file to filter out certain sections based on ANSI cursor commands.
@ -22,13 +23,16 @@ def process_file(input_path, output_path):
"""
skip_mode = False
target_pattern = None
ansi_pattern = re.compile(r'\u001b\[(\d+);(\d+)H')
ansi_pattern = re.compile(r"\u001b\[(\d+);(\d+)H")
is_first_line = True
last_timestamp = 0.0
time_offset = 0.0 # Accumulator for time to subtract
max_gap = 0.5 # Maximum allowed time gap between events
with open(input_path, 'r', encoding='utf-8') as infile, open(output_path, 'w', encoding='utf-8') as outfile:
with (
open(input_path, "r", encoding="utf-8") as infile,
open(output_path, "w", encoding="utf-8") as outfile,
):
for line in infile:
# Always include the header (first line)
if is_first_line:
@ -49,13 +53,13 @@ def process_file(input_path, output_path):
# If we're not in skip mode, check if we need to enter it
if not skip_mode:
if '\u001b[' in text and 'Atuin' in text:
if "\u001b[" in text and "Atuin" in text:
match = ansi_pattern.search(text)
if match:
row = match.group(1)
col = int(match.group(2))
# Create pattern for the ending sequence
target_pattern = f'\u001b[{row};{col-1}H'
target_pattern = f"\u001b[{row};{col-1}H"
skip_mode = True
# Start tracking time to subtract
skip_start_time = current_timestamp
@ -79,14 +83,14 @@ def process_file(input_path, output_path):
last_timestamp = adjusted_timestamp
# Apply speedup factor to the timestamp
record[0] = adjusted_timestamp / SPEEDUP
outfile.write(json.dumps(record) + '\n')
outfile.write(json.dumps(record) + "\n")
# If we're in skip mode, check if we should exit it
else:
if target_pattern in text:
skip_mode = False
# Calculate how much time to subtract from future timestamps
time_offset += (current_timestamp - skip_start_time)
time_offset += current_timestamp - skip_start_time
# Add a 0.5 second pause after each skip section
last_timestamp += 0.5
@ -108,13 +112,14 @@ def process_file(input_path, output_path):
last_timestamp = adjusted_timestamp
# Apply speedup factor to the timestamp
record[0] = adjusted_timestamp / SPEEDUP
outfile.write(json.dumps(record) + '\n')
outfile.write(json.dumps(record) + "\n")
# Otherwise we're still in skip mode, don't write anything
except json.JSONDecodeError:
# If we can't parse the line as JSON, include it anyway
outfile.write(line)
if __name__ == "__main__":
if len(sys.argv) != 3:
print(f"Usage: {os.path.basename(sys.argv[0])} input_file output_file")