Added a '--version' flag to the server.

linqs · Jun 27, 2024 · 2be4317 · 2be4317
1 parent 183973e
commit 2be4317
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/sharkclipper/cli/server.py b/sharkclipper/cli/server.py
@@ -6,6 +6,7 @@
 import sharkclipper.api.server
 import sharkclipper.util.log
 import sharkclipper.util.ffmpeg
+import sharkclipper.util.version
 
 def check_requirements():
     result = True
@@ -43,6 +44,7 @@ def _get_parser():
         help = 'Do not cleanup any intermitent results (good for debugging) (default: %(default)s).')
 
     sharkclipper.util.log.set_cli_args(parser)
+    sharkclipper.util.version.set_cli_args(parser)
 
     return parser
 

diff --git a/sharkclipper/util/version.py b/sharkclipper/util/version.py
@@ -35,6 +35,17 @@ def get_version():
 
     return UNKNOWN_VERSION
 
+def set_cli_args(parser):
+    """
+    Set common version args in an argparse parser.
+    """
+
+    parser.add_argument('--version',
+        action = 'version', version = get_version(),
+        help = 'Output the Shark Clipper version and exit.')
+
+    return parser
+
 def main():
     args = _get_parser().parse_args()
     sharkclipper.util.log.init_from_args(args)