def create_set_string(songlist):
setstring = ''
for song in songlist:
if setstring == '':
setstring = filehandle.remove_nr(song)
else:
setstring = "%s, %s" % (setstring, filehandle.remove_nr(song))
return setstring
def random_game(filename):
try:
options = filehandle.get_list(filename)
except IOError:
raise IOError("Could not open file for random_game")
randomResult = random.choice(options)
randomResult = filehandle.remove_nr(randomResult)
return randomResult
def acronym_replace(entry):
entry = filehandle.remove_nr(entry)
if entry == 'AP':
entry = 'Apocalypse Please'
elif entry == 'B&H':
entry = 'Butterflies & Hurricanes'
elif entry == 'CE':
entry = 'Citizen Erased'
elif entry == 'CoD':
entry = 'City of Delusion'
elif entry == 'DI':
entry = 'Dead Inside'
elif entry == 'FAWY':
entry = 'Falling Away With You'
elif entry == 'HTAILY':
entry = 'Hate This & I\'ll Love You'
elif entry == 'IBTY':
entry = 'I Belong To You'
elif entry == 'IS':
entry = 'The 2nd Law: Isolated System'
elif entry == 'KoC':
entry = 'Knights of Cydonia'
elif entry == 'MotP':
entry = 'Map of the Problematique'
elif entry == 'MM':
entry = 'Muscle Museum'
elif entry == 'NSC':
entry = 'Neutron Star Collision'
elif entry == 'PiB':
entry = 'Plug In Baby'
elif entry == 'RBS':
entry = 'Ruled By Secrecy'
elif (entry == 'SMBH' or entry == 'Supermassive'):
entry = 'Supermassive Black Hole'
elif entry == 'SS':
entry = 'Stockholm Syndrome'
elif entry == 'TaB':
entry = 'Take A Bow'
elif entry == 'TIRO':
entry = 'Time Is Running Out'
elif entry == 'TOADA':
entry = 'Thoughts of a Dying Atheist'
elif entry == 'UD':
entry = 'Undisclosed Desires'
return entry
def print_set(filename):
try:
fileList = filehandle.get_list(filename)
except IOError:
raise IOError("Error opening file to print")
fileLength = len(fileList)
if fileLength == 0:
print('EMPTY SETLIST')
return ''
else:
if filename == 'text/previous':
print('Printing PREVIOUS')
return "%s: %s" % (filehandle.remove_nr(fileList[0]),
create_set_string(fileList[1:]))
else:
print('Printing set')
return create_set_string(fileList)
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)
