akemi/dotfiles
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Screenshot: Add edit subcommand function

Browse source
This commit is contained in:
Akemi Izuko 2023-12-23 20:14:00 -07:00
parent 01b79a244b
commit c68ee16732
Signed by: akemi
GPG key ID: 8DE0764E1809E9FC
1 changed files with 31 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
bin/screenshot_wayland.py
View file

@ -3,10 +3,10 @@
# #
# Additional dependencies: # Additional dependencies:
# - Python's pillow library # - Python's pillow library
# - pip install --user pillow-avif-plugin # For avif support (recommended) # - pip install --user pillow-avif-plugin # Until PIL merges avif support
# - grim # - grim
# - swappy # - swappy
import argparse, os, time, re, shutil import argparse, os, time, re, shutil, pillow_avif
from subprocess import run, Popen, PIPE, DEVNULL from subprocess import run, Popen, PIPE, DEVNULL
from PIL import Image from PIL import Image
from PIL.ImageFilter import GaussianBlur from PIL.ImageFilter import GaussianBlur
@ -26,6 +26,8 @@ DS_SHADOW_LIGHT = 'black'
DS_BG_DARK = '#d3869b' DS_BG_DARK = '#d3869b'
DS_SHADOW_DARK = 'black' DS_SHADOW_DARK = 'black'
EDIT_QUALITY = 50
# Returns the path of the latest screenshot # Returns the path of the latest screenshot
def get_latest_sceenshot_path() -> Path: def get_latest_sceenshot_path() -> Path:
pics = [f for f in os.listdir(DIR) if os.path.isfile(DIR / f)] pics = [f for f in os.listdir(DIR) if os.path.isfile(DIR / f)]
@ -128,6 +130,32 @@ def take_subcommand(args):
if args.file is not None: if args.file is not None:
shutil.copyfile(save_path, args.file) shutil.copyfile(save_path, args.file)
# Applies an edit to the latest screenshot
def edit_subcommand(args):
ext = args.extension if args.extension else 'avif'
og_path, edit_path = get_edit_path(ext)
# Drop shadow
match args.drop_shadow:
case "light":
add_drop_shadow(og_path, edit_path, DS_BG_LIGHT, DS_SHADOW_LIGHT)
case "dark":
add_drop_shadow(og_path, edit_path, DS_BG_DARK, DS_SHADOW_DARK)
# Resize
with Image.open(edit_path if args.drop_shadow else og_path) as img:
if args.size:
img = img.resize(args.size)
elif args.rescale:
img = img.reduce(int(1 / (args.rescale / 100)))
img.save(edit_path, quality=EDIT_QUALITY, method=6)
if args.clipboard:
copy_to_clipboard(edit_path)
if args.file is not None:
shutil.copyfile(edit_path, args.file)
# Provides a drawing editor for the latest image # Provides a drawing editor for the latest image
def markup_latest(args): def markup_latest(args):
img = get_latest_sceenshot_path() img = get_latest_sceenshot_path()
@ -290,9 +318,7 @@ elif not os.path.exists(DIR):
match args.subcommand: match args.subcommand:
case "take": case "take":
take_subcommand(args) take_subcommand(args)
pass
case "edit": case "edit":
print("Edit is not yet ready for use") edit_subcommand(args)
pass
case "markup": case "markup":
markup_latest(args) markup_latest(args)