mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-29 16:54:59 +00:00
initial
This commit is contained in:
commit
2eb1513612
1 changed files with 51 additions and 0 deletions
51
redact.py
Executable file
51
redact.py
Executable file
|
@ -0,0 +1,51 @@
|
|||
#!/usr/bin/env python3
|
||||
import re
|
||||
import sys
|
||||
import os
|
||||
|
||||
def process_file(input_path, output_path):
|
||||
"""
|
||||
Process a text file to filter out certain sections based on ANSI cursor commands.
|
||||
|
||||
If a line contains "\u001b[ROW;COL]H" followed by "Atuin", skip it and all subsequent
|
||||
lines until finding a line with "\u001b[ROW;(COL-1)H".
|
||||
"""
|
||||
skip_mode = False
|
||||
target_pattern = None
|
||||
ansi_pattern = re.compile(r'\\u001b\[(\d+);(\d+)H')
|
||||
|
||||
with open(input_path, 'r', encoding='utf-8') as infile, open(output_path, 'w', encoding='utf-8') as outfile:
|
||||
for line in infile:
|
||||
# If we're not in skip mode, check if we need to enter it
|
||||
if not skip_mode:
|
||||
if '\\u001b[' in line and 'Atuin' in line:
|
||||
match = ansi_pattern.search(line)
|
||||
if match:
|
||||
row = match.group(1)
|
||||
col = int(match.group(2))
|
||||
# Create pattern for the line that will end the skip section
|
||||
target_pattern = f'\\u001b[{row};{col-1}H'
|
||||
skip_mode = True
|
||||
continue # Skip this line
|
||||
# If we're not skipping, write the line
|
||||
outfile.write(line)
|
||||
# If we're in skip mode, check if we should exit it
|
||||
else:
|
||||
if target_pattern in line:
|
||||
skip_mode = False
|
||||
outfile.write(line) # Include the matching line
|
||||
|
||||
if __name__ == "__main__":
|
||||
if len(sys.argv) != 3:
|
||||
print(f"Usage: {os.path.basename(sys.argv[0])} input_file output_file")
|
||||
sys.exit(1)
|
||||
|
||||
input_file = sys.argv[1]
|
||||
output_file = sys.argv[2]
|
||||
|
||||
if not os.path.exists(input_file):
|
||||
print(f"Error: Input file '{input_file}' does not exist")
|
||||
sys.exit(1)
|
||||
|
||||
process_file(input_file, output_file)
|
||||
print(f"Processed {input_file} -> {output_file}")
|
Loading…
Add table
Add a link
Reference in a new issue