feat: added package dependency search in index

src/index.py

@@ -85,3 +85,29 @@ def set_packages_apt_source(self, source):
         for package_versions in self.packages.values():
             for package in package_versions.values():
                 package.apt_source = source
+
+    def get_package_dependecies(self, name, packages=None):
+        current = self.search(name)
+
+        if current is None:
+            raise KeyError(name)
+
+        if packages is None:
+            packages = []
+
+        if current.priority in ["required", "important"]:
+            return None
+
+        if current in packages:
+            return None
+        else:
+            packages.append(current)
+
+        [
+            self.get_package_dependecies(dep, packages)
+            for dep in current.pre_dependencies
+        ]
+
+        [self.get_package_dependecies(dep, packages) for dep in current.dependencies]
+
+        return packages

tests/test_index.py

@@ -1,5 +1,7 @@
 # ruff: noqa: E501
 
+from pathlib import Path
+
 from src.index import Index
 from src.version import Version
 
@@ -126,6 +128,14 @@ def test_should_check_if_package_is_contained_in_list():
     assert "ca-certificates" not in index
 
 
+def test_should_get_package_dependencies():
+    index = valid_full_index_of_package()
+
+    packages = index.get_package_dependecies("python3")
+
+    assert len(packages) == 33
+
+
 def valid_index_file():
     return """Package: python3
 Source: python3-defaults (3.11.4-5)
@@ -253,3 +263,8 @@ def valid_index_with_other_package():
 Filename: ./ca-certificates_20230311_all.deb"""
 
     return Index(data)
+
+
+def valid_full_index_of_package():
+    data = Path("tests", "resources", "python_Packages").read_text()
+    return Index(data)