Allowing user to specify a target version

.github/workflows/tests.yml

@@ -8,7 +8,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: [3.6, 3.7, 3.8, 3.9]
+        python-version: ["3.6", "3.7", "3.8", "3.9", "3.10"]
     steps:
       - uses: actions/checkout@v2
       - name: Set up Python ${{ matrix.python-version }}

hpxml_version_translator/__init__.py

@@ -1,23 +1,34 @@
 import argparse
-from hpxml_version_translator.converter import convert_hpxml_to_3
+from hpxml_version_translator.converter import (
+    convert_hpxml_to_version,
+    get_hpxml_versions,
+)
 import sys
 
 
 def main(argv=sys.argv[1:]):
-    parser = argparse.ArgumentParser(description='HPXML Version Translator, convert an HPXML file to 3.0')
-    parser.add_argument(
-        'hpxml_input',
-        help='Filename of hpxml file'
+    parser = argparse.ArgumentParser(
+        description="HPXML Version Translator, convert an HPXML file to a newer version"
     )
+    parser.add_argument("hpxml_input", help="Filename of hpxml file")
     parser.add_argument(
-        '-o', '--output',
-        type=argparse.FileType('wb'),
+        "-o",
+        "--output",
+        type=argparse.FileType("wb"),
         default=sys.stdout.buffer,
-        help='Filename of output HPXML v3 file. If not provided, will go to stdout'
+        help="Filename of output HPXML v3 file. If not provided, will go to stdout",
+    )
+    parser.add_argument(
+        "-v",
+        "--to_hpxml_version",
+        type=str,
+        default="3.0",
+        choices=get_hpxml_versions(),
+        help="Major version of HPXML to translate to, default: 3.0",
     )
     args = parser.parse_args(argv)
-    convert_hpxml_to_3(args.hpxml_input, args.output)
+    convert_hpxml_to_version(args.to_hpxml_version, args.hpxml_input, args.output)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()  # pragma: no cover