Add function to get installed prerelease packages

Author: NeonDaniel

neon_utils/packaging_utils.py

@@ -276,3 +276,22 @@ def _pip_install(command_args: List[str]) -> int:
         stat = _pip_install(pip_args)
 
     return stat
+
+
+def get_installed_prereleases() -> List[Tuple[str, str]]:
+    """
+    Get a list of installed pre-release packages.
+    @return: List of tuple (pkg_name, version)
+    """
+    from subprocess import run
+    packages = run(["pip", "list"],
+                   capture_output=True).stdout.decode("utf-8")
+    prerelease_pkgs = list()
+    for line in packages.split('\n'):
+        if not line:
+            continue
+        name, version = line.split()
+        if not version.replace('.', '').isnumeric():
+            prerelease_pkgs.append((name, version))
+    return prerelease_pkgs
+

tests/packaging_util_tests.py

@@ -173,6 +173,20 @@ def test_install_packages_from_pip(self):
             self.assertEqual(0, test_result)
             self.assertEqual(mock_method.call_count, 2)
 
+    @patch("subprocess.run")
+    def test_get_installed_prereleases(self, run):
+        run.return_value.stdout = """stable_package          1.0.0
+beta_package            0.2.2b3
+alpha_package           0.0.0a0
+date_package            24.4.30
+""".encode("utf-8")
+        from neon_utils.packaging_utils import get_installed_prereleases
+        prereleases = get_installed_prereleases()
+        self.assertEqual(len(prereleases), 2)
+        for pkg in prereleases:
+            self.assertTrue(pkg[0].endswith("_package"))
+            self.assertEqual(len(pkg[1].split('.')), 3)
+
 
 if __name__ == '__main__':
     unittest.main()