mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-20 12:24:59 +00:00
refactor: Use regex to check for ordered "Atuin" characters in line
This commit is contained in:
parent
693a43efc8
commit
4ca229fd42
1 changed files with 4 additions and 3 deletions
|
@ -1,6 +1,7 @@
|
||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
import json
|
import json
|
||||||
import os
|
import os
|
||||||
|
import re
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
import pyte
|
import pyte
|
||||||
|
@ -35,7 +36,7 @@ def main():
|
||||||
|
|
||||||
# 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
|
check_terminal = False
|
||||||
atuin_chars = set("Atuin")
|
atuin_pattern = re.compile(r"A.*t.*u.*i.*n")
|
||||||
|
|
||||||
# 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):
|
||||||
|
@ -43,9 +44,9 @@ def main():
|
||||||
continue
|
continue
|
||||||
|
|
||||||
# Fast initial check on raw line before JSON parsing
|
# Fast initial check on raw line before JSON parsing
|
||||||
raw_line_has_atuin_chars = any(char in line for char in atuin_chars)
|
raw_line_has_atuin = bool(atuin_pattern.search(line))
|
||||||
|
|
||||||
if raw_line_has_atuin_chars:
|
if raw_line_has_atuin:
|
||||||
check_terminal = True
|
check_terminal = True
|
||||||
|
|
||||||
# Only parse JSON if we're checking terminal or need to check
|
# Only parse JSON if we're checking terminal or need to check
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue