def test_no(self):
self.assertEqual(parser.get_response('no'), parser.Response.No)
self.assertEqual(parser.get_response('NO'), parser.Response.No)
self.assertEqual(parser.get_response('n'), parser.Response.No)
self.assertEqual(parser.get_response('false'), parser.Response.No)
self.assertEqual(parser.get_response('f'), parser.Response.No)
self.assertEqual(parser.get_response('disable'), parser.Response.No)
self.assertEqual(parser.get_response('d'), parser.Response.No)
self.assertEqual(parser.get_response('off'), parser.Response.No)
self.assertEqual(parser.get_response('0'), parser.Response.No)
def test_yes(self):
self.assertEqual(parser.get_response('yes'), parser.Response.Yes)
self.assertEqual(parser.get_response('YES'), parser.Response.Yes)
self.assertEqual(parser.get_response('y'), parser.Response.Yes)
self.assertEqual(parser.get_response('true'), parser.Response.Yes)
self.assertEqual(parser.get_response('t'), parser.Response.Yes)
self.assertEqual(parser.get_response('enable'), parser.Response.Yes)
self.assertEqual(parser.get_response('e'), parser.Response.Yes)
self.assertEqual(parser.get_response('on'), parser.Response.Yes)
self.assertEqual(parser.get_response('1'), parser.Response.Yes)
async def commandSetAutoPurge(args: ChatCommandArgs) -> bool:
twitchUser: str = args.message.lower[1]
nick: str = args.message[1]
db: DatabaseMain
cursor: aioodbc.cursor.Cursor
async with DatabaseMain.acquire() as db, await db.cursor() as cursor:
query: str = '''\
SELECT 1 FROM auto_purge WHERE broadcaster=? AND twitchUser=?
'''
await cursor.execute(query, (args.chat.channel, twitchUser))
row: Optional[Tuple[int]] = await cursor.fetchone()
if row is None:
value: bool = False
if len(args.message) >= 3:
response: parser.Response
value = bool(
parser.get_response(args.message.lower[2],
default=parser.Yes))
query = '''
INSERT INTO auto_purge (broadcaster, twitchUser, stopcommands) VALUES (?, ?, ?)
'''
await cursor.execute(query, (args.chat.channel, twitchUser, value))
args.chat.send(f'Enabled Auto-Purge on {nick}')
else:
query = '''
DELETE FROM auto_purge WHERE broadcaster=? AND twitchUser=?
'''
await cursor.execute(query, (args.chat.channel, twitchUser))
args.chat.send(f'Disabled Auto-Purge on {nick}')
await db.commit()
await library.reset_auto_purges(args.chat.channel, args.data)
return True
async def feature(data: CacheStore, channel: str, message: Message,
send: Send) -> bool:
action: str = message.lower[2] if len(message) >= 3 else ''
feature_: str = message.lower[1]
features: Mapping[str, Optional[str]] = lib.items.feature.features()
if feature_ not in features or features[feature_] is None:
send('Unrecognized feature: ' + feature_)
return True
response: parser.Response = parser.get_response(action, default=parser.Yes)
if response == parser.Yes:
return await feature_add(data, channel, feature_, send)
if response == parser.No:
return await feature_remove(data, channel, feature_, send)
send('Unrecognized second parameter: ' + action)
return True
def test_unknown(self):
self.assertEqual(parser.get_response('Kappa'), parser.Response.Unknown)
def test_default(self):
self.assertEqual(parser.get_response('', default=parser.Response.Yes),
parser.Response.Yes)
self.assertEqual(parser.get_response('', default=parser.Response.No),
parser.Response.No)
def test(self):
self.assertEqual(parser.get_response(''), parser.Response.Unknown)