From bf58b6abcf057aaa35d2a5ebc4a067c370826d0e Mon Sep 17 00:00:00 2001 From: randomuser Date: Wed, 21 Jul 2021 16:24:23 -0500 Subject: [PATCH] add simple command line interface --- main.py | 19 +++++++++++++++---- 1 file changed, 15 insertions(+), 4 deletions(-) diff --git a/main.py b/main.py index de264ac..07a2fad 100644 --- a/main.py +++ b/main.py @@ -1,6 +1,7 @@ import asyncio import random import time +import argparse from db import DuckDB from db import DuckEvent @@ -113,11 +114,21 @@ class Bot(BaseBot): def create_server(self, name: str): return Server(self, name) -async def main(): +async def main(srv, port, tls, nick): + print(srv, port, tls, nick) bot = Bot() - params = ConnectionParams("test", "beepboop.systems", 6667, False) - await bot.add_server("beep", params) + params = ConnectionParams(nick, srv, port, tls) + await bot.add_server("main", params) await bot.run() +def parse_args(): + parser = argparse.ArgumentParser(description="cross-channel duck bot") + parser.add_argument('-s', '--host', default="beepboop.systems") + parser.add_argument('-p', '--port', default=6667, type=int) + parser.add_argument('-t', '--tls', action="store_true") + parser.add_argument('-n', '--nick', default="test") + return parser.parse_args() + if __name__ == "__main__": - asyncio.run(main()) + args = parse_args() + asyncio.run(main(args.host, args.port, args.tls, args.nick))