Merge pull request #152 from robotpy/manual-init

Provide 'version' argument for init command

Author: virtuald

robotpy_installer/cli_init.py

@@ -1,9 +1,13 @@
 import argparse
-import inspect
 import logging
 import pathlib
+import typing as T
 
+from packaging.version import Version
+
+from . import installer
 from . import pyproject
+from .errors import Error
 from .utils import handle_cli_error
 
 
@@ -16,10 +20,29 @@ class Init:
     """
 
     def __init__(self, parser: argparse.ArgumentParser):
-        pass
+        parser.add_argument(
+            "version",
+            nargs="?",
+            help="Manually specify the RobotPy package version to use instead of autodetecting it",
+            type=Version,
+        )
 
     @handle_cli_error
-    def run(self, main_file: pathlib.Path, project_path: pathlib.Path):
+    def run(
+        self,
+        main_file: pathlib.Path,
+        project_path: pathlib.Path,
+        version: T.Optional[Version],
+    ):
+
+        supported_year = int(installer._WPILIB_YEAR)
+        if version is not None and version.major != int(installer._WPILIB_YEAR):
+            msg = (
+                f"Only RobotPy {supported_year}.x is supported by this version "
+                f"of robotpy-installer (specified {version})"
+            )
+            raise Error(msg)
+
         project_path.mkdir(parents=True, exist_ok=True)
 
         # Create robot.py if it doesn't already exist
@@ -33,7 +56,9 @@ def run(self, main_file: pathlib.Path, project_path: pathlib.Path):
         # Create pyproject.toml if it doesn't already exist
         pyproject_path = pyproject.toml_path(project_path)
         if not pyproject_path.exists():
-            pyproject.write_default_pyproject(project_path)
+            pyproject.write_default_pyproject(
+                project_path, str(version) if version is not None else None
+            )
 
             logger.info("Created %s", pyproject_path)
 

robotpy_installer/pyproject.py

@@ -373,20 +373,28 @@ def write_default_gitignore(project_path: pathlib.Path):
 
 def write_default_pyproject(
     project_path: pathlib.Path,
+    robotpy_version: typing.Optional[str] = None,
 ):
     """
     Using the current environment, write a minimal pyproject.toml
 
     :param project_path: Path to robot project
     """
 
-    robotpy_version = robotpy_installed_version()
+    if robotpy_version is None:
+        robotpy_version = robotpy_installed_version()
+
+    try:
+        provides_extra = metadata("robotpy").get_all("Provides-Extra")
+    except PackageNotFoundError:
+        provides_extra = []
 
-    provides_extra = metadata("robotpy").get_all("Provides-Extra")
     if not provides_extra:
         extras = ""
     else:
-        extras = "\n    # ".join(f'"{extra}",' for extra in sorted(provides_extra))
+        extras = "    #" + "\n    # ".join(
+            f'"{extra}",' for extra in sorted(provides_extra)
+        )
 
     content = inspect.cleandoc(
         f"""
@@ -404,7 +412,7 @@ def write_default_pyproject(
             # Which optional robotpy components should be installed?
             # -> equivalent to `pip install robotpy[extra1, ...]
             components = [
-                # ##EXTRAS##
+            ##EXTRAS##
             ]
 
             # Other pip packages to install, such as vendor packages (each element