Add remove flag

saroad2 · May 13, 2020 · 9fdb495 · 9fdb495
1 parent 8de3a96
commit 9fdb495
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/eddington_static/__main__.py b/src/eddington_static/__main__.py
@@ -30,6 +30,9 @@
 parser.add_argument(
     "-c", "--commands", nargs="+", type=str, help="Specify the commands to run"
 )
+parser.add_argument(
+    "-r", "--remove", nargs="+", type=str, help="Remove commands from running"
+)
 parser.add_argument(
     "--commands-list",
     action="store_true",
@@ -74,6 +77,8 @@ def main() -> None:
         commands = COMMANDS
     if args.commands:
         commands = [command for command in commands if command.name in args.commands]
+    if args.remove:
+        commands = [command for command in commands if command.name not in args.remove]
     failed_commands = []
     for command in commands:
         if not silent: