[Feat] prompt the built version of mmcv-full

mim/commands/install.py

@@ -16,6 +16,7 @@
     PKG2PROJECT,
     WHEEL_URL,
     echo_warning,
+    get_all_wheel_version,
     get_torch_cuda_version,
 )
 
@@ -95,7 +96,16 @@ def install(
     importlib.reload(pip._vendor.pkg_resources)
 
     # add mmcv-full find links by default
-    install_args += ['-f', get_mmcv_full_find_link()]
+    mmcv_full_find_link = get_mmcv_full_find_link()
+    install_args += ['-f', mmcv_full_find_link]
+
+    try:
+        # Get and prompt the built version of mmcv-full.
+        all_built_versions = get_all_wheel_version(mmcv_full_find_link)
+        click.echo('The mmcv-full built package version that satisfies the '
+                   f'current environment: {all_built_versions}.')
+    except:  # noqa: E722
+        pass
 
     index_url_opt_names = ['-i', '--index-url', '--pypi-url']
     if any([opt_name in install_args for opt_name in index_url_opt_names]):

mim/utils/__init__.py

@@ -22,6 +22,7 @@
     ensure_installation,
     exit_with_error,
     extract_tar,
+    get_all_wheel_version,
     get_config,
     get_content_from_url,
     get_github_url,
@@ -88,4 +89,5 @@
     'parse_home_page',
     'ensure_installation',
     'rich_progress_bar',
+    'get_all_wheel_version',
 ]

mim/utils/utils.py

@@ -554,3 +554,32 @@ def module_full_name(abbr: str) -> str:
     elif abbr in names or is_installed(abbr):
         return abbr
     return ''
+
+
+def get_all_wheel_version(index_page_url: str):
+    """Get all wheel version from the given index page url.
+
+    Args:
+        index_page_url (str): The index page url.
+
+    Returns:
+        list: A list of all version of wheel in this page.
+    """
+    response = get_content_from_url(index_page_url, timeout=5)
+    anchors = re.findall(r'<a href="(.*?)">', response.text)
+
+    wheel_file_re = re.compile(
+        r"""^(?P<namever>(?P<name>.+?)-(?P<ver>.*?))
+        ((-(?P<build>\d[^-]*?))?-(?P<pyver>.+?)-(?P<abi>.+?)-(?P<plat>.+?)
+        \.whl|\.dist-info)$""",
+        re.VERBOSE,
+    )
+
+    all_versions = set()
+    for anchor in anchors:
+        wheel_info = wheel_file_re.match(anchor.split('/')[-1])
+        if wheel_info:
+            # TODO: check valid tags (pyversions, abis, plat)
+            all_versions.add(wheel_info.group('ver').replace('_', '-'))
+
+    return sorted(all_versions, key=parse_version)