#!/usr/bin/env python # -*- coding: utf-8 -*- import requests import discord import random import datetime import time import re import pickle import logging import thread import traceback import sys import wikipedia from subprocess import call from dateutil.parser import parse from discord.object import Object from discord.channel import PrivateChannel from ago import human import simplejson as json from collections import defaultdict from nltk.tag import pos_tag import wolframalpha import sqlite3 from blackjack import Blackjack import data from pankration import Pankration, HuntResponse, Jobs VERSION = 2.3 quitting = False grant_hour = 11 conn = sqlite3.connect('db.sqlite3') credentials = 'creds.json' muted_until = datetime.datetime.now() client = discord.Client() wolf = {} logging.basicConfig(filename='hellsbot.log', level=logging.WARNING) registered_commands = {'!help': 'do_help', '!commands': 'do_help', '!shutup': 'do_shutup', '!roll': 'do_roll', '!lastseen': 'do_lastseen', '!youtube': 'do_youtube', '!image': 'do_image', '!gif': 'do_gif', '!gameslist': 'do_gameslist', '!gamelist': 'do_gameslist', '!aliases': 'do_alias', '!alias': 'do_alias', '!addalias': 'do_addalias', '!games': 'do_games', '!reloadbot': 'do_reload', '!restartbot': 'do_reload', '!rebootbot': 'do_reload', '!whoplayed': 'do_whoplayed', '!gimmecredits': 'do_gimmecredits', '!gimmecredit': 'do_gimmecredits', '!grantcredits': 'do_grantcredits', '!ticketrank': 'do_ticketrank', '!startraffle': 'do_startraffle', '!raffle': 'do_raffle', '!pastraffles': 'do_pastraffle', '!pastraffle': 'do_pastraffle', '!buyticket': 'do_buyticket', '!balance': 'do_balance', '!slotsrules': 'do_slotsrules', '!slots': 'do_slots', '!hit': 'do_bj_hit', '!draw': 'do_bj_hit', '!stay': 'do_bj_stay', '!stand': 'do_bj_stay', '!bet': 'do_bj_bet', '!msg': 'do_msg', '!addfortune': 'do_addfortune', '!fortune': 'do_fortune', '!question': 'do_question', '!addjoke': 'do_addjoke', '!joke': 'do_joke', '!secret': 'do_secret', '!bemyirlwaifu': 'do_waifu', 'HILLARY 2016': 'do_hillary', '!squid': 'do_squid', '!stars': 'do_stars', '!rigged': 'do_rigged', '!listzones': 'do_list_zones', '!huntmonster': 'do_hunt_monster', } ##################### # Utility Functions ##################### def log(message): try: logging.warning("{} - {}".format(datetime.datetime.now().strftime('%Y/%m/%d %H:%M:%S'), message)) except: pass def send_message(client, target, message): if type(target) is not discord.channel.PrivateChannel: if target.id == '121468616414724100': return elif datetime.datetime.now() < muted_until: return client.send_message(target, message) def format_exception(e): exception_list = traceback.format_stack() exception_list = exception_list[:-2] exception_list.extend(traceback.format_tb(sys.exc_info()[2])) exception_list.extend(traceback.format_exception_only(sys.exc_info()[0], sys.exc_info()[1])) exception_str = "Traceback (most recent call last):\n" exception_str += "".join(exception_list) # Removing the last \n exception_str = exception_str[:-1] return exception_str def leaders(xs, top=20): counts = defaultdict(int) for x in xs: counts[x] += 1 return sorted(counts.items(), reverse=True, key=lambda tup: tup[1])[:top] def byteify(input): if isinstance(input, dict): return {byteify(key): byteify(value) for key, value in input.iteritems()} elif isinstance(input, list): return [byteify(element) for element in input] elif isinstance(input, unicode): return input.encode('utf-8') else: return input def search_youtube(query): query_string = {"search_query": query} r = requests.get("http://www.youtube.com/results", params=query_string) search_results = re.findall(r'href=\"\/watch\?v=(.{11})', r.content) log("http://www.youtube.com/watch?v=" + search_results[0]) return "http://www.youtube.com/watch?v=" + search_results[0] def search_google_images(query, animated=False): headers = {'User-Agent': "Mozilla/5.0 (X11; FreeBSD amd64; rv:12.0) Gecko/20100101 Firefox/12.0"} query_string = {"safe": "off", "tbm": "isch", "q": query} if animated: query_string = {"safe": "off", "tbm": "isch", "q": query, 'tbs': 'itp:animated'} r = requests.get("http://www.google.com/search", params=query_string, headers=headers) start_idx = r.content.find('imgurl=') + 7 if start_idx > 0: search_result = r.content[start_idx:r.content.find('&', start_idx)] if '/revision/' in search_result: search_result = search_result[:search_result.find('/revision/')] if '%' in search_result: search_result = search_result[:search_result.find('%')] log(search_result) return search_result return "boo you fail.." def check_msg_queue(client): messages = data.db_get_messages() if messages: for message in messages: try: member = data.db_get_member(filter(unicode.isalnum, message['message_to'])) if member: if message['message_to'] == member['discord_mention']: log("Found Message: {} - {} - Status: {} Channel: {}".format(message['message_to'], member['discord_mention'], member['status'], message['channel'])) if member['status'] == 'online': client.send_message(Object(message['channel']), '{}, {} asked me to tell you "{}"'.format(message['message_to'], message['message_from'], message['message'])) data.db_delete_sent_message(message['message_id']) except Exception as e: log("{}\nFailed to send message: {}".format(format_exception(e), message['message_id'],)) return def do_roll(client, message_parts, message): request = message_parts[0] count = 1 dice = 100 if request.strip() != '': if 'd' in request: dice_parts = request.split('d') if len(dice_parts) == 2: if dice_parts[0].isdigit() and dice_parts[1].isdigit(): count = int(dice_parts[0]) dice = int(dice_parts[1]) if count > 100000000000000000: send_message(client, message.channel, '{} stop fucking around with those stupid numbers...'.format(message.author.mention())) return if count > 100: send_message(client, message.channel, '{} 100 is the largest number of dice I will roll.'.format(message.author.mention())) return roll_results = [] for i in range(count): roll_results.append(random.randint(1, dice)) out_string = '{} your roll {}d{}: {} = {}'.format(message.author.mention(), count, dice, '+'.join(str(r) for r in roll_results), sum(roll_results)) send_message(client, message.channel, out_string) return def do_lastseen(client, message_parts, message): username = ' '.join(message_parts).replace('@', '').lower() member = data.db_get_member(username=username) if member: out_string = '' if member['is_afk'] == 1: out_string = 'Went AFK at: {}\n'.format(member['afk_at']) elif member['status'] == 'offline': out_string = 'Currently Offline\n' else: out_string = 'Currently Online\n' out_string += 'Last Status: {} at {} which was {}\nPrevious Status: {}\n'.format(member['status'], member['status_change_at'], human(datetime.datetime.strptime(member['status_change_at'], '%Y/%m/%d %H:%M:%S')), member['prev_status']) send_message(client, message.channel, 'Last Information on {}:\n{}'.format(byteify(username), out_string)) else: send_message(client, message.channel, 'I don\'t have any data on {} yet {}'.format(byteify(username), message.author.mention())) return def do_help(client, message_parts, message): send_message(client, message.channel, """{} Available Commands: You can ask compound or questions and I will choose. Example: HellsBot Rui is a Faggot or Rui is a faggot? User Info: !aliases <username> - Returns a list of all aliases a user has set for themselves. !addalias <alias> - Adds an alias to your list of aliases. !lastseen <username> - Returns info on when the user was last seen and their status. Messages: !msg <username> in 5 minutes Tea is ready !msg <username> in 45 seconds Your finished masterbating !msg <username> in 2 hours The movie is over !msg <username> on 12/22/2015 Happy Birthday! Games: !games <username> - Returns a list of games played for a username. !gameslist <count> - Returns a list of the top 20 games and the number of people who have played that game. if you pass a limit it will show that many games instead. !whoplayed <gamename> - Returns a list of players who have played the game. Minigames: !gimmecredits - Gives you some extra credits in case you run out. !bet <amount> - Start a game of BlackJack. !slots <amount> - Spin the slot machine (max 10 credit bet). !hit - Draw a card !stand - Show the cards """.format(message.author.mention())) send_message(client, message.channel, """ !balance - Lists your current credits and tickets !buyticket - Purchases a raffle ticket for 100 credits !raffle - Shows information about the current raffle !ticketrank - Shows the percentage of the tickets on the system Spam: !youtube <search term> - Returns the first video from the search results for the search term. !gif <search term> - Returns the first gif from the search results. !image <search term> - Returns the first image from the search results. Stuff: !addfortune <fortune> - Adds a new fortune. !fortune - Returns your fortune. !addjoke <joke> - Adds a new joke (it can be multiline but must all be in a single message). !joke - Returns a random joke. !roll <1d20> - Roll X number of dice of size X. 1d20 returns 1 roll 1-20. 3d6 returns 3 rolls of 1-6 etc... !secret !shutup - disables all image / gif / youtube spam for 5 minutes !rigged - Cause Study is no good at betting. !bemyirlwaifu""".format(message.author.mention())) def do_shutup(client, message_parts, message): global muted_until muted_until = datetime.datetime.now() + datetime.timedelta(minutes=5) send_message(client, message.channel, 'All image / gif / youtube spam disabled for 5 minutes') return def do_youtube(client, message_parts, message): if datetime.datetime.now() < muted_until: return send_message(client, message.channel, search_youtube(' '.join(message_parts))) return def do_image(client, message_parts, message): if datetime.datetime.now() < muted_until: return send_message(client, message.channel, search_google_images(' '.join(message_parts))) return def do_gif(client, message_parts, message): if datetime.datetime.now() < muted_until: return send_message(client, message.channel, search_google_images(' '.join(message_parts), True)) return def do_gameslist(client, message_parts, message): limit = 20 log("parts: {} {}".format(message_parts,message)) if len(message_parts) > 0 and message_parts[0].isdigit(): limit = int(message_parts[0]) games_list = data.db_get_games_list(limit) out_string = '' for game in games_list: out_string += ' {} - {}\n'.format(game[1], byteify(game[0])) send_message(client, message.channel, 'The games I have seen people playing are: ') while len(out_string) > 0: send_message(client, message.channel, out_string[:1900]) out_string = out_string[1900:] return def do_alias(client, message_parts, message): if len(message_parts) > 0: username = ' '.join(message_parts).replace('@', '').lower() member = data.db_get_member(username=username) else: username = message.author.name member = data.db_get_member(message.author.id) if member: aliases = data.db_get_aliases(member['member_id']) if aliases: send_message(client, message.channel, '{} has the following aliases: {}'.format(byteify(username), byteify(', '.join(aliases)))) else: send_message(client, message.channel, 'No known alises for {} yet {}'.format(byteify(username), message.author.mention())) else: send_message(client, message.channel, 'I don\'t know who you are speaking of {}!'.format(message.author.mention())) return def do_addalias(client, message_parts, message): alias = ' '.join(message_parts) username = message.author.name.lower() member = data.db_get_member(username=username) if member: data.db_add_aliases(member['member_id'], alias) send_message(client, message.channel, '{} has been added to your aliases'.format(byteify(alias))) else: send_message(client, message.channel, 'Something horrible happened and it is all your fault, try logging out / on again or play a game. (or fuck off i dunno i\'m just an error message. Who am I to tell you how to run your life...)') return def do_games(client, message_parts, message): if len(message_parts) > 0: username = ' '.join(message_parts).replace('@', '').lower() else: username = message.author.name games_list = data.db_get_games(username) if games_list: games = ', '.join(games_list) games = games.replace("FINAL FANTASY XIV", "**FINAL FANTASY XIV**") log(byteify(games)) send_message(client, message.channel, 'I have seen {} playing: {}'.format(byteify(username), byteify(games))) else: send_message(client, message.channel, 'I don\'t have any data on {} yet {}'.format(byteify(username), message.author.mention())) def do_reload(client, message_parts, message): if message.author.id != '78767557628133376': send_message(client, message.channel, "You shouldn't be calling this. You are clearly looking to piss Hellsbreath off.") return call(["service", "hellsbot", "restart"]) def do_whoplayed(client, message_parts, message): game_name = ' '.join(message_parts) member_list = data.db_get_whoplayed(game_name) if not member_list: send_message(client, message.channel, 'I don\'t have any data on {} yet {}'.format(byteify(game_name), message.author.mention())) else: out_string = '' for member in member_list: out_string += ' {} - {}\n'.format(byteify(member[1]), byteify(member[0])) send_message(client, message.channel, 'Below is a list of people who have played {} and the number of times they have launched the game:'.format(byteify(game_name),)) while len(out_string) > 0: send_message(client, message.channel, out_string[:1900]) out_string = out_string[1900:] return def do_gimmecredits(client, message_parts, message): member = data.db_get_member(message.author.id) if not member: send_message(client, message.author, "There was a problem looking up your information.") else: credits = data.db_get_credit(member['member_id']) if credits < 5: amount = random.randint(5, 50) data.db_update_credit(member['member_id'], amount) send_message(client, message.author, "You have been given {} credits.".format(amount,)) else: send_message(client, message.author, "You already have credits. Stop begging.") return def do_grantcredits(client, message_parts, message, channel=None): if not channel and message.author.id != '78767557628133376': send_message(client, message.channel, "You are not Hellsbreath. Use !gimmecredits to get a few extra if you run out.") return members = data.db_get_all_members() if len(members) < 0: send_message(client, message.channel, "There was a problem looking up your information.") else: for member in members: credits = data.db_get_credit(member['member_id']) if credits < 100: data.db_update_credit(member['member_id'], 100) if channel: send_message(client, channel, "{} has been given {} credits.".format(member['member_name'], 100)) else: send_message(client, message.channel, "{} has been given {} credits.".format(member['member_name'], 100)) return def do_ticketrank(client, message_parts, message): members = data.db_get_all_members() if len(members) < 0: send_message(client, message.channel, "There was a problem looking up your information.") else: ticket_count = 0 for member in members: if member['discord_id'] != '78767557628133376': ticket_count += member['tickets'] if ticket_count == 0: send_message(client, message.channel, "No Tickets have been sold for this raffle.") return out_string = "" for member in members: if member['tickets'] > 0 and member['discord_id'] != '78767557628133376': percent = (float(member['tickets']) / float(ticket_count)) * 100.0 out_string += "{} - {}%\n".format(byteify(member['member_name']), int(percent)) send_message(client, message.channel, out_string) return def do_startraffle(client, message_parts, message): if message.author.id != '78767557628133376': send_message(client, message.channel, "You are not Hellsbreath. Go die in an especially hot fire.") return members = data.db_get_all_members() if len(members) < 0: send_message(client, message.channel, "There was a problem looking up your information.") else: ticket_count = 0 for member in members: if member['discord_id'] != '78767557628133376': ticket_count += member['tickets'] if ticket_count == 0: send_message(client, message.channel, "No Tickets have been sold for this raffle.") return out_string = "The final standings are as follows: \n\n" ticket_reel = [] for member in members: if member['tickets'] > 0 and member['discord_id'] != '78767557628133376': ticket_reel += [member['discord_mention']] * member['tickets'] percent = (float(member['tickets']) / float(ticket_count)) * 100.0 out_string += "{} - {}%\n".format(byteify(member['member_name']), int(percent)) creds = json.loads(json_data) raffles = creds['raffles'] title = "" key = "" dlc = "" random_keys = creds['randomkeys'] for key, value in raffles.iteritems(): if value['current'] == 1: title = value['title'] game_key = value['key'] if 'dlc' in value: dlc = value['dlc'] if len(ticket_reel) > 0: winner = random.choice(ticket_reel) while winner in ticket_reel: ticket_reel.remove(winner) send_message(client, message.channel, "\n\n\n**The winner is....\n\n{}!**".format(byteify(winner))) if len(ticket_reel) > 0: second = random.choice(ticket_reel) while second in ticket_reel: ticket_reel.remove(second) time.sleep(0.5) send_message(client, message.channel, "\n\n*2nd Place:* {}".format(byteify(second))) if len(ticket_reel) > 0: third = random.choice(ticket_reel) time.sleep(0.5) send_message(client, message.channel, "\n*3rd Place:* {}".format(byteify(third))) for member in client.get_all_members(): log(member.id) if member.id == '78767557628133376': priv_message = "1st: {} key: {}\n2nd: {} keys: {}\n3rd: {} keys: {}".format(byteify(winner), game_key, byteify(second), ' '.join(random_keys[0:2]), byteify(third), random_keys[3]) log(priv_message) send_message(client, member, priv_message) break return def do_pastraffle(client, message_parts, message): creds = json.loads(json_data) raffles = creds['raffles'] out_str = "" for key, value in raffles.iteritems(): if value['current'] == 0: out_str += "{} - {}\n".format(key, value['title']) send_message(client, message.channel, "Past Raffles:\n\n{}".format(out_str)) return def do_raffle(client, message_parts, message): creds = json.loads(json_data) raffles = creds['raffles'] title = "" description = "" link = "" date = "" for key, value in raffles.iteritems(): if value['current'] == 1: date = key title = value['title'] description = value['description'] link = value['link'] send_message(client, message.channel, """Current Raffle Item: **1st Place** Game: **{}** Description: *{}* {} **2nd Place** 2 Random Steam Keys **3rd Place** 1 Random Steam Key Raffle Date: **{}(ish)** You will be contacted if you win. To win you must purchase tickets with the !buyticket command for 100 credits. You can get extra credits by playing !slots <amount> and !bet <amount> on BlackJack. **Disclaimer:** *If anything should go wrong you get no refund and there is no guarantee or warrantee on anything. 1 prize per person no matter how many tickets you have.* """.format(title, byteify(description), link, date)) return def do_buyticket(client, message_parts, message): log("Buying Ticket") member = data.db_get_member(message.author.id) log("Member: {}".format(member,)) if not member: log("there") send_message(client, message.author, "There was a problem looking up your information.") else: log("here") log("Buying Ticket for: {}".format(byteify(member['member_name']))) result, response = data.db_buy_ticket(member['member_id'], 1) log("Buy Result: {} - {}".format(result, response)) if not result: send_message(client, message.author, response) return credits = data.db_get_credit(member['member_id']) send_message(client, message.author, "Raffle ticket purchased. Tickets: {} Credits: {}".format(response, credits)) return def do_balance(client, message_parts, message): member = data.db_get_member(message.author.id) if not member: send_message(client, message.author, "There was a problem looking up your information.") else: send_message(client, message.author, "Credits: {}\nTickets: {}".format(member['credits'], member['tickets'])) return def do_slotsrules(client, message_parts, message): send_message(client, message.channel, """Paying Combinations: :moneybag:\t:moneybag:\t:moneybag:\t\t\t pays\t250 :bell:\t:bell:\t:bell:/:moneybag:\tpays\t20 :diamonds:\t:diamonds:\t:diamonds:/:moneybag:\tpays\t14 :spades:\t:spades:\t:spades:/:moneybag:\tpays\t10 :cherries:\t:cherries:\t:cherries:\t\t\t pays\t7 :cherries:\t:cherries:\t -\t\t\t\t pays\t5 :cherries:\t -\t\t -\t\t\t\t pays\t2 :black_square_button:\tblank space All payouts are in credits. Each pull costs 1 credit. Max Bet: 10 credits To Play: !slots <bet>""") return def do_slots(client, message_parts, message): member = data.db_get_member(message.author.id) if not member: send_message(client, message.author, "There was a problem looking up your information.") return elif type(message.channel) is not discord.channel.PrivateChannel: send_message(client, message.author, "You must make all bets / gaming via private message.") return bet_amount = message.content[7:] log("Member: {} Slots Bet: {}".format(member['member_name'], bet_amount)) if not bet_amount.isdigit() or int(bet_amount) > 10 or int(bet_amount) < 0: send_message(client, message.author, "Please provide a bet amount up to 10 credits.") return result, error_message = data.db_update_credit(member['member_id'], -int(bet_amount)) if not result: send_message(client, message.author, error_message) return reel1 = [':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':cherries:', ':cherries:', ':cherries:', ':spades:', ':spades:', ':spades:', ':hearts:', ':hearts:', ':diamonds:', ':bell:', ':moneybag:'] reel2 = [':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':cherries:', ':cherries:', ':cherries:', ':spades:', ':spades:', ':spades:', ':hearts:', ':hearts:', ':diamonds:', ':bell:', ':moneybag:'] reel3 = [':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':black_square_button:', ':cherries:', ':cherries:', ':cherries:', ':spades:', ':spades:', ':spades:', ':hearts:', ':hearts:', ':diamonds:', ':bell:', ':moneybag:'] #:moneybag: :bell: :diamonds: :spades: :hearts: val1 = random.choice(reel1) val2 = random.choice(reel2) val3 = random.choice(reel3) winnings = 0 reels = [val1, val2, val3] cherries = reels.count(":cherries:") spades = reels.count(":spades:") diamonds = reels.count(":diamonds:") bells = reels.count(":bell:") moneybags = reels.count(":moneybag:") out_string = "" if moneybags == 3: out_string += "JACKPOT!!" winnings = 250 elif bells == 3 or (bells == 2 and moneybags == 1): winnings = 20 elif diamonds == 3 or (diamonds == 2 and moneybags == 1): winnings = 14 elif spades == 3 or (spades == 2 and moneybags == 1): winnings = 10 elif cherries == 3: winnings = 7 elif cherries == 2: winnings = 5 elif cherries == 1: winnings = 2 out_string = """| {} | {} | {} |\n\n""".format(val1, val2, val3) winnings = int(winnings * int(bet_amount)) if winnings > 0: out_string += "You Won! Total Winnings: {}".format(winnings,) log("Member: {} Wins: {}".format(member['member_name'], winnings)) else: out_string += "You lose. Total Winnings: {}".format(winnings,) if winnings > 0: result, error_message = data.db_update_credit(member['member_id'], winnings) if not result: send_message(client, message.author, error_message) return credits = data.db_get_credit(member['member_id']) out_string += "\nCredits: {}".format(credits) send_message(client, message.author, out_string) return def do_bj_hit(client, message_parts, message): member = data.db_get_member(message.author.id) if not member: send_message(client, message.author, "There was a problem looking up your information.") elif type(message.channel) is not discord.channel.PrivateChannel: send_message(client, message.author, "You must make all bets / gaming via private message.") else: state = data.db_get_minigame_state(member['member_id'], 'blackjack') if state: out_string = "" bj = pickle.loads(str(state)) bj.draw() out_string += bj.print_hand() actions = bj.get_actions() if len(actions) > 0: out_string += '\n\nPlease choose an option [{}]\n'.format(', '.join(actions)) data.db_add_minigame(member['member_id'], 'blackjack', bj.serialize()) else: win, response = bj.is_win() out_string += "\n\n" + bj.print_hand(show_dealer=True) out_string += "\n\n" + response.format(win,) data.db_delete_minigame_state(member['member_id'], 'blackjack') log("{} - {}".format(member['member_name'], response.format(win,))) result, error_message = data.db_update_credit(member['member_id'], int(win)) if not result: send_message(client, message.author, error_message) return credits = data.db_get_credit(member['member_id']) out_string += "\nCredits: {}".format(credits) send_message(client, message.author, out_string) else: send_message(client, message.author, "You must start a game with !bet before you can ask for a new card.") return def do_bj_stay(client, message_parts, message): member = data.db_get_member(message.author.id) if not member: send_message(client, message.author, "There was a problem looking up your information.") elif type(message.channel) is not discord.channel.PrivateChannel: send_message(client, message.author, "You must make all bets / gaming via private message.") else: state = data.db_get_minigame_state(member['member_id'], 'blackjack') if state: out_string = "" bj = pickle.loads(str(state)) win, response = bj.is_win() out_string += "\n\n" + bj.print_hand(show_dealer=True) out_string += "\n\n" + response.format(win,) data.db_delete_minigame_state(member['member_id'], 'blackjack') log("{} - {}".format(member['member_name'], response.format(win,))) result, error_message = data.db_update_credit(member['member_id'], int(win)) if not result: send_message(client, message.author, error_message) return credits = data.db_get_credit(member['member_id']) out_string += "\nCredits: {}".format(credits) send_message(client, message.author, out_string) return def do_bj_bet(client, message_parts, message): member = data.db_get_member(message.author.id) if not member: send_message(client, message.author, "There was a problem looking up your information.") elif type(message.channel) is not discord.channel.PrivateChannel: send_message(client, message.author, "You must make all bets / gaming via private message.") else: state = data.db_get_minigame_state(member['member_id'], 'blackjack') if state: send_message(client, message.author, "You are already playing a game!") out_string = "" bj = pickle.loads(str(state)) out_string += bj.print_hand() actions = bj.get_actions() if len(actions) > 0: out_string += '\n\nPlease choose an option [{}]\n'.format(', '.join(actions)) data.db_add_minigame(member['member_id'], 'blackjack', bj.serialize()) else: win, response = bj.is_win() out_string += "\n\n" + bj.print_hand(show_dealer=True) out_string += "\n\n" + response.format(win,) data.db_delete_minigame_state(member['member_id'], 'blackjack') log("{} - {}".format(member['member_name'], response.format(win,))) result, error_message = data.db_update_credit(member['member_id'], int(win)) if not result: send_message(client, message.author, error_message) return credits = data.db_get_credit(member['member_id']) out_string += "\nCredits: {}".format(credits) send_message(client, message.author, out_string) return out_string = "" bet_amount = message.content[5:] log("Member: {} Bet: {}".format(member['member_name'], bet_amount)) if not bet_amount.isdigit(): send_message(client, message.author, "Please provide a bet amount. !bet 10") return result, error_message = data.db_update_credit(member['member_id'], -int(bet_amount)) if not result: send_message(client, message.author, error_message) return out_string += "Welcome to BlackJack! :flower_playing_cards: You have placed a bet of: {}\n".format(bet_amount) bj = Blackjack(bet_amount) out_string += bj.print_hand() actions = bj.get_actions() if len(actions) > 0: out_string += '\n\nPlease choose an option [{}]\n'.format(', '.join(actions)) data.db_add_minigame(member['member_id'], 'blackjack', pickle.dumps(bj)) else: win, response = bj.is_win() out_string += "\n\n" + bj.print_hand(show_dealer=True) out_string += "\n\n" + response.format(win,) data.db_delete_minigame_state(member['member_id'], 'blackjack') log("{} - {}".format(member['member_name'], response.format(win,))) result, error_message = data.db_update_credit(member['member_id'], int(win)) if not result: send_message(client, message.author, error_message) return credits = data.db_get_credit(member['member_id']) out_string += "\nCredits: {}".format(credits) send_message(client, message.author, out_string) return def do_msg(client, message_parts, message): channel = message.channel author = message.author username = '' try: # TODO: Switch message_bits with message_parts message_bits = message.content.split(" ") msg_datetime = datetime.datetime.now() msg_idx = 2 if message_bits[2] == 'in' and message_bits[3].isdigit(): time = int(message_bits[3]) msg_idx = 4 if message_bits[4].startswith('sec'): msg_datetime = msg_datetime + datetime.timedelta(seconds=time) msg_idx = 5 elif message_bits[4].startswith('hour'): msg_datetime = msg_datetime + datetime.timedelta(hours=time) msg_idx = 5 else: # minutes by default msg_datetime = msg_datetime + datetime.timedelta(minutes=time) msg_idx = 5 elif message_bits[2] == 'on': try: tmp_date = parse(message_bits[3]) msg_datetime = tmp_date msg_idx = 4 except ValueError: send_message(client, channel, 'Your shitty message has been rejected {}. Next time learn how to date...MM\\DD\\YYYY'.format(message.author.mention())) return username = message_bits[1] user_mention = '' # TODO: have it look in the database. Do this AFTER on startup we add all users. for member in client.get_all_members(): if username.lower() == member.name.lower(): user_mention = member.mention() user_id = member.id break if user_mention == '': send_message(client, channel, 'Your shitty message has been rejected {}. That user does not exist.'.format(message.author.name)) return msg_text = byteify(' '.join(message_bits[msg_idx:])) message = {'user_id': user_id, 'channel': channel.id, 'delivery_time': msg_datetime.strftime('%Y/%m/%d %H:%M:%S'), 'message': msg_text} log("Message: %s" % byteify(message)) data.db_add_message(msg_text, msg_datetime.strftime('%Y-%m-%d %H:%M:%S'), channel.id, author.mention(), user_mention, user_id) # print("Data: %s" % data) #test_ch = Object(channel.id) #client.send_message(test_ch, 'Test Message {}.'.format(author)) except Exception as e: send_message(client, channel, 'Your shitty message has been rejected {}. {}'.format(author.name, format_exception(e))) return if msg_datetime < datetime.datetime.now(): send_message(client, channel, '{} your message will be delivered to {} as soon as they are available.'.format(author.name, user_mention)) else: send_message(client, channel, '{} your message will be delivered to {} {}.'.format(author.name, user_mention, human(msg_datetime))) check_msg_queue(client) return def do_addfortune(client, message_parts, message): try: fortune = message.content[9:] if 'aa737a5846' in fortune: send_message(client, message.channel, '{} you stop it, you are a pedofile, stop looking at little girls.'.format(message.author.mention())) return date_added = datetime.datetime.now().strftime('%Y/%m/%d %H:%M:%S') c = conn.cursor() c.execute("INSERT INTO fortunes (fortune, date_added) VALUES (?, ?)", (fortune, date_added)) conn.commit() log("Added fortune") except Exception as e: log(e.message) send_message(client, message.channel, 'Your shitty fortune has been rejected {}.'.format(message.author.mention())) return send_message(client, message.channel, 'Your shitty fortune has been added {}.'.format(message.author.mention())) return def do_fortune(client, message_parts, message): fortune = None try: c = conn.cursor() # TODO: Move this shit to data fortune = c.execute("SELECT fortune FROM fortunes ORDER BY RANDOM() LIMIT 1;").fetchone()[0] log(fortune) except Exception as e: log(e) pass if not fortune: send_message(client, message.channel, 'Try adding a fortune with "!addfortune <fortune>" {}!'.format(message.author.mention())) else: send_message(client, message.channel, '{} Your fortune is... {}'.format(message.author.mention(), byteify(fortune))) return def do_question(client, message_parts, message): question = message.content[10:] if "is it gay" in question: send_message(client, message.channel, 'Yes {}, it is gay.'.format(message.author.mention())) return res = wolf.query(question) try: if len(res.pods): pod_text = [] for pod in res.pods: if pod.text: pod_text.append(pod.text) send_message(client, message.channel, '{} {}.'.format(message.author.mention(), byteify("\n".join(pod_text)[:1990]))) else: tagged_sent = pos_tag(question.replace('?', '').split()) proper_nouns = [word for word, pos in tagged_sent if pos == 'NNP'] wiki_search = " ".join(proper_nouns) if wiki_search.strip() != "": log("Looking up {}".format(wiki_search)) wiki_out = wikipedia.summary(wiki_search, sentences=3) send_message(client, message.channel, '{} {}.'.format(message.author.mention(), byteify(wiki_out))) else: send_message(client, message.channel, 'I don\'t know {}.'.format(message.author.mention())) return except Exception as e: log(format_exception(e)) send_message(client, message.channel, 'I don\'t know {}.'.format(message.author.mention())) return def do_addjoke(client, message_parts, message): try: joke = message.content[9:] if 'aa737a5846' in joke: send_message(client, message.channel, '{} you stop it, you are a pedofile, stop looking at little girls.'.format(message.author.mention())) return c = conn.cursor() c.execute("INSERT INTO jokes (joke) VALUES (?)", (joke,)) conn.commit() log("Added joke") except Exception as e: log(e.message) send_message(client, message.channel, 'Your shitty joke has been rejected {}.'.format(message.author.mention())) return send_message(client, message.channel, 'Your shitty joke has been added {}.'.format(message.author.mention())) return def do_joke(client, message_parts, message): joke = None try: c = conn.cursor() joke = c.execute("SELECT joke FROM jokes ORDER BY RANDOM() LIMIT 1;").fetchone()[0] log(joke) except Exception as e: log(e) pass if not joke: send_message(client, message.channel, 'Try adding a joke with "!addjoke <joke>" {}!'.format(message.author.mention())) else: send_message(client, message.channel, '{} {}'.format(message.author.mention(), byteify(joke))) return def do_secret(client, message_parts, message): send_message(client, message.channel, 'git gud {}! My source is here: http://git.savsoul.com/barry/discordbot\nVersion: {}'.format(message.author.mention(), VERSION)) return def do_waifu(client, message_parts, message): send_message(client, message.channel, 'http://orig13.deviantart.net/b25e/f/2014/175/3/d/no_waifu_no_laifu_by_imtheonenexttome-d7nsx3b.gif {}!'.format(message.author.mention())) return def do_hillary(client, message_parts, message): send_message(client, message.channel, ':bomb: Ohhhhhh, now you done it...:bomb:'.format(message.author.mention())) return def do_squid(client, message_parts, message): send_message(client, message.channel, 'くコ:彡 くコ:彡 くコ:彡 くコ:彡') return def do_stars(client, message_parts, message): send_message(client, message.channel, '✮═━┈ ✰═━┈ ✮═━┈ ✰═━┈ ✮═━┈ ✰═━┈ ✮═━┈ ✰═━┈ ✮═━┈ ✰═━┈ ✰═━┈┈ ✰═━┈┈') return def do_rigged(client, message_parts, message): lines = open('studylyrics.txt').read().splitlines() send_message(client, message.channel, ":musical_note: {} :musical_note:".format(random.choice(lines))) return def do_hunt_monster(client, message_parts, message): p = Pankration() hunt_response = p.hunt_monster(' '.join(message_parts)) str_out = hunt_response.message if hunt_response.result == HuntResponse.SUCCESS: monster = hunt_response.monster str_out += "\n\n**Stats:**\nFamily: {}\nLevel: {}\nMain Job: {}\nSupport Job: {}".format(monster.family_name, monster.level, Jobs().get_job_name(monster.main_job), Jobs().get_job_name(monster.support_job)) send_message(client, message.channel, str_out) return def do_list_zones(client, message_parts, message): p = Pankration() send_message(client, message.channel, ', '.join(p.list_zones())) return def start_timer(client): needs_loot = True while not quitting: if datetime.datetime.now().hour == grant_hour: needs_loot = True # This should fire anytime they need loot and it isn't 9.. this will change to 11 after testing if datetime.datetime.now().hour != grant_hour and needs_loot: for channel in client.get_all_channels(): if channel.id == '47934985176354816': do_grantcredits(client, None, None, channel) needs_loot = False time.sleep(10000) #log('Sleeping') def thread_exception_handler(method, client, message_parts, message): try: if message: globals()[method](client, message_parts, message) else: globals()[method](client) except Exception as e: log("{} - {}".format(format_exception(e), e.message)) ################# # Client Events ################# @client.event def on_socket_raw_send(payload, binary=False): check_msg_queue(client) @client.event def on_status(member): for member in client.get_all_members(): try: db_member = data.db_get_member(member.id) if not db_member: log("Creating new member: {}".format(member)) data.db_create_member(member) else: data.db_update_member(member, db_member) check_msg_queue(client) except Exception as e: log("Exception: {}".format(format_exception(e))) pass @client.event def on_message(message): #print message.content #print message.author #print client.user # we do not want the bot to reply to itself if message.author == client.user: return message_parts = message.content.split(" ") for command, method in registered_commands.iteritems(): if message_parts[0] == command: try: log("Calling {}".format(method,)) thread.start_new_thread(thread_exception_handler, (method, client, message_parts[1:], message)) except Exception as e: log("{} - {}".format(format_exception(e), e.message)) return if message.content.lower().startswith(client.user.name.lower()): log('Someone is talking to %s' % (client.user.name.lower(),)) if ' or ' in message.content: questions = message.content[len(client.user.name)+1:].replace('?', '').split(' or ') send_message(client, message.channel, '{} I choose: {}'.format(message.author.mention(), random.choice(questions).encode('utf-8', errors='ignore'))) @client.event def on_ready(): log('Logged in as') log(client.user.name) log(client.user.id) log('------') for member in client.get_all_members(): if member.id == '78767557628133376': send_message(client, member, "Bot Started") break # for channel in client.get_all_channels(): # client.send_message(member, "Bot Started {} {}".format(channel.id, channel.name)) check_msg_queue(client) thread.start_new_thread(thread_exception_handler, ('start_timer', client, None, None)) retries = 0 while retries < 1000: try: json_data = open(credentials).read() creds = json.loads(json_data) wolf = wolframalpha.Client(creds['wolframkey']) client.login(creds['username'], creds['password']) client.run() except KeyboardInterrupt: quitting = True conn.close quit() except Exception as e: retries += 1 log("Shit I crashed: {}\nRetry {}".format(e, retries)) #time.sleep(1)