diff --git a/doomcc/thumb.py b/doomcc/thumb.py index 5ee8265..6138278 100644 --- a/doomcc/thumb.py +++ b/doomcc/thumb.py @@ -7,7 +7,7 @@ import wand.image class Thumb(doomcc.doom_base.WadMap): def get_parser(self, prog_name): parser = super().get_parser(prog_name) - parser.add_argument("--index", action="store_true") + parser.add_argument("--index", nargs="?", const="none", default="") parser.add_argument("--noov", action="store_true") return parser @@ -28,11 +28,20 @@ class Thumb(doomcc.doom_base.WadMap): mdi.border(tc, 5, 5) bi.composite(mdi, gravity="north_west") - if parsed_args.index: + if parsed_args.index != "": with wand.image.Image( filename=self.fabricate.joinpath("doomed_index.png") ) as di: di.border(tc, 1, 1) bi.composite(di, gravity="north_east") + if parsed_args.index != "none": + ird = self._config.get("index_rating", {}) + if parsed_args.index in ird: + with wand.image.Image( + filename=self.fabricate.joinpath(self.fabricate.joinpath(ird[parsed_args.index]))) as ri: + ri.border(tc, 1, 1) + bi.composite(ri, top=di.height+5, left=int(bi.width-ri.width-(di.width-ri.width)/2)) + + bi.save(filename=self.thumb_path())