yrriban
2438995093
Most of this is now in doom_base.py, which stores wad/map/name, exposes them as properties, and also takes care of most common tasks needed for building command lines and such.
36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
from tkinter import messagebox
|
|
import dcc.config
|
|
import dcc.doom_base
|
|
import sys
|
|
import wand.display
|
|
import wand.image
|
|
|
|
class SS(dcc.doom_base.WadMap):
|
|
def get_parser(self, prog_name):
|
|
parser = super().get_parser(prog_name)
|
|
parser.add_argument("-g", "--gravity", default="center")
|
|
parser.add_argument("-y", "--yolo", action="store_true")
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
def try_screenshot():
|
|
with wand.image.Image(width=dcc.config.THUMB_WIDTH, height=dcc.config.THUMB_HEIGHT, pseudo="x:") as img:
|
|
img.reset_coords()
|
|
if img.size[0] < dcc.config.THUMB_WIDTH or img.size[1] < dcc.config.THUMB_HEIGHT:
|
|
if not messagebox.askretrycancel(title="DCC", message="Image too small. Try again?"):
|
|
sys.exit("Gave up trying to select an image.")
|
|
return False
|
|
img.crop(width=dcc.config.THUMB_WIDTH,height=dcc.config.THUMB_HEIGHT,gravity=parsed_args.gravity)
|
|
img.reset_coords()
|
|
if not parsed_args.yolo:
|
|
wand.display.display(img)
|
|
accepted = messagebox.askyesnocancel(title="DCC", message="Is this image acceptable?")
|
|
if accepted is None:
|
|
sys.exit("Gave up on image verification")
|
|
if not accepted:
|
|
return False
|
|
img.save(filename=self.base_thumb_path())
|
|
return True
|
|
|
|
while not try_screenshot():
|
|
pass
|