Changed totalbugs/totalrfes to a single total command.

This commit is contained in:
Jeremy Fincher 2004-08-23 19:53:29 +00:00
parent d28ddf718e
commit 5861a5e07c
2 changed files with 22 additions and 6 deletions

View File

@ -265,7 +265,23 @@ class Sourceforge(callbacks.PrivmsgCommandAndRegexp):
# TODO: consolidate total* into one command which takes options for all # TODO: consolidate total* into one command which takes options for all
# the viable statistics that can be snarfed from the project page # the viable statistics that can be snarfed from the project page
def totalbugs(self, irc, msg, args): def total(self, irc, msg, args):
"""[bugs|rfes] [<project>]
Returns the total count of open bugs or rfes. <project> is only
necessary if a default project is not set.
"""
if not args:
raise callbacks.ArgumentError
type = args.pop(0)
if type == 'bugs':
self._totalbugs(irc, msg, args)
elif type == 'rfes':
self._totalrfes(irc, msg, args)
else:
raise callbacks.ArgumentError
def _totalbugs(self, irc, msg, args):
"""[<project>] """[<project>]
Returns a count of the open/total bugs. <project> is not needed if a Returns a count of the open/total bugs. <project> is not needed if a
@ -281,7 +297,7 @@ class Sourceforge(callbacks.PrivmsgCommandAndRegexp):
else: else:
irc.error('Could not find bug statistics.') irc.error('Could not find bug statistics.')
def totalrfes(self, irc, msg, args): def _totalrfes(self, irc, msg, args):
"""[<project>] """[<project>]
Returns a count of the open/total RFEs. <project> is not needed if a Returns a count of the open/total RFEs. <project> is not needed if a

View File

@ -170,10 +170,10 @@ if network:
original) original)
def testTotal(self): def testTotal(self):
self.assertRegexp('totalbugs gaim', r'\d+ open / \d+ total') self.assertRegexp('total bugs gaim', r'\d+ open / \d+ total')
self.assertRegexp('totalrfes gaim', r'\d+ open / \d+ total') self.assertRegexp('total rfes gaim', r'\d+ open / \d+ total')
self.assertError('totalbugs lkjfad') self.assertError('total bugs lkjfad')
self.assertError('totalrfes lkjfad') self.assertError('total rfes lkjfad')
def testFight(self): def testFight(self):
self.assertRegexp('fight gaim opengaim', self.assertRegexp('fight gaim opengaim',