install: allow space-separated packages too

This enables users to copy and paste a package list which they'd
normally provide to their package manager, directly into the -I
argument.

Signed-off-by: Stephen Brennan <[email protected]>

Author: brenns10

yo/api.py

@@ -194,8 +194,9 @@ def __str__(self) -> str:
 
 
 def flex_list(arg: t.Union[str, t.List[str]]) -> t.List[str]:
+    splitter = re.compile(r"[,\s]+")
     if isinstance(arg, str):
-        return arg.split(",")
+        return splitter.split(arg)
     else:
         return arg
 

yo/main.py

@@ -1816,7 +1816,7 @@ def add_args(self, parser: argparse.ArgumentParser) -> None:
             default=[],
             action="append",
             help="Install packages after instance creation (may be specified "
-            "multiple times)",
+            "multiple times, or with space/comma separated values)",
         )
 
     def standardize_wait(self, tasks: bool) -> None:
@@ -1835,8 +1835,9 @@ def maybe_install_packages(
     ) -> None:
         packages = []
         packages.extend(profile.install)
+        splitter = re.compile(r"[,\s]+")
         for pkgspec in self.args.install:
-            packages.extend(pkgspec.split(","))
+            packages.extend(splitter.split(pkgspec))
         if packages:
             args = " ".join(packages)
             tasks.append(