main.py

import asyncio
import time

from src.utils import REQUESTS, Request
from src.logs import create_logger
from src.connectors import HostResponse, get_request_latency


logger = create_logger("app.log")


async def main():

    requests = REQUESTS
    max_host_len = max([r.host_len() for r in requests])
    tasks = {}

    while True:
        # Append tasks that
        for req in requests:
            if req.running:
                continue
            
            if time.time() - req.prev_time > req.sleep_s:
                task = asyncio.create_task(get_request_latency(req))
                tasks[task] = req
                req.running = True

        
        if len(tasks) == 0:
            await asyncio.sleep(1)
            continue
                
        done, pending = await asyncio.wait(
            tasks.keys(), timeout=1, return_when=asyncio.FIRST_COMPLETED
        )
                

        for task in done:
            req: Request = tasks.pop(task)
            req.running = False
            req.prev_time = time.time()
            
            result: HostResponse = task.result()
            
            if result.is_error():
                logger.info(
                    f"Wifi {result.wifi_name}, Host {result.host.ljust(max_host_len)} latency {result.latency:.2f} ms, host not reachable,  Error - {result.get_err_message()}"
                )
            else:
                logger.info(
                    f"Wifi {result.wifi_name}, Host {result.host.ljust(max_host_len)} latency {result.latency:.2f} ms"
                )
            
            if result.dns_resp:
                logger.info(
                    f"Wifi {result.wifi_name}, Host {result.host.ljust(max_host_len)} DNS {result.get_dns_info()} "
                )


if __name__ == "__main__":
    asyncio.run(main())