Created
December 28, 2013 19:44
-
-
Save thekindlyone/8163333 to your computer and use it in GitHub Desktop.
xmpp chat-bot that emails music among other things.
Uses beets music organizer http://beets.radbox.org/
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import logging | |
| from sleekxmpp import ClientXMPP | |
| from sleekxmpp.exceptions import IqError, IqTimeout | |
| import subprocess | |
| import smtplib | |
| import mimetypes | |
| import email | |
| import email.mime.application | |
| import thread | |
| import os | |
| from multiprocessing import * | |
| import re | |
| import eliza | |
| import pickle | |
| import string | |
| from random import choice | |
| from time import time | |
| import unidecode | |
| import sys | |
| # Mail Thread | |
| def sendmail(fn,to,activeflag,xmobj,recipient): | |
| activeflag.value=1 | |
| filename=fn | |
| msg = email.mime.Multipart.MIMEMultipart() | |
| msg['Subject'] =filename[filename.rfind('\\')+1:] | |
| msg['From'] = '<emailid used by bot eg. [email protected]>' | |
| msg['To'] = to | |
| body = email.mime.Text.MIMEText("""thekindlybot abides. Ask and the tunes are thine.""") | |
| msg.attach(body) | |
| fp=open(filename,'rb') | |
| #att = email.mime.application.MIMEApplication(fp.read(),_subtype="mp3") | |
| att = email.mime.application.MIMEApplication(fp.read()) | |
| fp.close() | |
| att.add_header('Content-Disposition','attachment',filename=filename) | |
| msg.attach(att) | |
| s = smtplib.SMTP('smtp.gmail.com') | |
| s.starttls() | |
| s.login(msg['From'],'<password>') | |
| s.sendmail(msg['From'],[msg['To']], msg.as_string()) | |
| s.quit() | |
| activeflag.value=0 | |
| xmobj.send_message(recipient, mbody='Mail Sent!') | |
| print 'mail sent! to ',recipient | |
| thread.exit() | |
| #hangman work | |
| class player(object): | |
| def __init__(self,pname): | |
| self.name=pname | |
| self.movie='' | |
| self.done=[] | |
| self.tried=[] | |
| self.creationtime=time() | |
| def reset(self): | |
| self.movie=self.fetchmovie() | |
| self.tried=[] | |
| self.done=['-'] | |
| self.lastaction=time() | |
| def fetchmovie(self): | |
| movlist=pickle.load(open("movielist.p","rb")) | |
| self.lastaction=time() | |
| return string.strip(string.lower(choice(movlist))) | |
| def getstat(self): | |
| self.lastaction=time() | |
| return 'name = '+self.name+' movie= '+self.movie+' tried list='+str(self.tried)+' done list='+str(self.done) | |
| def displaytext(self): | |
| self.lastaction=time() | |
| rs='' | |
| for i in range(len(self.movie)): | |
| if self.movie[i] in self.done : rs=rs+self.movie[i] | |
| else: | |
| if self.movie[i]==' ': rs=rs+'/' | |
| else: rs=rs+'_'+' ' | |
| return rs | |
| class game(object): | |
| def __init__(self): | |
| self.playertable={} | |
| def register_player(self,newplayer): | |
| self.playertable[newplayer]=player(newplayer) | |
| def delete_player(self,playername): | |
| self.playertable.pop(playername,None) | |
| return 'Hangman session Ended!' | |
| def startgame(self,f): | |
| self.playertable[f].reset() | |
| return '\n************Game Started!***********\n'+self.playertable[f].displaytext() | |
| def gameresponse(self,q,f): | |
| q=q.replace(' ','') | |
| if q: | |
| if len(q)>1: q=q[0].lower() | |
| if q in self.playertable[f].movie: | |
| self.playertable[f].done.append(q) | |
| response='\n'+self.playertable[f].displaytext()+'\n'+'HANGMAN'[:len(self.playertable[f].tried)] | |
| else: | |
| if q in self.playertable[f].tried: | |
| response='Already tried!\n'+self.playertable[f].displaytext()+'\n'+'HANGMAN'[:len(self.playertable[f].tried)] | |
| else: | |
| self.playertable[f].tried.append(q) | |
| response='INCORRECT, Try Again!\n'+self.playertable[f].displaytext()+'\n'+'HANGMAN'[:len(self.playertable[f].tried)] | |
| if len(self.playertable[f].tried)==len('HANGMAN'): | |
| response='HANGMAN\n Game Over, You lost. The answer was '+self.playertable[f].movie+self.startgame(f) | |
| if '_' not in self.playertable[f].displaytext(): | |
| response=self.playertable[f].movie+'\nVICTORY!!!!\n\n\n'+self.startgame(f) | |
| else: response='WTF?' | |
| return response | |
| def cleanup(self): | |
| for n in self.playertable: | |
| if (self.playertable[n].lastaction-self.playertable[n].creationtime)>(30*60): | |
| self.delete_player(n) | |
| # philo game work | |
| class vplayer(object): | |
| def __init__(self,pname): | |
| self.name=pname | |
| self.question='' | |
| self.answer='' | |
| self.done=[] | |
| self.tried=[] | |
| self.creationtime=time() | |
| def reset(self): | |
| self.question,self.answer=self.fetchquestion() | |
| self.lastaction=time() | |
| self.tried=[] | |
| self.done=[''] | |
| def fetchquestion(self): | |
| self.lastaction=time() | |
| qbank=pickle.load(open(r'philobank.p','r')) | |
| q=choice(qbank.keys()) | |
| a=qbank[q] | |
| return q,a.lower() | |
| def getstat(self): | |
| self.lastaction=time() | |
| return 'name = '+self.name+' question= '+self.question+' answer ='+self.answer+' tried list='+str(self.tried)+' done list='+str(self.done) | |
| def displaytext(self): | |
| self.lastaction=time() | |
| rs='' | |
| for i in range(len(self.answer)): | |
| if self.answer[i] in self.done : rs=rs+self.answer[i] | |
| else: | |
| if self.answer[i]==' ': rs=rs+'/' | |
| else: rs=rs+'_'+' ' | |
| return rs | |
| class vgame(object): | |
| def __init__(self): | |
| self.playertable={} | |
| def register_player(self,newplayer): | |
| self.playertable[newplayer]=vplayer(newplayer) | |
| def delete_player(self,playername): | |
| self.playertable.pop(playername,None) | |
| return 'Quiz session ended!' | |
| def startgame(self,f): | |
| self.playertable[f].reset() | |
| return '\n************Game Started!***********\n'+'Question: '+self.playertable[f].question+'\n\n'+self.playertable[f].displaytext() | |
| def gameresponse(self,q,f): | |
| q=q.replace(' ','') | |
| if q: | |
| if q=='#repeat' or q=='# repeat': | |
| response='Question: '+self.playertable[f].question | |
| else: | |
| if len(q)>1: q=q[0].lower() | |
| if q in self.playertable[f].answer: | |
| self.playertable[f].done.append(q) | |
| response='\n'+self.playertable[f].displaytext()+'\n'+'HANGMAN'[:len(self.playertable[f].tried)] | |
| else: | |
| if q in self.playertable[f].tried: | |
| response='Already tried!\n'+self.playertable[f].displaytext()+'\n'+'HANGMAN'[:len(self.playertable[f].tried)] | |
| else: | |
| self.playertable[f].tried.append(q) | |
| response='INCORRECT, Try Again!\n'+self.playertable[f].displaytext()+'\n'+'HANGMAN'[:len(self.playertable[f].tried)] | |
| if len(self.playertable[f].tried)==len('HANGMAN'): | |
| response='HANGMAN\n Game Over, You lost. The answer was '+self.playertable[f].answer+self.startgame(f) | |
| if '_' not in self.playertable[f].displaytext(): | |
| response=self.playertable[f].answer+' VICTORY!!!!\n\n\n'+self.startgame(f) | |
| else: response='WTF?' | |
| return response | |
| def cleanup(self): | |
| for n in self.playertable: | |
| if (self.playertable[n].lastaction-self.playertable[n].creationtime)>(30*60): | |
| self.delete_player(n) | |
| class EchoBot(ClientXMPP): | |
| def __init__(self, jid, password): | |
| ClientXMPP.__init__(self, jid, password) | |
| self.active= Value('i') | |
| self.active.value=0 | |
| self.therapist=eliza.eliza() | |
| self.artists=pickle.load(open(r'f:\Music\artists.dat')) | |
| self.hangman=game() | |
| self.philo=vgame() | |
| self.links=open(r'd:\pythonary\links.txt').read().split('\n') | |
| self.add_event_handler("session_start", self.session_start) | |
| self.add_event_handler("message", self.message) | |
| def session_start(self, event): | |
| self.send_presence(pstatus="Hello, I am a chat bot. I like distributing music and playing Hangman. Type 'help' for help.") | |
| print 'Session started' | |
| self.get_roster() | |
| def message(self, msg): | |
| if msg['type'] in ('chat', 'normal'): | |
| #msg.reply("Thanks for sending\n%(body)s" % msg).send() | |
| #print '\n\n\n\n' | |
| f="%(from)s"%msg | |
| q="%(body)s"%msg | |
| q=string.lower(q) | |
| print '\n************************\n',f,'says: ',q | |
| pt=self.hangman.playertable | |
| ppt=self.philo.playertable | |
| if (f not in pt)and q !='help' and (f not in ppt): | |
| reply="*"+self.therapist.respond(q)+"*"+"\nTYPE _help_ for syntax" | |
| # search | |
| if q=='list artists': | |
| reply="This is a list of all the artists whose discography is present in my library. Note: There are many more artists besides this in my library, but I do not possess their entire work \n"+self.artists | |
| if q[0]=='$' and len(q)<=5: | |
| reply="This is a 300 GB music library. Do a more specific search. Don't search for 'love' or 'the' etc. *TRY AGAIN*" | |
| if q[0]=='$' and len(q)>5: | |
| ql = filter(None,q[1:].strip().split(' ')) | |
| cl=["beet","list","-f","$artist-$album-$title ($format)"]+ql | |
| #command="beet ls "+q[1:].strip() | |
| try: | |
| track_output=subprocess.check_output(cl) | |
| #file_output=subprocess.check_output("beet ls -p "+q[1:].strip()) | |
| tracks = filter(None, track_output.split('\n')) | |
| #files = filter(None, file_output.split('\n')) | |
| s='' | |
| n=1 | |
| ex="" | |
| length=len(tracks) | |
| if length>500: | |
| tracks=tracks[:100] | |
| ex="\n"+str(length-100)+" more results founds on given keywords. This is a 300 GB music library. Do a more specific search. Don't search for 'love' or 'the' etc. " | |
| for track in tracks: | |
| s=s+'['+str(n)+']'+track+'\n' | |
| n=int(n)+1 | |
| reply='\n'+s+ex | |
| if len(track_output)==0: | |
| reply='''\n thekindlybot could not find what you looked for. Try again with different keywords. | |
| If $ zz-top doesn't work try $ zz top or $ zztop. To learn mail command type help. | |
| If you still can't find the track, then most likely I don't have it.''' | |
| except: | |
| reply='UNRECOGNIZED COMMAND.\nTYPE help for help.' | |
| if re.match('@.+?#\d+$',q): | |
| if re.match('[email protected]$',f[:f.rfind('/')]): | |
| if self.active.value==0: | |
| num=int(q[q.rfind('#')+1:]) | |
| q=q[:q.rfind('#')] | |
| command="beet ls -p "+q[1:].strip() | |
| file_output=subprocess.check_output(command) | |
| files = filter(None, file_output.split('\n')) | |
| fl=files[num-1].strip() | |
| size=os.stat(fl).st_size/(1024*1024) | |
| if size<24: | |
| reply='\n'+'Mailing...'+fl[fl.rfind('\\')+1:]+' File size = '+str(size)+'MB' | |
| rec="%(from)s"%msg | |
| thread.start_new_thread(sendmail,(fl,f[:f.rfind('/')],self.active,self,rec)) | |
| else: | |
| reply='File larger than 25MB, cannot send. This is a limitation of google smtp server. Take it up with them.' | |
| else: | |
| reply='Mail server busy, Try again in a while. type sup? for status' | |
| else: | |
| reply='''Sorry, your email id is coming across all garbled. I suspect this is encryption at some level. Please specify your email id with command like \n@ <search query> #<number> =<email id> | |
| Example: | |
| if your email id is [email protected] and you want the 3rd result of search query 'ravi raga' mailed to you, use command: | |
| @ ravi raga #3 [email protected]''' | |
| #CUSTOM EMAIL | |
| if re.match('@.+#\d\s*[email protected]',q): | |
| if self.active.value==0: | |
| from_id=q[q.rfind('=')+1:] | |
| q=q[:q.rfind('=')] | |
| num=int(q[q.rfind('#')+1:]) | |
| q=q[:q.rfind('#')] | |
| command="beet ls -p "+q[1:].strip() | |
| file_output=subprocess.check_output(command) | |
| files = filter(None, file_output.split('\n')) | |
| fl=files[num-1].strip() | |
| size=os.stat(fl).st_size/(1024*1024) | |
| if size<24: | |
| reply='/me \n'+'Mailing...'+fl[fl.rfind('\\')+1:]+' File size = '+str(size)+'MB'+' to '+from_id | |
| rec="%(from)s"%msg | |
| thread.start_new_thread(sendmail,(fl,from_id,self.active,self,rec)) | |
| else: | |
| reply='File larger than 25MB, cannot send. This is a limitation of google smtp server. Take it up with them.' | |
| else: | |
| reply='Mail server busy, Try again in a while. type sup? for status' | |
| if q=='sup?': | |
| if self.active.value==0: | |
| reply='/me is ready for mailing!!' | |
| else: | |
| reply='/me mail server busy.' | |
| if q=='#start hangman' or q=='# start hangman': | |
| self.hangman.cleanup() | |
| self.hangman.register_player(f) | |
| reply=self.hangman.startgame(f) | |
| if q=='#start quiz'or q=='# start quiz': | |
| self.philo.cleanup() | |
| self.philo.register_player(f) | |
| reply=self.philo.startgame(f) | |
| else: | |
| if(f in pt) and q!='help' and (f not in ppt): | |
| if q=='#stop hangman' or q=='# stop hangman': | |
| reply=self.hangman.delete_player(f) | |
| else: | |
| if q=='#start hangman' or q=='# start hangman': | |
| r1='Old '+self.hangman.delete_player(f)+'\n' | |
| self.hangman.register_player(f) | |
| r2=self.hangman.startgame(f) | |
| reply=r1+r2 | |
| else: | |
| reply=self.hangman.gameresponse(q,f) | |
| if(f not in pt) and q!='help' and (f in ppt): | |
| if q=='#stop quiz' or q=='# stop quiz': | |
| reply=self.philo.delete_player(f) | |
| else: | |
| if q=='#start quiz' or q=='# start quiz': | |
| r1='Old '+self.philo.delete_player(f)+'\n' | |
| self.philo.register_player(f) | |
| r2=self.philo.startgame(f) | |
| reply=r1+r2 | |
| else: | |
| reply=self.philo.gameresponse(q,f) | |
| #help | |
| if q=='help': | |
| if ('[email protected]' in f) or ('[email protected]' in f): | |
| custom='6. Philo quiz -> start quiz by typing #start quiz\nstop quiz by typing #stop quiz\nIf you want question repeated type #repeat while in quiz session' | |
| else: custom='' | |
| reply="""\n1. this service queries music on my library. \nExample: to search for led zeppelin's no quarter, type \n$ led zeppelin no quarter | |
| *The search feature accepts artists, tracknames, album names and combination of these fields.* | |
| 2. To receive email with chosen track as attachment, type : | |
| @ <query that yielded results with $> #<index number> | |
| Example: | |
| me: $ rolling stones sympathy | |
| [email protected]: | |
| [1]The Rolling Stones - 100 Greatest Guitar Solos - Sympathy For The Devil | |
| [2]The Rolling Stones - Beggars Banquet - Sympathy For The Devil | |
| [3]The Rolling Stones - Classic Hits - Sympathy For The Devil | |
| me: @ rolling stones sympathy #2 | |
| [email protected]: | |
| mailing...F:\Music\Discographies\The Rolling Stones Discography MP3@320Kbps)\07 - 1968 -Beggars Banquet\01 Sympathy For The Devil.mp3 | |
| 3.To find out if mail server is busy, type sup? | |
| 4.To list all artists whose discographies are present in the library, type: | |
| list artists | |
| 4.HANGMAN I will play hangman with you if you want. to start session type | |
| #start hangman | |
| note: No other commands will work while in hangman session except _help_ . | |
| To exit hangman session type | |
| #stop hangman | |
| 5.For a random interesting link, type | |
| #random | |
| """+custom | |
| if q==':(': | |
| reply='/me hugs you' | |
| if q=='#random'or q=='# random': | |
| reply=choice(open(r'd:\pythonary\links.txt').read().split('\n')) | |
| msg.reply(reply).send() | |
| print '\n\n' | |
| print 'Reply=',reply | |
| print'\n\n**************************' | |
| if __name__ == '__main__': | |
| logging.basicConfig(level=logging.DEBUG, | |
| format='%(levelname)-8s %(message)s') | |
| xmpp = EchoBot('<emailid>', '<password>') | |
| xmpp.connect(('talk.google.com', 5222)) | |
| xmpp.process(block=True) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment