import argparse
from moonlight_analytics import nightly_router_iperf, nightly_pi_iperf
import asyncio
import sys


if __name__ == '__main__':
    time = sys.argv[2] if len(sys.argv) > 2 else 10
    if sys.argv[1] == "router":
        asyncio.get_event_loop().run_until_complete(nightly_router_iperf(time=time))
    elif sys.argv[1] == "pi":
        asyncio.get_event_loop().run_until_complete(nightly_pi_iperf(time=time))