def add_song(argument): songList = argument.split(', ') for song in songList: song = acronym_replace(song) try: filehandle.list_append('text/setlist', song) except IOError: print("Error adding song to setlist")
def set_previous(): setlist = filehandle.get_list('text/setlist') gig = filehandle.get_list('text/gig') tour = filehandle.get_list('text/tour') print('GIG: %s' % (gig)) if len(gig) == 0: print('SetPrevious Error: GIG not found') outputString = 'ERROR: No gig set.' return outputString if len(setlist) == 0: print('SetPrevious Error: Empty setlist') outputString = 'ERROR: Setlist is empty.' return outputString gig = gig[0] tour = tour[0].strip() sys.stdout.write('FORMATTING GIG NAME...') try: gigParse = gig.split('/') if len(gigParse) > 1: i = 1 gigFileName = gigParse[0] while i < len(gigParse): gigFileName = '%s-%s' % (gigFileName, gigParse[i]) i += 1 else: gigFileName = gigParse[0] except: raise RuntimeError("Error formatting gig") sys.stdout.write('\t[DONE]\n') gigFileName = gigFileName.split('\x0A')[0] gigFileName = gigFileName.split('\r')[0] print('GIG FILE NAME: %s' % (gigFileName)) sys.stdout.write('Writing previous setlist...') archivePath = "GigArchive/" + tour + "/" + gigFileName gig = [gig] try: filehandle.put_list('text/previous', gig) filehandle.put_list(archivePath, gig) except IOError: raise IOError("Error writing setlists for set_previous") for entry in setlist: try: filehandle.list_append('text/previous', entry) filehandle.list_append(archivePath, entry) except IOError: raise IOError("Error appending song for set_previous") sys.stdout.write('\t[DONE]\nUpdating database...') database = filehandle.get_list('text/archive.db') for entry in database: if entry.startswith('c!'): song = entry.split('!')[1].split('=')[0] if song in [s.lower() for s in setlist]: count = int(entry.split('!')[1].split('=')[1]) count = count + 1 database[database.index(entry)] = 'c!%s=%d' % (song, count) elif entry.startswith('lp!'): song = entry.split('!')[1].split('::')[0] if song in [s.lower() for s in setlist]: database[database.index(entry)] = 'lp!%s::%s' % (song, archivePath) filehandle.put_list('text/archive.db', database) sys.stdout.write('\t[DONE]\nTweeting setlist...') twit.tweet_setlist() sys.stdout.write('\t[DONE]\n') outputString = "Current setlist copied over as previous set." return outputString
def command_run(text, irc, commands): if text.IRCcmd == "JOIN" and irc.joinmsg() and irc.isAwake(): if text.nick != irc.nick(): irc.msg("Welcome to the sexy plane %s. " "Enter !commands to view the bot functions." % text.nick) if text.nick in irc.modlist: if text.command in ("!wake\r\n", "!wake"): irc.wake() irc.resumeTimers() irc.msg("BellamyBot is online.") elif text.command in ("!sleep\r\n", "!sleep"): irc.sleep() irc.pauseTimers() irc.msg("Sleep mode activated") elif text.command == "!joinmsg": if text.argument in ("on\r\n", "on"): irc.activateJoinMsg() print("Join messages ON") elif text.argument in ("off\r\n", "off"): irc.deactivateJoinMsg() print("Join messages OFF") elif text.command == "!timers": if text.argument in ("on\r\n", "on"): irc.resumeTimers() elif text.argument in ("off\r\n", "off"): irc.pauseTimers() elif text.command == "!gamemode": if text.argument in ("on\r\n", "on"): irc.activateGames() print("Game mode ON") elif text.argument in ("off\r\n", "off"): irc.deactivateGames() print("Game mode OFF") elif text.command == "!setgig": try: filehandle.clear_file('text/gig') filehandle.list_append('text/gig', text.argument) except IOError as e: print(e) elif text.command == "!settour": try: filehandle.clear_file('text/tour') filehandle.list_append('text/tour', text.argument) except IOError as e: print(e) # Setlist controls elif text.command == "!add": txtfunctions.add_song(text.argument) undo.add() elif text.command == "!exp": original = filehandle.remove_nr(text.argument) check = txtfunctions.acronym_replace(text.argument) if check == original: irc.msg("There is no expansion for " + original) else: irc.msg(original + " expands to " + check) elif text.command in ("!clearset\r\n", "!clearset"): try: filehandle.clear_file('text/setlist') undo.refresh() except IOError as e: print(e) irc.msg("Current setlist has been cleared.") elif text.command in ("!pop\r\n", "!pop"): try: txtfunctions.song_pop() except IOError as e: print(e) return except IndexError as i: print(i) return undo.add() irc.msg("The last song has been erased") elif text.command == "!insert": try: response = txtfunctions.insert_song(text.argument) except IOError as e: print(e) return if response.find('ERROR') == -1: undo.add() irc.msg(response) elif text.command == "!replace": try: response = txtfunctions.replace_song(text.argument) except IOError as e: print(e) if response.find('ERROR') == -1: undo.add() irc.msg(response) elif text.command == "!delete": try: response = txtfunctions.delete_song(text.argument) except IOError as e: print(e) if response != 'Could not delete any songs.': undo.add() irc.msg(response) elif text.command in ("!setprevious\r\n", "!setprevious"): try: response = txtfunctions.set_previous() except IOError as e: print(e) except RuntimeError as r: print(r) irc.msg(response) elif text.command in ("!undo\r\n", "!undo"): if undo.undo(): irc.msg("Undid last change to setlist.") elif text.command in ("!redo\r\n", "!redo"): if undo.redo(): irc.msg("Redid last change to setlist.") elif text.command == "!cmd": cmd.function(irc, commands, text.argument) # All user commands if text.command in ("!bot\r\n", "!bot"): statePhrase = ("BellamyBot version %s created by Kueller917. Status: " % irc.version()) if irc.isAwake(): statePhrase = statePhrase + "ONLINE" else: statePhrase = statePhrase + "OFFLINE" irc.msg(statePhrase) elif text.command in ("!gig\r\n", "!gig"): try: gig = filehandle.get_list('text/gig') except IOError as e: print(e) irc.msg(gig[0]) elif text.command in ("!previous\r\n", "!previous"): gigarchive.print_recent_setlist(irc) elif text.command == "!message": try: filehandle.list_append('text/ircmsg', "%s: %s" % (text.nick, filehandle.remove_nr(text.argument))) except IOError as e: print(e) if len(irc.owners) > 0: irc.memo(irc.owners[0], "You have a message from %s" % text.nick) elif text.command in ("!source\r\n", "!source"): sourceMsg = ("Get your own copy of BellamyBot today! %s" % irc.source()) irc.msg(sourceMsg) # State dependent commands if irc.isAwake(): # General commands if text.command in ("!setlist\r\n", "!setlist"): setmsg = "CURRENT SETLIST: " try: currentset = txtfunctions.print_set('text/setlist') except IOError as e: print(e) if currentset == '': currentset = "...is empty" setmsg = setmsg + currentset irc.msg(setmsg) elif text.command == "!count": gigarchive.song_count(irc, text.argument) elif text.command == "!lastplayed": song = txtfunctions.acronym_replace(text.argument) gigarchive.last_played(irc, song) elif text.command == "!findset": gigarchive.find_setlist(irc, text.argument.strip()) elif text.command == "!loadset": date = text.argument.strip() setlist = gigarchive.print_set_by_date(irc, date) elif text.command == "!info": try: infolist = filehandle.get_list('text/info') except IOError: print('Error opening file info') return command = text.argument.strip() for line in infolist: if line.split(':')[0] == command: irc.msg(line.split(':')[1]) elif text.command == "!eval": ev = boteval.BotEval() ev.eval(irc, text.argument) elif text.command in ("!commands\r\n", "!commands"): irc.msg("Set commands: !gig, !setlist, !previous, !findset, " "!loadset, !count, !lastplayed.") irc.msg("Other: !bot, !source, !closer, !opener, !realfan, " "!roulette, !setfm, !ru-roulette, !setgen. Use !info " "for a description of any command.") elif text.command.strip() in commands: cmd.execute(irc, commands, text) if irc.gamesActive(): # Games if text.command in ("!closer\r\n", "!closer"): try: closer = musegames.random_game('text/gigcloser') except IOError as e: print(e) irc.msg("%s\'s gig has closed with %s!" % (text.nick, closer)) elif text.command in ("!opener\r\n", "!opener"): try: opener = musegames.random_game('text/opener') except IOError as e: print(e) irc.msg("%s\'s gig has opened with %s!" % (text.nick, opener)) elif text.command in ("!realfan\r\n", "!realfan"): if randint(0,1): irc.msg("%s is a REAL FAN. Good for you." % text.nick) else: irc.msg("%s is not a REAL FAN. Go away." % text.nick) elif text.command in ("!roulette\r\n", "!roulette"): output = musegames.T2L_roulette() if output != -1: irc.msg(output) else: irc.greenOn() irc.greenNick(text.nick) irc.msg("You landed on GREEN! Type !green to get your song") elif text.command in ("!green\r\n", "!green"): if irc.greenActive(): if irc.checkGreen(text.nick): irc.greenOff() irc.greenNick(None) irc.msg(musegames.roulette_green(text.nick)) else: irc.msg("You did not land on green.") elif text.command in ("!setfm\r\n", "!setfm"): setlist = setlistfm.get_setlist(irc.mbid()) for messagePart in setlist: irc.msg(messagePart) elif text.command in ("!manson\r\n", "!manson"): try: mansons = musegames.manson_game(text.nick) except IOError as e: print(e) except IndexError as i: print(i) irc.msg(mansons) elif text.command in ("!ru-roulette\r\n", "!ru-roulette"): timercommands.russian_roulette(irc, text.nick) elif text.command in ("!setgen\r\n", "!setgen"): setlistgenerator.generate(irc)