Merge pull request #185 from henryiii/henryiii/feat/python_extras

feat: support python extras

pre_commit_mirror_maker/languages.py

@@ -4,6 +4,7 @@
 import subprocess
 import urllib.request
 
+from packaging import requirements
 from packaging import version
 
 
@@ -20,7 +21,8 @@ def node_get_package_versions(package_name: str) -> list[str]:
 
 
 def python_get_package_versions(package_name: str) -> list[str]:
-    url = f'https://pypi.org/pypi/{package_name}/json'
+    pypi_name = requirements.Requirement(package_name).name
+    url = f'https://pypi.org/pypi/{pypi_name}/json'
     resp = json.load(urllib.request.urlopen(url))
     return sorted(resp['releases'], key=lambda k: version.parse(k))
 

tests/languages_test.py

@@ -23,6 +23,12 @@ def test_python_get_package_version_output():
     assert_all_text(ret)
 
 
+def test_python_get_package_version_extras_output():
+    ret = python_get_package_versions('bandit[yaml]')
+    assert ret
+    assert_all_text(ret)
+
+
 def test_ruby_get_package_version_output():
     ret = ruby_get_package_versions('scss-lint')
     assert ret