diff --git a/tbot.py b/tbot.py index 3e0f60c..5902413 100644 --- a/tbot.py +++ b/tbot.py @@ -1,3 +1,5 @@ +import argparse + from twisted.words.protocols import irc from twisted.internet import protocol, reactor @@ -39,5 +41,15 @@ class IDRTBotFactory(protocol.ClientFactory): print('could not connect: {}'.format(reason)) -reactor.connectTCP('localhost', 6667, IDRTBotFactory(['#smb', '#linux'])) +parser = argparse.ArgumentParser() +parser.add_argument('-H', '--host', type=str, default='localhost') +parser.add_argument('-p', '--port', type=int, default=6667) +parser.add_argument('channels', nargs='+') + +args = parser.parse_args() + +reactor.connectTCP( + args.host, int(args.port), + IDRTBotFactory('#{}'.format(c) for c in args.channels) +) reactor.run()