Merge "Implemented version printing in fuelclient"

fuelclient/fuel

@@ -17,12 +17,14 @@ from __future__ import print_function
 import argparse
 import curses
 from functools import partial
+from functools import wraps
 from itertools import chain
 from itertools import groupby
 import json
 import math
 from operator import itemgetter
 import os
+import pkg_resources
 import shutil
 import sys
 from time import sleep
@@ -42,6 +44,11 @@ else:
     defaults.update(os.environ)
 ROOT = "http://{LISTEN_ADDRESS}:{LISTEN_PORT}".format(**defaults)
 
+try:
+    __version__ = pkg_resources.get_distribution("python-fuelclient").version
+except pkg_resources.DistributionNotFound:
+    __version__ = ""
+
 OSTF_ROOT = ROOT + "/ostf/"
 API_ROOT = ROOT + "/api/v1/"
 DEBUG = False
@@ -65,6 +72,15 @@ class DeployProgressError(Exception):
     pass
 
 
+class FuelVersionAction(argparse._VersionAction):
+    """Custom argparse._VersionAction subclass to compute fuel server version
+
+    :returns: prints fuel server version
+    """
+    def __call__(self, parser, namespace, values, option_string=None):
+        parser.exit(message=get_fuel_version())
+
+
 class NodeAction(argparse.Action):
     """Custom argparse.Action subclass to store node identity
 
@@ -126,6 +142,16 @@ def handle_exceptions(exc):
         raise exc
 
 
+def exceptions_decorator(func):
+    @wraps(func)
+    def wrapper(*args, **kwargs):
+        try:
+            return func(*args, **kwargs)
+        except Exception as exc:
+            handle_exceptions(exc)
+    return wrapper
+
+
 def recur_get(multi_level_dict, key_chain):
     """Method accesses some field in nested dictionaries
 
@@ -1915,6 +1941,14 @@ substitutions = {
 }
 
 
+@exceptions_decorator
+def get_fuel_version():
+    return yaml.safe_dump(
+        json_api_get_request("version"),
+        default_flow_style=False
+    )
+
+
 def prepare_args():
     # replace some args from dict substitutions
     sys.argv = map(
@@ -1988,26 +2022,27 @@ if __name__ == '__main__':
     prepare_args()
 
     parser = argparse.ArgumentParser(
-        usage="fuel [optional args] <namespace> [action] [flags]",
+        usage="fuel [optional args] <namespace> [action] [flags]"
     )
+    parser.add_argument("-v", "--version",
+                        action="version",
+                        version=__version__)
+    parser.add_argument("--fuel-version", action=FuelVersionAction)
     parser.add_argument("--json",
                         dest="json",
                         action="store_true",
                         help="prints to only json to stdout",
-                        default=False
-                        )
+                        default=False)
     parser.add_argument("--yaml",
                         dest="yaml",
                         action="store_true",
                         help="prints to only yaml to stdout",
-                        default=False
-                        )
+                        default=False)
     parser.add_argument("--debug",
                         dest="debug",
                         action="store_true",
                         help="prints details of all HTTP request",
-                        default=False
-                        )
+                        default=False)
     subparsers = parser.add_subparsers(
         title="Namespaces",
         metavar="",
@@ -2036,7 +2071,6 @@ if __name__ == '__main__':
     if parsed_params.action not in actions:
         parser.print_help()
         sys.exit(0)
-    try:
-        actions[parsed_params.action]["action"](parsed_params)
-    except Exception as e:
-        handle_exceptions(e)
+    exceptions_decorator(
+        actions[parsed_params.action]["action"]
+    )(parsed_params)