Skip to content

Advanced energy scan #49

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
puddly wants to merge 6 commits into
base: dev
Choose a base branch
from
Draft

Advanced energy scan #49

Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
1de3134
WIP: Advanced energy scan
puddly Jul 29, 2024
d8bc230
List nearby networks as well
puddly Jul 29, 2024
db2e486
Merge branch 'dev' into puddly/advanced-energy-scan
puddly Aug 12, 2024
f775fb5
Switch to JSON and dump network information
puddly Aug 12, 2024
aa58eaa
Properly deduplicate scanned networks
puddly Aug 12, 2024
ebc953d
Add the current channel as well
puddly Aug 15, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
WIP: Advanced energy scan
  • Loading branch information
@puddly
puddly committed Jul 29, 2024
commit 1de3134b3bbf798b09f0b63d58270c997fc17760
53 changes: 53 additions & 0 deletions zigpy_cli/radio.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,8 @@
import itertools
import json
import logging
import random
import time

import click
import zigpy.state
Expand Down Expand Up @@ -227,6 +229,57 @@ async def energy_scan(app, num_scans):
print()


@radio.command()
@click.pass_obj
@click.option("-n", "--num-scans", type=int, default=10 * 2**8)
@click.option("-r", "--randomize", type=bool, default=True)
@click.argument("output", type=click.File("w"), default="-")
@click_coroutine
async def advanced_energy_scan(app, output, num_scans, randomize):
await app.startup()
LOGGER.info("Running scan...")

channels = zigpy.types.Channels.ALL_CHANNELS
scan_counts = {channel: num_scans for channel in channels}

if randomize:

def iter_channels():
while scan_counts:
channel = random.choice(tuple(scan_counts))
scan_counts[channel] -= 1

yield channel

if scan_counts[channel] <= 0:
del scan_counts[channel]

else:

def iter_channels():
for channel, count in scan_counts.items():
for i in range(count):
yield channel

with click.progressbar(
iterable=iter_channels(),
length=len(list(channels)) * num_scans,
item_show_func=lambda item: None if item is None else f"Channel {item}",
) as bar:
output.write("Timestamp,Channel,Energy\n")

for channel in bar:
results = await app.energy_scan(
channels=zigpy.types.Channels.from_channel_list([channel]),
duration_exp=0,
count=1,
)

energy = results[channel]
timestamp = time.time()
output.write(f"{timestamp:0.4f},{channel},{energy:0.4f}\n")


@radio.command()
@click.pass_obj
@click.option("-c", "--channel", type=int)
Expand Down