#!/usr/bin/env python

"""Example of announcing 250 services (in this case, a fake HTTP server)."""

from __future__ import annotations

import argparse

import asyncio

import logging

import socket

from zeroconf import IPVersion

from zeroconf.asyncio import AsyncServiceInfo, AsyncZeroconf

class AsyncRunner:

def __init__(self, ip_version: IPVersion) -> None:

self.ip_version = ip_version

self.aiozc: AsyncZeroconf | None = None

async def register_services(self, infos: list[AsyncServiceInfo]) -> None:

self.aiozc = AsyncZeroconf(ip_version=self.ip_version)

tasks = [self.aiozc.async_register_service(info) for info in infos]

background_tasks = await asyncio.gather(*tasks)

await asyncio.gather(*background_tasks)

print("Finished registration, press Ctrl-C to exit...")

await asyncio.Event().wait()

async def unregister_services(self, infos: list[AsyncServiceInfo]) -> None:

assert self.aiozc is not None

tasks = [self.aiozc.async_unregister_service(info) for info in infos]

background_tasks = await asyncio.gather(*tasks)

await asyncio.gather(*background_tasks)

await self.aiozc.async_close()

if __name__ == "__main__":

logging.basicConfig(level=logging.DEBUG)

parser = argparse.ArgumentParser()

parser.add_argument("--debug", action="store_true")

version_group = parser.add_mutually_exclusive_group()

version_group.add_argument("--v6", action="store_true")

version_group.add_argument("--v6-only", action="store_true")

args = parser.parse_args()

if args.debug:

logging.getLogger("zeroconf").setLevel(logging.DEBUG)

if args.v6:

ip_version = IPVersion.All

elif args.v6_only:

ip_version = IPVersion.V6Only

else:

ip_version = IPVersion.V4Only

infos = []

for i in range(250):

infos.append(

AsyncServiceInfo(

"_http._tcp.local.",

f"Paul's Test Web Site {i}._http._tcp.local.",

addresses=[socket.inet_aton("127.0.0.1")],

port=80,

properties={"path": "/~paulsm/"},

server=f"zcdemohost-{i}.local.",

)

)

print("Registration of 250 services...")

loop = asyncio.get_event_loop()

runner = AsyncRunner(ip_version)

try:

loop.run_until_complete(runner.register_services(infos))

except KeyboardInterrupt:

loop.run_until_complete(runner.unregister_services(infos))