maxsplit

Author: hauntsaninja

mypy/stubinfo.py

@@ -2,19 +2,19 @@
 
 
 def is_module_from_legacy_bundled_package(module: str) -> bool:
-    top_level = module.split(".")[0]
+    top_level = module.split(".", 1)[0]
     return top_level in legacy_bundled_packages
 
 
 def approved_stub_package_exists(module: str) -> bool:
-    components = module.split(".")
-    top_level = components[0]
+    top_level = module.split(".", 1)[0]
     if top_level in legacy_bundled_packages:
         return True
     if top_level in non_bundled_packages_flat:
         return True
     if top_level in non_bundled_packages_namespace:
         namespace = non_bundled_packages_namespace[top_level]
+        components = module.split(".")
         for i in range(len(components), 0, -1):
             module = ".".join(components[:i])
             if module in namespace:
@@ -23,8 +23,7 @@ def approved_stub_package_exists(module: str) -> bool:
 
 
 def stub_distribution_name(module: str) -> str | None:
-    components = module.split(".")
-    top_level = components[0]
+    top_level = module.split(".", 1)[0]
 
     dist = legacy_bundled_packages.get(top_level)
     if dist:
@@ -35,6 +34,7 @@ def stub_distribution_name(module: str) -> str | None:
 
     if top_level in non_bundled_packages_namespace:
         namespace = non_bundled_packages_namespace[top_level]
+        components = module.split(".")
         for i in range(len(components), 0, -1):
             module = ".".join(components[:i])
             dist = namespace.get(module)