Fix the extract command to work with the new nested config classes.
This commit is contained in:
parent
30baa49588
commit
114fec8f7e
1 changed files with 5 additions and 6 deletions
|
@ -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}"
|
||||||
)
|
)
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue