wzykubek/pysheller
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Sort arguments passed to script in defined way

Browse Source
This commit is contained in:
samedamci 2021-09-01 14:43:42 +02:00
parent 3309ad8662
commit 4a2d31e29b
No known key found for this signature in database
GPG Key ID: FCB4A9A20D00E894
1 changed files with 18 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
pysheller/app.py
View File

@ -2,6 +2,7 @@ from os import listdir, path
from argparse import ArgumentParser from argparse import ArgumentParser
from subprocess import call from subprocess import call
import re import re
from collections import OrderedDict
class PySheller: class PySheller:
@ -34,14 +35,18 @@ class PySheller:
if line.startswith("#") and not line.startswith("#!") if line.startswith("#") and not line.startswith("#!")
] ]
parameters = {"help": None, "args": []} parameters = {"help": None, "args": {}}
for key, val in [line.split(" : ") for line in comments]: for key, val in [line.split(" : ") for line in comments]:
key, val = key.strip(), val.strip() key, val = key.strip(), val.strip()
if key == "help": if key == "help":
parameters["help"] = val parameters["help"] = val
elif bool(re.match(r"arg\d+$", key)): elif bool(re.match(r"arg\d+$", key)):
name, desc = [x.strip() for x in val.split(" - ")] name, desc = [x.strip() for x in val.split(" - ")]
parameters["args"].append({"name": name, "help": desc}) id_ = int(key.split("arg")[1])
parameters["args"][id_] = {"name": name, "help": desc}
parameters["args"] = dict(OrderedDict(sorted(parameters["args"].items())))
return parameters return parameters
@ -57,13 +62,20 @@ class PySheller:
subcommands[cmd] = subparsers.add_parser(cmd, help=p["help"]) subcommands[cmd] = subparsers.add_parser(cmd, help=p["help"])
if "args" in p: if "args" in p:
for arg in p["args"]: for arg in p["args"].items():
subcommands[cmd].add_argument(arg["name"], help=arg["help"]) subcommands[cmd].add_argument(arg[1]["name"], help=arg[1]["help"])
self.args = self.arg_parser.parse_args() self.args = self.arg_parser.parse_args()
def run_command(self, command: str) -> None: def run_command(self, command: str) -> None:
args = [x[1] for x in self.args._get_kwargs() if x[1] is not None] args = []
call(self.get_path(args[0]) + " " + " ".join(args[1:]), shell=True) for exp_args in self.parse_comments(command)["args"].items():
id_, arg = exp_args
passed_arg = self.args.__getattribute__(re.sub(r"^--", "", arg["name"]))
if passed_arg is None:
args.append("")
else:
args.append(passed_arg)
call(self.get_path(command) + " " + " ".join(args), shell=True)
def run(self) -> None: def run(self) -> None:
self.run_command(self.args.command) self.run_command(self.args.command)