Add rvss command line tool

Author: vmayoral

cvsslib/command2.py

@@ -0,0 +1,30 @@
+from cvsslib.vector import detect_vector, calculate_vector, VectorError
+from cvsslib import cvss2, cvss3
+import argparse
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Calculate RVSS scores from a vector")
+    parser.add_argument('vector')
+    parser.add_argument('-v', default=None, dest="version", type=int, help="RVSS version to use (default: autodetect)")
+    args = parser.parse_args()
+
+    module = None
+
+    if args.version is not None:
+        module = {2: cvss2, 3: cvss3}[args.version]
+    else:
+        module = detect_vector(args.vector)
+
+    try:
+        results = calculate_vector(args.vector, module)
+    except VectorError as e:
+        print("Error parsing vector: {0}".format(e.message))
+    else:
+        print("Base Score:\t{0}".format(results[0]))
+        print("Temporal:\t{0}".format(results[1]))
+        print("Environment:\t{0}".format(results[2]))
+
+
+if __name__ == "__main__":
+    main()

setup.py

@@ -23,7 +23,7 @@
                      '(https://github.com/ctxis/cvsslib) for examples and documentation.',
     install_requires=requirements,
     entry_points={
-        'console_scripts': ['cvss=cvsslib.command:main'],
+        'console_scripts': ['cvss=cvsslib.command:main', 'rvss=cvsslib.command2:main'],
     },
     classifiers=[
         'Framework :: Django',