added argparse

This commit is contained in:
Stephen McQuay 2013-02-23 17:03:18 -08:00
parent 24123e84c9
commit 0df5cce6af
1 changed files with 13 additions and 1 deletions

14
tbot.py
View File

@ -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()