dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-30 17:24:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

feat: Add support for custom filenames with image extensions in /clipboard command

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-09 14:47:38 -03:00
parent a94b871991
commit 8bae297e5d
1 changed files with 13 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

16
aider/commands.py
View file

@ -903,10 +903,20 @@ class Commands:
# Check for image first
image = ImageGrab.grabclipboard()
if isinstance(image, Image.Image):
basename = args.strip() if args.strip() else "clipboard_image"
if args.strip():
filename = args.strip()
ext = os.path.splitext(filename)[1].lower()
if ext in ('.jpg', '.jpeg', '.png'):
basename = filename
else:
basename = f"{filename}.png"
else:
basename = "clipboard_image.png"
temp_dir = tempfile.mkdtemp()
temp_file_path = os.path.join(temp_dir, f"{basename}.png")
image.save(temp_file_path, "PNG")
temp_file_path = os.path.join(temp_dir, basename)
image_format = 'PNG' if basename.lower().endswith('.png') else 'JPEG'
image.save(temp_file_path, image_format)
abs_file_path = Path(temp_file_path).resolve()