mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-20 12:24:59 +00:00
feat: Add script to filter asciinema cast files for "Atuin" visibility
This commit is contained in:
parent
97291b806a
commit
70d10a0bb2
1 changed files with 45 additions and 0 deletions
|
@ -5,3 +5,48 @@ import re
|
|||
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:
|
||||
# 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
|
||||
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')
|
||||
for event in filtered_events:
|
||||
f.write(json.dumps(event) + '\n')
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue