Add filter option to band script

wladerer · Feb 23, 2024 · abb648d · abb648d
1 parent f4d598a
commit abb648d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/vsh/scripts/__init__.py b/vsh/scripts/__init__.py
@@ -101,6 +101,8 @@ def band(subparsers):
         "--dpi", type=int, default=800, help="DPI of the output file"
     )
 
+    subp_band.add_argument("--filter", action="store_true", help="Filter the data")
+
 
 def db(subparsers):
     """Parse command line arguments"""

diff --git a/vsh/scripts/band.py b/vsh/scripts/band.py
@@ -117,4 +117,8 @@ def filter_procar(args):
 def run(args):
     functions = {"plot": plot_procar, "filter": filter_procar}
 
-    functions[args.function](args)
+    #if filter, run filter function, else plot
+    if args.filter:
+        functions["filter"](args)
+    else:
+        functions["plot"](args)