[python] mark chip cluster library as type checking ready

Author: agners

build/chip/python_wheel.gni

@@ -52,11 +52,18 @@ template("chip_python_wheel_action") {
       _py_scripts = []
     }
 
+    if (defined(invoker.py_typed_packages)) {
+      _py_typed_packages = py_typed_packages
+    } else {
+      _py_typed_packages = []
+    }
+
     _py_manifest = {
       files = _py_manifest_files_rebased
       packages = py_packages
       scripts = _py_scripts
       package_reqs = py_package_reqs
+      typed_packages = _py_typed_packages
     }
 
     write_file(_py_manifest_file, _py_manifest, "json")

src/controller/python/BUILD.gn

@@ -318,6 +318,7 @@ chip_python_wheel_action("chip-clusters") {
         "chip/clusters/TestObjects.py",
         "chip/clusters/Types.py",
         "chip/clusters/enum.py",
+        "chip/clusters/py.typed",
         "chip/tlv/__init__.py",
         "chip/tlv/tlvlist.py",
       ]
@@ -328,6 +329,8 @@ chip_python_wheel_action("chip-clusters") {
     },
   ]
 
+  py_typed_packages = [ "chip.clusters" ]
+
   inputs = []
 
   py_packages = [

src/controller/python/build-chip-wheel.py

@@ -115,6 +115,12 @@ def has_ext_modules(foo):
     # Build the chip package...
     #
     packages = manifest['packages']
+    package_data = {}
+    for package in manifest['typed_packages']:
+        package_data[package] = ["py.typed"]
+
+    if libName:
+        package_data[packages[0]]: [libName]
 
     print("packageName: {}".format(packageName))
     print("libName: {}".format(libName))
@@ -140,11 +146,7 @@ def has_ext_modules(foo):
             # By default, look in the tmp directory for packages/modules to be included.
             '': tmpDir,
         },
-        package_data={
-            packages[0]: [
-                libName
-            ]
-        } if libName else {},
+        package_data=package_data,
         scripts=[name for name in map(
             lambda script: os.path.join(tmpDir, script.installName),
             installScripts