Fix the extract command to work with the new nested config classes.

This commit is contained in:
yrriban 2025-07-23 18:26:11 -04:00
parent 30baa49588
commit 114fec8f7e

View file

@ -1,14 +1,13 @@
import dcc.config import dcc.doom_base
import omg import omg
import numpy as np import numpy as np
import wand.color import wand.color
import wand.image import wand.image
class Extract(dcc.config.base): class Extract(dcc.doom_base.Wad):
def get_parser(self, prog_name): def get_parser(self, prog_name):
parser = super().get_parser(prog_name) parser = super().get_parser(prog_name)
parser.add_argument('wad')
parser.add_argument('lump') parser.add_argument('lump')
return parser return parser
@ -30,17 +29,17 @@ class Extract(dcc.config.base):
) as img: ) as img:
img.transparent_color(wand.color.Color("#ff00ff"), 0.0) img.transparent_color(wand.color.Color("#ff00ff"), 0.0)
img.save( img.save(
filename=self.output.joinpath(parsed_args.wad) filename=self.fabricate.joinpath(parsed_args.wad)
.joinpath(parsed_args.lump + ".png") .joinpath(parsed_args.lump + ".png")
) )
return return
except Exception as e: except Exception as e:
print( print(
f"Wad {w} likely has no lump {parsed_args.lump}" f"Wad {w} likely has no lump {parsed_args.lump} "
+ f"(exception {e})." + f"(exception {e})."
) )
print( print(
"Lump {parsed_args.lump} not found in any wad in" "Lump {parsed_args.lump} not found in any wad in "
+ f"{parsed_args.wad}" + f"{parsed_args.wad}"
) )