Add a means of indicating the tier of map when adding the index overlay.

This commit is contained in:
yrriban 2026-04-16 01:46:55 -04:00
parent 9bc41264e5
commit a982c5b20c

View file

@ -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())