micropython-stubber 1.23.1__py3-none-any.whl → 1.23.2__py3-none-any.whl

stubber/publish/pypi.py

@@ -1,38 +1,42 @@
-"""
-Read versions published to PyPi or test.PyPi.
-uses the pypi-simple package to get the versions from the simple index.
-"""
-
-from typing import Optional
-
-from packaging.version import Version, parse
-from pypi_simple import PyPISimple, NoSuchProjectError
-from loguru import logger as log
-
-
-def get_pypi_versions(package_name: str, base: Optional[Version] = None, production: bool = True):
-    """Get all versions of a package from a PyPI endpoint."""
-    package_info = None
-    if production:
-        endpoint = "https://pypi.org/simple/"
-    else:
-        endpoint = "https://test.pypi.org/simple/"
-
-    try:
-        with PyPISimple(endpoint=endpoint) as client:
-            package_info = client.get_project_page(project=package_name)
-    except NoSuchProjectError:
-        log.debug(f"Package {package_name} not found on {endpoint}")
-        return []
-
-    if not package_info:
-        return []
-
-    versions = [parse(pkg.version) for pkg in package_info.packages if pkg.package_type == "wheel" and pkg.version]
-    # print(versions)
-
-    if base:
-        # if base provided then filter
-        versions = [v for v in versions if v.base_version == base.base_version]
-
-    return sorted(versions)
+"""
+Read versions published to PyPi or test.PyPi.
+uses the pypi-simple package to get the versions from the simple index.
+"""
+
+from typing import Optional
+
+from packaging.version import Version, parse
+from pypi_simple import PyPISimple, NoSuchProjectError
+from mpflash.logger import log
+
+
+def get_pypi_versions(package_name: str, base: Optional[Version] = None, production: bool = True):
+    """Get all versions of a package from a PyPI endpoint."""
+    package_info = None
+    if production:
+        endpoint = "https://pypi.org/simple/"
+    else:
+        endpoint = "https://test.pypi.org/simple/"
+
+    try:
+        with PyPISimple(endpoint=endpoint) as client:
+            package_info = client.get_project_page(project=package_name)
+    except NoSuchProjectError:
+        log.debug(f"Package {package_name} not found on {endpoint}")
+        return []
+
+    if not package_info:
+        return []
+
+    versions = [
+        parse(pkg.version)
+        for pkg in package_info.packages
+        if pkg.package_type == "wheel" and pkg.version
+    ]
+    # print(versions)
+
+    if base:
+        # if base provided then filter
+        versions = [v for v in versions if v.base_version == base.base_version]
+
+    return sorted(versions)