diff --git a/dcc/ls.py b/dcc/ls.py index 9fb1a81..62fdb37 100644 --- a/dcc/ls.py +++ b/dcc/ls.py @@ -1,27 +1,39 @@ import dcc.config import os + class List(dcc.config.Base): - def get_parser(self, prog_name): - parser = super().get_parser(prog_name) - parser.add_argument("target") - parser.add_argument("--wad") - return parser + def get_parser(self, prog_name): + parser = super().get_parser(prog_name) + parser.add_argument("target") + parser.add_argument("--wad") + return parser - def take_action(self, parsed_args): - match parsed_args.target: - case "pwads": - self.list(x.name for x in os.scandir(self.pwads) if x.is_dir()) - case "iwads": - self.list(x.name for x in os.scandir(self.iwads) if x.is_file()) - case "demos": - self.list(x.name for x in os.scandir(self.demos.joinpath(parsed_args.wad)) if x.name.endswith(".lmp")) - case "videos": - self.list(x.name for x in os.scandir(self.fabricate.joinpath(parsed_args.wad)) if x.name.endswith(".mp4")) - case _: - raise Exception(f"unknown target {parsed_args.target}") + def take_action(self, parsed_args): + match parsed_args.target: + case "pwads": + self.list(x.name for x in os.scandir(self.pwads) if x.is_dir()) + case "iwads": + self.list( + x.name for x in os.scandir(self.iwads) + if x.is_file() + ) + case "demos": + self.list( + x.name for x in + os.scandir(self.demos.joinpath(parsed_args.wad)) + if x.name.endswith(".lmp") + ) + case "videos": + self.list( + x.name for x in + os.scandir(self.fabricate.joinpath(parsed_args.wad)) + if x.name.endswith(".mp4") + ) + case _: + raise Exception(f"unknown target {parsed_args.target}") - def list(self, gen): - # TODO: fancy text? - for i in sorted(gen): - print(i) + def list(self, gen): + # TODO: fancy text? + for i in sorted(gen): + print(i)