From dba5fb9dfad81b3fc404f46397e02665dfed20d8 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Thu, 13 Mar 2025 16:00:40 -0700 Subject: [PATCH] style: Format code with consistent quotes and spacing --- scripts/redact-cast.py | 30 ++++++++++++++++-------------- 1 file changed, 16 insertions(+), 14 deletions(-) diff --git a/scripts/redact-cast.py b/scripts/redact-cast.py index f85708279..1887fc086 100755 --- a/scripts/redact-cast.py +++ b/scripts/redact-cast.py @@ -6,47 +6,49 @@ import sys import pyte + def main(): if len(sys.argv) != 3: print(f"Usage: {sys.argv[0]} input_cast_file output_cast_file") sys.exit(1) - + input_file = sys.argv[1] output_file = sys.argv[2] - + # Initialize pyte screen and stream screen = pyte.Screen(80, 24) stream = pyte.Stream(screen) - + # Read and parse the cast file - with open(input_file, 'r') as f: + with open(input_file, "r") as f: # First line is header header = f.readline().strip() - + # Read the events events = [json.loads(line) for line in f if line.strip()] - + filtered_events = [] - + # Process each event through the terminal emulator for event in events: # Process the event in the terminal - if len(event) >= 3 and event[1] == 'o': # Output event + if len(event) >= 3 and event[1] == "o": # Output event stream.feed(event[2]) - + # Check if "Atuin" is visible on screen display_content = "\n".join("".join(line) for line in screen.display) if "Atuin" in display_content: continue # Skip this event - + # Keep this event filtered_events.append(event) - + # Write the filtered content to the output file - with open(output_file, 'w') as f: - f.write(header + '\n') + with open(output_file, "w") as f: + f.write(header + "\n") for event in filtered_events: - f.write(json.dumps(event) + '\n') + f.write(json.dumps(event) + "\n") + if __name__ == "__main__": main()