From 0df5cce6af289c842200ebb63abf979defada995 Mon Sep 17 00:00:00 2001 From: Stephen McQuay Date: Sat, 23 Feb 2013 17:03:18 -0800 Subject: [PATCH] added argparse --- tbot.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) 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()