python-kasa · rytilahti · Aug 27, 2023 · Nov 17, 2022 · Aug 27, 2023
diff --git a/devtools/README.md b/devtools/README.md
@@ -15,6 +15,28 @@ Options:
   --help       Show this message and exit.
 ```
 
+## create_module_fixtures
+
+* Queries the device for all supported modules and outputs module-based fixture files for each device.
+* This could be used to create fixture files for module-specific tests, but it might also be useful for other use-cases.
+
+```shell
+Usage: create_module_fixtures.py [OPTIONS] OUTPUTDIR
+
+  Create module fixtures for given host/network.
+
+Arguments:
+  OUTPUTDIR  [required]
+
+Options:
+  --host TEXT
+  --network TEXT
+  --install-completion  Install completion for the current shell.
+  --show-completion     Show completion for the current shell, to copy it or
+                        customize the installation.
+  --help                Show this message and exit.
+```
+
 ## parse_pcap
 
 * Requires dpkt (pip install dpkt)

diff --git a/devtools/create_module_fixtures.py b/devtools/create_module_fixtures.py
@@ -0,0 +1,60 @@
+"""Create fixture files for modules supported by a device.
+
+This script can be used to create fixture files for individual modules.
+"""
+
+import asyncio
+import json
+from pathlib import Path
+
+import typer
+
+from kasa import Discover, SmartDevice
+
+app = typer.Typer()
+
+
+def create_fixtures(dev: SmartDevice, outputdir: Path):
+    """Iterate over supported modules and create version-specific fixture files."""
+    for name, module in dev.modules.items():
+        module_dir = outputdir / name
+        if not module_dir.exists():
+            module_dir.mkdir(exist_ok=True, parents=True)
+
+        sw_version = dev.hw_info["sw_ver"]
+        sw_version = sw_version.split(" ", maxsplit=1)[0]
+        filename = f"{dev.model}_{dev.hw_info['hw_ver']}_{sw_version}.json"
+        module_file = module_dir / filename
+
+        if module_file.exists():
+            continue
+
+        typer.echo(f"Creating {module_file} for {dev.model}")
+        with module_file.open("w") as f:
+            json.dump(module.data, f, indent=4)
+
+
+@app.command()
+def create_module_fixtures(
+    outputdir: Path,
+    host: str = typer.Option(default=None),
+    network: str = typer.Option(default=None),
+):
+    """Create module fixtures for given host/network."""
+    devs = []
+    if host is not None:
+        dev: SmartDevice = asyncio.run(Discover.discover_single(host))
+        devs.append(dev)
+    else:
+        if network is None:
+            network = "255.255.255.255"
+        devs = asyncio.run(Discover.discover(target=network)).values()
+        for dev in devs:
+            asyncio.run(dev.update())
+
+    for dev in devs:
+        create_fixtures(dev, outputdir)
+
+
+if __name__ == "__main__":
+    app()