Merge pull request #3 from multi-py/pre-release-fix

Properly exclude prereleases

Author: tedivm

scripts/latest_versions.py

@@ -1,27 +1,27 @@
 #!/usr/bin/python3
 import os
-import sys
 import re
+import sys
 import urllib.request
 
 PACKAGE = sys.argv[1]
 MAX_VERSIONS = sys.argv[2] if len(sys.argv) > 2 else 10
-EXCLUDE_VERSIONS=os.environ.get("EXCLUDE_VERSIONS", "").split(",")
+EXCLUDE_VERSIONS = os.environ.get("EXCLUDE_VERSIONS", "").split(",")
 
 # Load PyPI website for the package
 fp = urllib.request.urlopen(f"https://pypi.org/simple/{PACKAGE}/")
 html_string = fp.read().decode("utf8")
 fp.close()
 
 # Pick a regex- either include prerelease versions or not
-if os.environ.get("INCLUDE_PRERELEASE", False) == 'true':
-  version_regex = re.compile(r'-(\d+\.\d+\.\d+(a|b|rc)?\d?).+[\.whl|\.tar\.gz]')
-  versions = [x[0] for x in version_regex.findall(html_string) if x[0] not in EXCLUDE_VERSIONS]
+if os.environ.get("INCLUDE_PRERELEASE", False) == "true":
+    version_regex = re.compile(r"-(\d+\.\d+\.\d+(a|b|rc)?\d?).+[\.whl|\.tar\.gz]")
+    versions = [x[0] for x in version_regex.findall(html_string) if x[0] not in EXCLUDE_VERSIONS]
 else:
-  version_regex = re.compile(r'-(\d+\.\d+\.\d+).+[\.whl|\.tar\.gz]')
-  versions = [x for x in version_regex.findall(html_string) if x not in EXCLUDE_VERSIONS]
+    version_regex = re.compile(r"-(\d+\.\d+\.\d+)\..+[\.whl|\.tar\.gz]")
+    versions = [x for x in version_regex.findall(html_string) if x not in EXCLUDE_VERSIONS]
 
 unique_versions = list(dict.fromkeys(versions))
 
-for version in unique_versions[-(int(MAX_VERSIONS)):]:
-  print(version)
+for version in unique_versions[-(int(MAX_VERSIONS)) :]:
+    print(version)