Qbot2.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
#	   Qbot2.py
#	   
#	   Copyright 2011 James McClain <jamezmcclain@gmail.com>
#	   
#	   This program is free software; you can redistribute it and/or modify
#	   it under the terms of the GNU General Public License as published by
#	   the Free Software Foundation; either version 3 of the License, or
#	   (at your option) any later version.
#	   
#	   This program is distributed in the hope that it will be useful,
#	   but WITHOUT ANY WARRANTY; without even the implied warranty of
#	   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#	   GNU General Public License for more details.
#	   
#	   You should have received a copy of the GNU General Public License
#	   along with this program; if not, write to the Free Software
#	   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
#	   MA 02110-1301, USA.

# TODO ###########################
#
# LISTING WHAT I HAVE PLANNED: 
# option tags "-f, -x", piping !|, evaluation, chat logging and statistics. 
# Commands that are planned:
# !say !fortune !qfortune !vote !poll !uno !cleverbot !welcome 
# !roll !spin !google !spelz !copy !werewolf !connect4 !holdem !blackjack
#
##################################
import u413lib
import time
import random
CK = '@'

laioni = False
def Command(data):
	test_data = data.split(' ')
	if test_data[0][0] == CK:
		if len(test_data) > 1:
			return {'Command': test_data[0][1:].lower(),'Args': ' '.join(test_data[1:])}
		else:
			return {'Command': test_data[0][1:].lower(),'Args': False}
	else:
			return {'Command': False,'Args': False}
			
def laioni_mode(old_string):
	to_mix = old_string.split(' ')
	mixed = ""
	
	for string in to_mix:
		if len(string) > 3:
			mix = string[1:-1]
			mix = list(mix)
			random.shuffle(mix)
			mix = ''.join(mix)
			string = string[0]+mix+string[-1]
			mixed += string+" "
		elif len(string) == 3:
			mix = string[1:]
			mix = list(mix)
			random.shuffle(mix)
			mix = ''.join(mix)
			string = string[0]+mix
			mixed += string+" "
		else:
			mix = string
			mix = list(mix)
			random.shuffle(mix)
			mix = ''.join(mix)
			string = mix
			mixed += string+" "
	return mixed[:-1]
	
def send(string):
	global chat
	if laioni:
		string = laioni_mode(string)
		chat.send(string)
	else:
		chat.send(string)
def main():
	global laioni
	global chat
	client = u413lib.createclient()
	if not client.login('USER','PASS'):
		exit()
	chat = client.joinchat('general')
	client.sendRawCommand('channel general')
	while True:
		chatget = chat.get()
		if chatget:
			for text in chatget:
				if text['Type'] == 'Message':
					command = Command(text['Msg'])
					if command['Command'] == 'say':
						if command['Args']:
							print repr(command['Args'])
							send(command['Args'])
					elif command['Command'] == 'hw':
						if command['Args']:
							send('hw args')
						else:
							string = 'Hello World!'
							for i in range(len(string)+1):
								send(string[0:i])
					elif command['Command'] == 'roll':
						if command['Args']:
							pass
						else:
							num = random.choice(range(100))
							send(str(num))
					elif command['Command'] == 'laioni':
						if command['Args']:
							pass
						else:
							if laioni:
								laioni = False
								send("Laioni mode switched off")
							else:
								send("Laioni mode switched on")
								laioni = True
		
				print text['User'],text['Msg'],text['Timestamp']
		time.sleep(7)
	return 0

if __name__ == '__main__':
	main()