Daniel Folkinshteyn 2011-01-24 16:09:18 -05:00
parent fc1a3ab258
commit a6d361573d
2 changed files with 5 additions and 1 deletions

View File

@ -196,7 +196,7 @@ def banmask(hostmask):
L[-1] = '*' L[-1] = '*'
return '*!*@' + ':'.join(L) return '*!*@' + ':'.join(L)
else: else:
if '.' in host: if len(host.split('.')) > 2: # If it is a subdomain
return '*!*@*%s' % host[host.find('.'):] return '*!*@*%s' % host[host.find('.'):]
else: else:
return '*!*@' + host return '*!*@' + host

View File

@ -233,6 +233,10 @@ class FunctionsTestCase(SupyTestCase):
msg.prefix), msg.prefix),
'%r didn\'t match %r' % (msg.prefix, banmask)) '%r didn\'t match %r' % (msg.prefix, banmask))
self.assertEqual(ircutils.banmask('foobar!user@host'), '*!*@host') self.assertEqual(ircutils.banmask('foobar!user@host'), '*!*@host')
self.assertEqual(ircutils.banmask('foobar!user@host.tld'),
'*!*@host.tld')
self.assertEqual(ircutils.banmask('foobar!user@sub.host.tld'),
'*!*@*.host.tld')
self.assertEqual(ircutils.banmask('foo!bar@2001::'), '*!*@2001::*') self.assertEqual(ircutils.banmask('foo!bar@2001::'), '*!*@2001::*')
def testSeparateModes(self): def testSeparateModes(self):