Add a new option --json-pdt

src/python_inspector/resolution.py

@@ -627,13 +627,49 @@ def format_resolution(
         return dependencies
 
 
+def pdt_dfs(mapping, graph, src):
+    """
+    Return a nested mapping of dependencies.
+
+    This takes ``mapping`` and ``graph`` as input. And do a dfs
+    on the ``graph`` to get the dependencies of the given ``src``.
+    And use the ``mapping`` to get the version of the given dependency.
+    """
+    children = list(graph.iter_children(src))
+    if not children:
+        return dict(
+            key=src, package_name=src, installed_version=str(mapping[src].version), dependencies=[]
+        )
+
+    return dict(
+        key=src,
+        package_name=src,
+        installed_version=str(mapping[src].version),
+        dependencies=sorted([pdt_dfs(mapping, graph, c) for c in children], key=lambda d: d["key"]),
+    )
+
+
+def format_pdt_tree(results):
+    """
+    Return a formatted tree of dependencies.
+    """
+    mapping = results.mapping
+    graph = results.graph
+    dependencies = []
+    for src in get_all_srcs(mapping=mapping, graph=graph):
+        dependencies.append(pdt_dfs(mapping=mapping, graph=graph, src=src))
+    dependencies.sort(key=lambda d: d["key"])
+    return dependencies
+
+
 def get_resolved_dependencies(
     requirements: List[Requirement],
     environment: Environment = None,
     repos: Sequence[PypiSimpleRepository] = tuple(),
     as_tree: bool = False,
     max_rounds: int = 200000,
     debug: bool = False,
+    pdt_output: bool = False,
 ):
     """
     Return resolved dependencies of a ``requirements`` list of Requirement for
@@ -648,9 +684,12 @@ def get_resolved_dependencies(
             provider=PythonInputProvider(environment=environment, repos=repos),
             reporter=BaseReporter(),
         )
-        results = resolver.resolve(requirements=requirements, max_rounds=max_rounds)
-        results = format_resolution(results, as_tree=as_tree, environment=environment, repos=repos)
-        return results
+        resolver_results = resolver.resolve(requirements=requirements, max_rounds=max_rounds)
+        if pdt_output:
+            return format_pdt_tree(resolver_results)
+        return format_resolution(
+            resolver_results, as_tree=as_tree, environment=environment, repos=repos
+        )
     except Exception as e:
         if debug:
             import click

src/python_inspector/resolve_cli.py

@@ -99,7 +99,16 @@
     "--json",
     "json_output",
     type=FileOptionType(mode="w", encoding="utf-8", lazy=True),
-    required=True,
+    required=False,
+    metavar="FILE",
+    help="Write output as pretty-printed JSON to FILE. "
+    "Use the special '-' file name to print results on screen/stdout.",
+)
+@click.option(
+    "--json-pdt",
+    "pdt_output",
+    type=FileOptionType(mode="w", encoding="utf-8", lazy=True),
+    required=False,
     metavar="FILE",
     help="Write output as pretty-printed JSON to FILE. "
     "Use the special '-' file name to print results on screen/stdout.",
@@ -138,6 +147,7 @@ def resolve_dependencies(
     operating_system,
     index_urls,
     json_output,
+    pdt_output,
     max_rounds,
     use_cached_index=False,
     use_pypi_json_api=False,
@@ -161,6 +171,11 @@ def resolve_dependencies(
 
         dad --spec "flask==2.1.2" --json -
     """
+    if not (json_output or pdt_output):
+        if debug:
+            click.secho("No output file specified. Use --json or --json-pdt.", err=True)
+        return
+
     if debug:
         click.secho(f"Resolving dependencies...")
 
@@ -237,6 +252,7 @@ def resolve_dependencies(
         as_tree=False,
         max_rounds=max_rounds,
         debug=debug,
+        pdt_output=pdt_output,
     )
 
     cli_options = [f"--requirement {rf}" for rf in requirement_files]
@@ -262,19 +278,35 @@ def resolve_dependencies(
         errors=[],
     )
 
-    write_output(
-        headers=headers,
-        requirements=requirements,
-        resolved_dependencies=resolved_dependencies,
-        json_output=json_output,
-    )
+    if json_output:
+        write_output(
+            headers=headers,
+            requirements=requirements,
+            resolved_dependencies=resolved_dependencies,
+            json_output=json_output,
+        )
+
+    if pdt_output:
+        write_output(
+            headers=headers,
+            requirements=requirements,
+            resolved_dependencies=resolved_dependencies,
+            json_output=pdt_output,
+            pdt_output=True,
+        )
 
     if debug:
         click.secho("done!")
 
 
 def resolve(
-    direct_dependencies, environment, repos=tuple(), as_tree=False, max_rounds=200000, debug=False
+    direct_dependencies,
+    environment,
+    repos=tuple(),
+    as_tree=False,
+    max_rounds=200000,
+    debug=False,
+    pdt_output=False,
 ):
     """
     Resolve dependencies given a ``direct_dependencies`` list of
@@ -293,8 +325,11 @@ def resolve(
         as_tree=as_tree,
         max_rounds=max_rounds,
         debug=debug,
+        pdt_output=pdt_output,
     )
 
+    print(resolved_dependencies)
+
     initial_requirements = [d.to_dict() for d in direct_dependencies]
 
     return initial_requirements, resolved_dependencies
@@ -312,16 +347,19 @@ def get_requirements_from_direct_dependencies(
         yield Requirement(requirement_string=dependency.extracted_requirement)
 
 
-def write_output(headers, requirements, resolved_dependencies, json_output):
+def write_output(headers, requirements, resolved_dependencies, json_output, pdt_output=False):
     """
     Write headers, requirements and resolved_dependencies as JSON to ``json_output``.
     Return the output data.
     """
-    output = dict(
-        headers=headers,
-        requirements=requirements,
-        resolved_dependencies=resolved_dependencies,
-    )
+    if not pdt_output:
+        output = dict(
+            headers=headers,
+            requirements=requirements,
+            resolved_dependencies=resolved_dependencies,
+        )
+    else:
+        output = resolved_dependencies
 
     json.dump(output, json_output, indent=2)
     return output