diff --git a/dcc/thumb.py b/dcc/thumb.py index 9ff3b07..a1afa95 100644 --- a/dcc/thumb.py +++ b/dcc/thumb.py @@ -3,28 +3,34 @@ import dcc.doom_base import wand.color import wand.image + class Thumb(dcc.doom_base.WadMap): - def get_parser(self, prog_name): - parser = super().get_parser(prog_name) - parser.add_argument("--index", action="store_true") - return parser - - def take_action(self, parsed_args): - base = self.base_thumb_path() - text = self.text_thumb_path() - mdoom = self.m_doom_path() - with wand.image.Image(filename=base) as bi, wand.color.Color("transparent") as tc: - with wand.image.Image(filename=text) as ti: - ti.border(tc, 5, 5) - bi.composite(ti, gravity="south_west") + def get_parser(self, prog_name): + parser = super().get_parser(prog_name) + parser.add_argument("--index", action="store_true") + return parser - with wand.image.Image(filename=mdoom) as mdi: - mdi.border(tc, 5, 5) - bi.composite(mdi, gravity="north_west") + def take_action(self, parsed_args): + base = self.base_thumb_path() + text = self.text_thumb_path() + mdoom = self.m_doom_path() + with ( + wand.image.Image(filename=base) as bi, + wand.color.Color("transparent") as tc + ): + with wand.image.Image(filename=text) as ti: + ti.border(tc, 5, 5) + bi.composite(ti, gravity="south_west") - if parsed_args.index: - with wand.image.Image(filename=self.output.joinpath("doomed_index.png")) as di: - di.border(tc, 1, 1) - bi.composite(di, gravity="north_east") + with wand.image.Image(filename=mdoom) as mdi: + mdi.border(tc, 5, 5) + bi.composite(mdi, gravity="north_west") - bi.save(filename=self.thumb_path()) + if parsed_args.index: + with wand.image.Image( + filename=self.output.joinpath("doomed_index.png") + ) as di: + di.border(tc, 1, 1) + bi.composite(di, gravity="north_east") + + bi.save(filename=self.thumb_path())