Support deps -r <requirements file>

honesty/cmdline.py

@@ -429,21 +429,27 @@ def checkcache() -> None:
 @click.option("--historical", help="yyyy-mm-dd of a historical date to simulate")
 @click.option("--have", help="pkg==ver to assume already installed", multiple=True)
 @click.option("--use-json", is_flag=True, default=True, show_default=True)
-@click.argument("package_name")
+@click.option("-r", "--requirement_file")
+@click.argument("package_names", nargs=-1)
 def deps(
     include_extras: bool,
     verbose: bool,
     flat: bool,
     pick: bool,
     python_version: str,
     sys_platform: str,
-    package_name: str,
+    package_names: List[str],
     historical: str,
     have: List[str],
     use_json: bool,
+    requirement_file: str,
 ) -> None:
     logging.basicConfig(level=logging.DEBUG if verbose else logging.WARNING)
 
+    if requirement_file:
+        # TODO handle comments and whatnot
+        package_names = Path(requirement_file).read_text().splitlines()
+
     trim_newer: Optional[datetime]
     if historical:
         trim_newer = datetime.strptime(historical, "%Y-%m-%d").replace(
@@ -469,7 +475,7 @@ def current_versions_callback(p: str) -> Optional[str]:
     seen: Set[Tuple[str, Optional[Tuple[str, ...]], Version]] = set()
     assert python_version.count(".") == 2
     deptree = DepWalker(
-        package_name,
+        package_names,
         python_version,
         sys_platform,
         only_first=pick,
@@ -480,7 +486,7 @@ def current_versions_callback(p: str) -> Optional[str]:
         use_json=use_json,
     )
     # TODO record constraints on DepEdge, or put in lib to avoid this nonsense
-    fake_root = DepNode("", version=Version("0"), deps=[DepEdge(target=deptree)])
+    fake_root = DepNode("", version=Version("0"), deps=[DepEdge(target=x) for x in deptree])
     if pick:
         print(f"{deptree.name}=={deptree.version}")
     elif flat:

honesty/deps.py

@@ -90,15 +90,15 @@ def _all_current_versions_unknown(cn: str) -> Optional[str]:
 class DepWalker:
     def __init__(
         self,
-        starting_package: str,
+        starting_packages: List[str],
         python_version: str,
         sys_platform: Optional[str] = None,
         only_first: bool = False,
         trim_newer: Optional[datetime] = None,
     ) -> None:
         self.nodes: Dict[KeyType, DepNode] = {}
-        self.queue: List[Tuple[Optional[DepNode], str]] = [(None, starting_package)]
-        self.root: Optional[DepNode] = None
+        self.queue: List[Tuple[Optional[DepNode], str]] = [(None, pkg) for pkg in starting_packages]
+        self.roots: List[DepNode] = []
         # TODO support unusual versions.
         t = ".".join(python_version.split(".")[:2])
         self.markers = EnvironmentMarkers(
@@ -116,7 +116,7 @@ def walk(
         include_extras: bool,
         current_versions_callback: Optional[VersionCallback] = None,
         use_json: bool = True,
-    ) -> DepNode:
+    ) -> List[DepNode]:
         if current_versions_callback is None:
             current_versions_callback = _all_current_versions_unknown
         already_chosen: Dict[str, Version] = {}
@@ -194,7 +194,7 @@ def walk(
                     self.nodes[key] = node
 
                 if parent is None:
-                    self.root = node
+                    self.roots.append(node)
                 else:
                     parent.deps.append(
                         DepEdge(
@@ -234,8 +234,7 @@ def walk(
                         LOG.info(f"enqueue {d!r} for {node!r}")
                 node.done = True
 
-        assert self.root is not None
-        return self.root
+        return self.roots
 
     def _do_markers_match(self, marker: Marker, extras: Sequence[str] = ()) -> bool:
         env = dict(**asdict(self.markers), extras=Extras(extras))