Reset environment with permenant deps

Author: soapy1

conda_self/cli/main_reset.py

@@ -20,12 +20,13 @@ def execute(args: argparse.Namespace) -> int:
     from conda.core.link import PrefixSetup, UnlinkLinkTransaction
     from conda.base.context import context
 
-    from ..constants import PERMANENT_PACKAGES
+    from ..query import permanent_dependencies
 
     print("Resetting 'base' environment...")
 
+    uninstallable_packages = permanent_dependencies()
     installed = sorted(PrefixData(sys.prefix).iter_records(), key=lambda x: x.name)
-    packages_to_remove = [pkg for pkg in installed if pkg.name not in PERMANENT_PACKAGES]
+    packages_to_remove = [pkg for pkg in installed if pkg.name not in uninstallable_packages]
 
     stp = PrefixSetup(
         target_prefix=sys.prefix,

conda_self/constants.py

@@ -1,26 +1,5 @@
 from typing import Final
 
 PERMANENT_PACKAGES: Final =  (
-    "archspec",
-    "boltons",
     "conda",
-    "charset-normalizer",
-    "conda-libmamba-solver",
-    "conda-package-handling",
-    "conda-self",
-    "distro",
-    "frozendict",
-    "jsonpatch",
-    "menuinst",
-    "packaging",
-    "platformdirs",
-    "pluggy",
-    "pycosat"
-    "python",
-    "requests",
-    "ruamel.yaml"
-    "setuptools",
-    "tqdm",
-    "truststore",
-    "zstandard",
 )

conda_self/query.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 import sys
+from functools import cache
 from typing import TYPE_CHECKING
 
 from conda.base.context import context
@@ -13,8 +14,11 @@
     PackagesNotFoundError,
 )
 from conda.models.channel import Channel
+from conda.models.prefix_graph import PrefixGraph
 from conda.models.version import VersionOrder
 
+from .constants import PERMANENT_PACKAGES
+
 if TYPE_CHECKING:
     from collections.abc import Iterable
 
@@ -56,3 +60,16 @@ def latest(
     if best is None:
         raise PackagesNotFoundError(package_name, channels)
     return best
+
+
+@cache
+def permanent_dependencies() -> list[str]:
+    """Get the full list of dependencies for all the permanent packages."""
+    installed = PrefixData(sys.prefix)
+    prefix_graph = PrefixGraph(installed.iter_records())
+
+    packages = []
+    for pkg in PERMANENT_PACKAGES:
+        node = prefix_graph.get_node_by_name(pkg)
+        packages.extend([record.name for record in prefix_graph.all_ancestors(node)])
+    return set(packages)