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

Add global constants to screenshot_wayland

Browse source
This commit is contained in:
Akemi Izuko 2022-11-23 17:53:29 -07:00
parent d9bf5d121b
commit 242d6aa191
No known key found for this signature in database
GPG key ID: 905D444F6A5E4BE4
1 changed files with 13 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
bin/screenshot_wayland.py
View file

@ -13,15 +13,24 @@ from PIL.ImageFilter import GaussianBlur
from pathlib import Path from pathlib import Path
from typing import * from typing import *
# Global constants
DIR = Path(f"{os.environ['HOME']}/Desktop/screenshots_tmp") # Default DIR = Path(f"{os.environ['HOME']}/Desktop/screenshots_tmp") # Default
EXTENSION_REGEX = "\.([A-z0-9]+)$"
ORIGINAL_REGEX = "([0-9]+(_[0-9]{0,3})?)\.png"
EDIT_REGEX = "([0-9]+(_[0-9]{0,3})?)_edit_([0-9])+" + EXTENSION_REGEX
DS_BG_LIGHT = 'white'
DS_SHADOW_LIGHT = 'black'
DS_BG_DARK = '#d3869b'
DS_SHADOW_DARK = 'black'
# 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)]
pngs = [p for p in pics if re.fullmatch("[0-9_]+\.png", p)] pngs = [p for p in pics if re.fullmatch("[0-9_]+\.png", p)]
pngs.sort() pngs.sort()
return Path(pngs[-1]) return DIR / pngs[-1]
return pngs[-1]
# Returns a path to an unused screenshot in DIR # Returns a path to an unused screenshot in DIR
def get_sceenshot_path() -> Path: def get_sceenshot_path() -> Path:
@ -90,9 +99,9 @@ def take_subcommand(args):
match args.drop_shadow: match args.drop_shadow:
case "light": case "light":
add_drop_shadow(save_path, save_path, 'white', 'black') add_drop_shadow(save_path, save_path, DS_BG_LIGHT, DS_SHADOW_LIGHT)
case "dark": case "dark":
add_drop_shadow(save_path, save_path, '#d3869b', 'black') add_drop_shadow(save_path, save_path, DS_BG_DARK, DS_SHADOW_DARK)
if args.clipboard: if args.clipboard:
copy_to_clipboard(save_path) copy_to_clipboard(save_path)