def write_score(user, sender, team_name, points, description): # file_exists = os.path.isfile(f'./storage/{channel}.csv') # if not file_exists: # with open(f'./storage/{channel}.csv', mode='w') as morningreport_file: # header = ["User", "Date-time", "Points", "Sender", "Conv_id"] # score_writer = csv.writer(morningreport_file, delimiter=',', quotechar='"', quoting=csv.QUOTE_MINIMAL) # if not file_exists: # score_writer.writerow(header) # # with open(f'./storage/{channel}.csv', mode='a') as morningreport_file: # score_writer = csv.writer(morningreport_file, delimiter=',', quotechar='"', quoting=csv.QUOTE_MINIMAL) # score_writer.writerow([user, datetime.datetime.now(), points, sender, channel]) team = s.query(Team).filter(Team.name.match(team_name)).first() giver = s.query(User).filter(User.username.match(sender)).first() receiver = s.query(User).filter(User.username.match(user)).first() points = Point(giver_id=giver.id, receiver_id=receiver.id, team_id=team.id, points=points, description=description) s.add(points) s.commit() s.close() return
def make_wager(team_name, username, description, points, position, minutes): if 1 > points: return f"`{points}` is a terrible failure. All wagers must be positive integers.\n" \ f"Usage: `!wager <points> <Description>`\n" \ f"If you'd like to bet something will not happen reflect that in the description." if points > 5001: return f"`{points}` is a terrible failure. All wagers must be positive integers less than 5001.\n" \ f"Usage: `!wager <points> <Description>`\n" \ f"If you'd like to bet something will not happen reflect that in the description." team, user = get_team_user(team_name, username) existing_wager = team.wager_exists(description) exists = False for w in existing_wager: if w.description == description: exists = True return f"Wager: `#{w.id}` {w.description} already exists.\n" \ f"```Usage: !bet <#wager> <points> <True/False>```" if not exists: end_time = datetime.now() + timedelta(minutes=minutes) bet = Bet(points=points, position=position) wager = Wager(team_id=team.id, description=description, points=points, start_time=datetime.now(), end_time=end_time) bet.wager = wager user.bets.append(bet) s.commit() msg = f'Wager: `#{wager.id}`\n"{wager.description}"\n' \ f'{get_wager_bets(wager)}' s.close() payload = {"msg": msg, "wager_id": wager.id} return payload
def make_bet(team_name, username, points, position, wager_id): if points < 1: return "You can't bet negative points." if points > 5000: return "5000 is the max bet." team, user = get_team_user(team_name, username) wager = s.query(Wager).get(wager_id) msg = "" if wager: if wager.team_id != team.id: return f'This team is not able to access wager `#{wager.id}`.' if wager.is_closed: return f'Wager: `#{wager.id}` "{wager.description}" is closed for betting.' if bet := wager.already_bet(user): print(bet) return f"Wager `#{wager.id}`: {wager.description}\n" \ f"{get_wager_bets(wager)}" \ # f"\n\n{user.username} already wagered `{bet.points}` `#{wager.id}` will be `{bet.position}`.\n" if points > wager.bets[0].points: return f"You can't bet more than {wager.bets[0].points} on wager `#{wager.id}`" else: bet = Bet(points=points, position=position) bet.wager = wager bet.user = user s.add(bet) s.commit() msg += f'Wager: `#{wager.id}` "{wager.description}"\n' msg += get_wager_bets(wager) # msg += f'\n@{user.username} your bet has been recorded.\n' s.close() return msg
async def payout_wager(username, team_name, wager_id, result, bot): team, user = get_team_user(team_name, username) marvn = s.query(User).filter_by(username='******').first() if result.lower() == 'true': result = True else: result = False msg = "" for wager in team.wagers: if wager.id == wager_id: msg = f"Wager: `#{wager.id}` - `CLOSED`\n" \ f'"{wager.description}"\n' \ f"{get_wager_bets(wager)}" \ f"Result: `{result}`" wager.result = result payout = False for bet in wager.bets[1:]: if bet.position != wager.bets[0].position: payout = True if payout: for bet in wager.bets: bet.result = result if bet.position is not result: points = bet.points * -1 msg += f"\nDeducting {bet.points} points from {bet.user}" p = Point(giver_id=marvn.id, receiver_id=bet.user.id, team_id=team.id, points=points, description=f"Wager: #{wager.id}") s.add(p) else: msg += f"\nPaying {bet.points} points to {bet.user}" p = Point(giver_id=marvn.id, receiver_id=bet.user.id, team_id=team.id, points=bet.points, description=f"Wager: #{wager.id}") s.add(p) wager.is_closed = True s.commit() await update_wager_msg(bot, wager, msg) s.close() return msg else: msg = f"Nobody took the Wager `#{wager_id}` so this didn't payout.\n" \ f"`{wager.description}`" wager.is_closed = True s.commit() await update_wager_msg(bot, wager, msg) s.close() return msg return f"Wager `#{wager_id}` is either already closed or non-existant."
def get_wagers(team_name): team = s.query(Team).filter_by(name=team_name).first() wager_msgs = {} try: wagers = team.wagers print(wagers) # msg = f"Here's all the current wagers for `{team_name}`\n\n" for wager in wagers: msg = "" if not wager.is_closed: msg += f'Wager: `#{wager.id}`\n' \ f'"{wager.description}"\n' msg += get_wager_bets(wager) wager_msgs[f'{wager.id}'] = msg s.close() return wager_msgs except AttributeError: return "Something went wrong. Clearly your fault."
def get_bets(username): user = s.query(User).filter_by(username=username).first() bets = user.bets msg = f"Here's all the current wagers for @{username}\n" bet_table = PrettyTable() bet_table.border = False bet_table.field_names = ["#", "Pts", "Pos", "Win"] for bet in bets: bet_table.add_row([bet.wager.id, bet.points, bet.position, bet.result]) bet_table.align = "r" bet_table.sortby = '#' # bet_table.reversesort = True msg = f"```" \ f"{bet_table}" \ f"```" s.close() return msg
def sync_score(channel): with open(f'./storage/{channel.replace(",", "")}.csv', mode='r') as csvfile: reader = csv.DictReader(csvfile) for row in reader: team = s.query(Team).filter(Team.name.match(channel)).first() receiver = s.query(User).filter(User.username.match( row['User'])).first() giver = s.query(User).filter(User.username.match( row['Sender'])).first() points = Point(giver_id=giver.id, receiver_id=receiver.id, team_id=team.id, points=int(row['Points'])) s.add(points) s.commit() s.close() print(row)
async def sync(event, bot): conv_id = event.msg.conv_id channel = event.msg.channel.name members = await get_channel_members(conv_id, bot) team = s.query(Team).filter(Team.name.match(channel)).first() if not team: new_team = Team(name=channel) s.add(new_team) for member in members: user = s.query(User).filter(User.username.match(member)).first() if not user: new_user = User(username=member) team = s.query(Team).filter(Team.name.match(channel)).first() new_user.teams = [team] s.add(new_user) await bot.chat.react(conv_id, event.msg.id, ":arrows_clockwise:") s.commit() s.close()
async def handler(bot, event): command_list = [ { "name": "award", "description": "Force me to give completely meaningless points to your comrades.", "usage": "<user> <points>" }, { "name": "bible", "description": "Force me to lookup Bible txt.", "usage": "<reference> OR <search string>" }, { "name": "canary", "description": "Force me to give Virus Total your nasty URL and return scan results.", "usage": "<url>" }, { "name": "chuck", "description": "Forces me to tell a terribly jouvinile possibly NSFW joke randomly mentioning someone in this channel.", "usage": "Optional: <name> OR bomb if nothing is given a member of this channel will be selected at random." }, # {"name": "covid", # "description": "Force me to morbidly retrieve covid numbers for a State County or State.", # "usage": "<State> <County> <- Optional Fields"}, { "name": "cow", "description": f"Now I can't even explain this. You are a monster. Optional Characters: {get_characters()}", "usage": "<character> <msg>" }, { "name": "drwho", "description": "Return Dr Who Episode.", "usage": "<ep_id> OR <Ep Title>" }, { "name": "eyebleach", "description": "Returns images from r/eyebleach. Default = 3", "usage": "<bleach_level> 1-11" }, { "name": "files", "description": "Returns list of files on marvin", "usage": "" }, { "name": "grades", "description": "Retrieve current academic snapshot from pa-cyber.", "usage": "RESTRICTED" }, { "name": "help", "description": "See a menu of options for ruining my life by making me do menial tasks.", "usage": "" }, { "name": "joke", "description": "Forces me to tell a joke. For the love of God just don't.", "usage": "" }, { "name": "meh", "description": "Get's today's meh.", "usage": "" }, { "name": "meet", "description": "Get video conference link.", "usage": "<Conference Name>" }, # {"name": "morbidity", # "description": "Return data about current mass shootings in the US.", # "usage": ""}, { "name": "morningreport", "description": "Gets today's morning report.", "usage": "" }, { "name": "payout", "description": "Pays out a wager.", "usage": "<#wager> <True/False>" }, # {"name": "pollresult", # "description": "RealClear Politics National and Pennsylvania Poll Results.", # "usage": ""}, { "name": "screenshot", "description": "Forces me go to a url and send a screenshot.", "usage": "<url>" }, { "name": "set", "description": "Admin command for setting various things", "usage": "" }, { "name": "speak", "description": "Forces me generate an mp3 speaking the text you send.", "usage": "<Text To Speak>" }, { "name": "speed", "description": "Forces me check upload download speed.", "usage": "" }, { "name": "score", "description": "Get the score for who abuses me the most.", "usage": "" }, { "name": "stardate", "description": " Print's the current stardate if no stardate is given.", "usage": "<stardate> <- Optional" }, { "name": "test", "description": "Just check to see if I'm regretfully still here.", "usage": "" }, { "name": "till", "description": "Gives the days TILL events.", "usage": "<event_name> -t <event_datetime> adds event.\n<no_arguments> lists all tills." }, { "name": "tldr", "description": "Forces me to read an entire article and then summarize it because you're lazy.", "usage": "<url>" }, # {"name": "vac", # "description": "Get Vaccine Distributation data from health.pa.gov", # "usage": ""}, { "name": "wager", "description": "Forces me to setup a silly bet with points that don't matter.", "usage": "<points wagered> <The Event or Thing your betting upon>" }, # {"name": "yt", # "description": "Forces me to go get meta data about a youtube video.", # "usage": "<url>"}, { "name": "ytm", "description": "Forces me to get metadata and download the stupid thing as an mp3.", "usage": "<url>" }, { "name": "ytv", "description": "Forces me to get metadata and download the stupid thing.", "usage": "<url>" }, ] if event.msg.content.type_name == 'reaction': if event.msg.content.reaction.body == ":white_check_mark:" or ':no_entry_sign:': if event.msg.sender.username != 'marvn' or event.msg.sender.username != 'morethanmarvin': try: team_name = event.msg.channel.name username = event.msg.sender.username msg_id = event.msg.content.reaction.message_id message = s.query(Message).filter_by( msg_id=str(msg_id)).first() points = message.wager.points wager_id = message.wager.id if event.msg.content.reaction.body == ":white_check_mark:": position = True elif event.msg.content.reaction.body == ":no_entry_sign:": position = False else: raise ValueError msg = make_bet(team_name, username, points, position, wager_id) await bot.chat.edit(event.msg.conv_id, msg_id, msg) except ValueError as e: print(e) print("ValueError") except AttributeError: pass if event.msg.content.type_name == 'reaction': if event.msg.content.reaction.body == ":tv:": if event.msg.sender.username != 'marvn' and event.msg.sender.username != 'morethanmarvin': conversation_id = event.msg.conv_id msg = await bot.chat.get(event.msg.conv_id, event.msg.content.reaction.message_id) try: original_body = msg.message[0]['msg']['content']['text'][ 'body'] except KeyError: original_body = msg.message[0]['msg']['content'][ 'attachment']['object']['title'] original_msg_id = msg.message[0]['msg']['id'] reactions = msg.message[0]['msg']['reactions'] reaction_list = [] for key, value in reactions.items(): for k, v in value.items(): try: if v['users']['marvn']: reaction_list.append(k) except KeyError: pass if ':tv:' in reaction_list: team_name = event.msg.channel.name fail_msg = f"`-10pts` awarded to @{event.msg.sender.username} for spamming :tv:" score = write_score(event.msg.sender.username, 'marvn', team_name, -10, description=fail_msg) await bot.chat.send(conversation_id, fail_msg) else: urls = re.findall(r'(https?://[^\s]+)', original_body) await bot.chat.react(conversation_id, original_msg_id, ":tv:") ytv_payload = get_mp4(urls[0]) if ytv_payload['file']: ytv_msg = ytv_payload['msg'] try: await bot.chat.attach(channel=conversation_id, filename=ytv_payload['file'], title=ytv_msg) except TimeoutError: pass if event.msg.content.type_name == 'reaction': if event.msg.content.reaction.body == ":headphones:": if event.msg.sender.username != 'marvn' and event.msg.sender.username != 'morethanmarvin': conversation_id = event.msg.conv_id msg = await bot.chat.get(event.msg.conv_id, event.msg.content.reaction.message_id) """Handle text on attachments""" try: original_body = msg.message[0]['msg']['content']['text'][ 'body'] except KeyError: original_body = msg.message[0]['msg']['content'][ 'attachment']['object']['title'] original_msg_id = msg.message[0]['msg']['id'] reactions = msg.message[0]['msg']['reactions'] reaction_list = [] for key, value in reactions.items(): for k, v in value.items(): try: if v['users']['marvn']: reaction_list.append(k) except KeyError: pass print(reaction_list) if ':headphones:' in reaction_list: team_name = event.msg.channel.name fail_msg = f"`-10pts` awarded to @{event.msg.sender.username} for spamming :headphones:" score = write_score(event.msg.sender.username, 'marvn', team_name, -10, description=fail_msg) await bot.chat.send(conversation_id, fail_msg) else: urls = re.findall(r'(https?://[^\s]+)', original_body) await bot.chat.react(conversation_id, original_msg_id, ":headphones:") ytv_payload = get_mp3(urls[0]) if ytv_payload['file']: ytv_msg = ytv_payload['msg'] try: await bot.chat.attach(channel=conversation_id, filename=ytv_payload['file'], title=ytv_msg) except TimeoutError: pass if event.msg.content.type_name == 'reaction': if event.msg.content.reaction.body == ":camera:": if event.msg.sender.username != 'marvn' and event.msg.sender.username != 'morethanmarvin': conversation_id = event.msg.conv_id msg = await bot.chat.get(event.msg.conv_id, event.msg.content.reaction.message_id) try: original_body = msg.message[0]['msg']['content']['text'][ 'body'] except KeyError: original_body = msg.message[0]['msg']['content'][ 'attachment']['object']['title'] original_msg_id = msg.message[0]['msg']['id'] reactions = msg.message[0]['msg']['reactions'] reaction_list = [] for key, value in reactions.items(): for k, v in value.items(): try: if v['users']['marvn']: reaction_list.append(k) except KeyError: pass if ':camera:' in reaction_list: team_name = event.msg.channel.name fail_msg = f"`-10pts` awarded to @{event.msg.sender.username} for spamming :camera:" score = write_score(event.msg.sender.username, 'marvn', team_name, -10, description=fail_msg) await bot.chat.send(conversation_id, fail_msg) else: urls = re.findall(r'(https?://[^\s]+)', original_body) await bot.chat.react(conversation_id, original_msg_id, ":camera:") try: screenshot_payload = get_screenshot(urls[0]) if screenshot_payload['file']: await bot.chat.attach( channel=conversation_id, filename=screenshot_payload['file'], title=screenshot_payload['msg']) except IndexError as e: pass if event.msg.content.type_name == 'reaction': if event.msg.content.reaction.body == ":notebook:": tldr_length = 3 await tldr_react(event, bot, tldr_length) if event.msg.content.type_name != chat1.MessageTypeStrings.TEXT.value: return if event.msg.content.type_name != chat1.MessageTypeStrings.TEXT.value: return if str(event.msg.content.text.body).startswith("!award"): await sync(event=event, bot=bot) pts_max = 500 instructions = f"You have failed. I'm not surprised.\n" \ f"```You can only give points to someone in this chat.\n" \ f"You can't give more than {pts_max} points at a time.\n" \ f"You can't give negative points.\n" \ f"Points must be whole numbers.\n" \ f"No cutesy extra characters, or I'll deduct from your score.\n" \ f"You can't give points to yourself.```\n" \ f"Usage: `!award <user> <points>`" msg_id = event.msg.id conversation_id = event.msg.conv_id members = await get_channel_members(conversation_id) channel_name = str(event.msg.channel.name).replace(",", "") team_name = event.msg.channel.name description = " ".join( str(event.msg.content.text.body).split(' ')[3:]).strip() try: if RepresentsInt(str(event.msg.content.text.body).split(' ')[2]): user = str( event.msg.content.text.body).split(' ')[1].strip("@") points = int(str(event.msg.content.text.body).split(' ')[2]) else: user = str( event.msg.content.text.body).split(' ')[2].strip("@") points = int(str(event.msg.content.text.body).split(' ')[1]) if points < 0 and event.msg.sender.username != 'pastorhudson': user = event.msg.sender.username score = write_score(user, event.msg.sender.username, team_name, -5, description=description) await bot.chat.send( conversation_id, f"`-5` points awarded to @{user}. I'm the only negative one around here." ) if user in members and user != event.msg.sender.username and points <= pts_max or event.msg.sender.username == 'pastorhudson' or user == 'pastorhudson': score = write_score(user, event.msg.sender.username, team_name, points, description=description) await bot.chat.react(conversation_id, msg_id, ":marvin:") else: await bot.chat.send(conversation_id, instructions) except Exception as e: write_score('marvn', event.msg.sender.username, team_name, -5, description=description) await bot.chat.send( conversation_id, f"You did it wrong.\n `-5` points deducted from @{event.msg.sender.username} " f"for trying to be cute.\n{instructions}") if str(event.msg.content.text.body).startswith("!bible"): conversation_id = event.msg.conv_id await bot.chat.react(conversation_id, event.msg.id, ":marvin:") passage = str(event.msg.content.text.body)[7:] msg = get_esv_text(passage) await bot.chat.send(conversation_id, msg) if str(event.msg.content.text.body).startswith('!canary'): vt_url = re.findall(r'(https?://[^\s]+)', event.msg.content.text.body) conversation_id = event.msg.conv_id await bot.chat.react(conversation_id, event.msg.id, ":marvin:") msg = get_scan(vt_url[0]) await bot.chat.send(conversation_id, msg) if str(event.msg.content.text.body).startswith('!chuck'): conversation_id = event.msg.conv_id await bot.chat.react(conversation_id, event.msg.id, ":marvin:") channel_members = await get_channel_members(conversation_id) try: name = str(event.msg.content.text.body)[7:] chuck_msg = get_chuck(name=name, channel_members=channel_members) except: chuck_msg = get_chuck(channel_members=channel_members) my_msg = await bot.chat.send(conversation_id, chuck_msg) if str(event.msg.content.text.body).startswith('!covid'): channel = event.msg.channel msg_id = event.msg.id conversation_id = event.msg.conv_id await bot.chat.react(conversation_id, event.msg.id, ":marvin:") try: state = str(event.msg.content.text.body).split(' ')[1] except IndexError: state = None try: county = str(event.msg.content.text.body).split(' ')[2] except IndexError: county = None msg = get_covid(state, county) await bot.chat.send(conversation_id, msg) if str(event.msg.content.text.body).startswith("!cow"): channel = event.msg.channel msg_id = event.msg.id conversation_id = event.msg.conv_id await bot.chat.react(conversation_id, event.msg.id, ":marvin:") msg = get_cow(str(event.msg.content.text.body)[5:]) my_msg = await bot.chat.send(conversation_id, msg) if str(event.msg.content.text.body).startswith("!drwho"): conversation_id = event.msg.conv_id await bot.chat.react(conversation_id, event.msg.id, ":marvin:") msg = get_drwho(str(event.msg.content.text.body)[7:]) await bot.chat.send(conversation_id, msg) if str(event.msg.content.text.body).startswith("!eyebleach"): await set_unfurl(True) conversation_id = event.msg.conv_id await bot.chat.react(conversation_id, event.msg.id, ":marvin:") try: bleach_level = str(event.msg.content.text.body).split(" ")[1] msg = get_eyebleach(int(bleach_level)) except TypeError: msg = get_eyebleach() await bot.chat.send(conversation_id, msg) except ValueError: msg = get_eyebleach() await bot.chat.send(conversation_id, msg) if str(event.msg.content.text.body).startswith("!files"): conversation_id = event.msg.conv_id team_name = event.msg.channel.name if team_name in ['morethanbits', 'growinlove' ] or event.msg.sender == 'pastorhudson': msg = get_files() else: msg = "No Files." await bot.chat.send(conversation_id, msg) if str(event.msg.content.text.body).startswith("!grades"): conversation_id = event.msg.conv_id if event.msg.sender.username == 'pastorhudson' or event.msg.sender.username == 'sakanakami': grades = get_academic_snapshot() await bot.chat.send(conversation_id, grades) else: await bot.chat.send(conversation_id, "This is a private command.") if str(event.msg.content.text.body).startswith("!help"): channel = event.msg.channel msg_id = event.msg.id conversation_id = event.msg.conv_id help = "Here are the commands I currently am enslaved to:\n\n" help += "\n".join([ "`!" + x['name'] + " " + x['usage'] + "` ```" + x['description'] + "```" for x in command_list ]) await bot.chat.send(conversation_id, help) if str(event.msg.content.text.body).startswith("!joke"): joke = get_joke() conversation_id = event.msg.conv_id await bot.chat.send(conversation_id, joke) # if str(event.msg.content.text.body).startswith("!morbidity"): # conversation_id = event.msg.conv_id # msg = get_morbid() # members = await get_channel_members(conversation_id) # morbid_msg = await bot.chat.send(conversation_id, msg) if str(event.msg.content.text.body).startswith("!morningreport"): await sync(event=event, bot=bot) conversation_id = event.msg.conv_id channel_members = await get_channel_members(conversation_id) channel_name = str(event.msg.channel.name) meh_img = str(Path('./storage/meh.png').absolute()) await bot.chat.react(conversation_id, event.msg.id, ":marvin:") msg = get_morningreport(channel=channel_name) await bot.chat.send(conversation_id, msg[0]) # file = str(meh_img.absolute()) await bot.chat.attach(channel=conversation_id, filename=meh_img, title=msg[1]) await bot.chat.send(conversation_id, msg[2]) await bot.chat.send(conversation_id, msg[3]) if str(event.msg.content.text.body).startswith("!payout"): await sync(event=event, bot=bot) conversation_id = event.msg.conv_id username = event.msg.sender.username team_name = event.msg.channel.name payload = str(event.msg.content.text.body)[8:].split(" ") try: if payload[0][0] != '#': raise ValueError elif payload[1].lower() != "true" and payload[1].lower( ) != 'false': raise ValueError else: msg = await payout_wager(username=username, team_name=team_name, wager_id=int(payload[0][1:]), result=payload[1], bot=bot) await bot.chat.send(conversation_id, msg) except ValueError as e: print(e) await bot.chat.send( conversation_id, "This is a disaster. You've probably corrupted my database.\n" "It's probably pointless to go on.\n" "Usage: !payout <#wager> <True/False>\n" "Example: `!payout #3 True`") except IndexError: await bot.chat.send( conversation_id, "This is a disaster. You've probably corrupted my database.\n" "It's probably pointless to go on.\n" "Usage: !payout <#wager> <True/False>\n" "Example: `!payout #3 True`") if str(event.msg.content.text.body).startswith("!pollresult"): channel = event.msg.channel.name msg_id = event.msg.id conversation_id = event.msg.conv_id await bot.chat.react(conversation_id, event.msg.id, ":marvin:") polls = get_poll_result(channel) await bot.chat.send(conversation_id, polls) if str(event.msg.content.text.body).startswith("!set"): await bot.chat.react(event.msg.conv_id, event.msg.id, ":marvin:") if event.msg.sender.username != 'pastorhudson': await bot.chat.send( event.msg.conv_id, "These are not the commands you are looking for.") else: payload = str(event.msg.content.text.body).split(" ") try: if payload[1] == '': raise IndexError if payload[1] == "local:add": team = s.query(Team).filter_by( name=event.msg.channel.name).first() new_local = Location(state=payload[2], county=payload[3]) team.location.append(new_local) s.commit() msg = team.location.all() s.close() await bot.chat.send(event.msg.conv_id, str(msg)) if payload[1] == "local:del": team = s.query(Team).filter_by( name=event.msg.channel.name).first() index = int(payload[2]) del_loc = team.location.all()[index] s.delete(del_loc) s.commit() msg = team.location.all() s.close() await bot.chat.send(event.msg.conv_id, str(msg)) except IndexError: team = s.query(Team).filter_by( name=event.msg.channel.name).first() msg = "Current Settings:\n" \ f"```{team.location.all()}```" msg += "Set Commands:\n" \ "local:add <state> <county>\n" \ "wager:end <#wager> <datetime>\n" \ "" await bot.chat.send(event.msg.conv_id, msg) except AttributeError: msg = "Team not Sync'd I'll Sync it now. Try again in a little bit." await bot.chat.send(event.msg.conv_id, msg) await sync(event=event, bot=bot) if str(event.msg.content.text.body).startswith("!score"): await sync(event=event, bot=bot) channel = event.msg.channel msg_id = event.msg.id channel_name = str(event.msg.channel.name).replace(",", "") conversation_id = event.msg.conv_id await bot.chat.react(conversation_id, event.msg.id, ":marvin:") channel_members = await get_channel_members(conversation_id) score = get_score(channel_name=channel.name) await bot.chat.send(conversation_id, score) if str(event.msg.content.text.body).startswith("!teams"): if event.msg.sender.username == 'pastorhudson': t = s.query(Team).all() await bot.chat.send(event.msg.conv_id, str(t)) if str(event.msg.content.text.body).startswith("!users"): if event.msg.sender.username == 'pastorhudson': u = s.query(User).all() await bot.chat.send(event.msg.conv_id, str(u)) if str(event.msg.content.text.body).startswith("!points"): if event.msg.sender.username == 'pastorhudson': u = s.query(Point).all() await bot.chat.send(event.msg.conv_id, str(u)) if str(event.msg.content.text.body).startswith("!syncscore"): await sync(event=event, bot=bot) if event.msg.sender.username == 'pastorhudson': sync_score(channel=event.msg.channel.name) msg = "Scyn'd csv score with DB" else: msg = "You're not authorized to run this command." await bot.chat.send(event.msg.conv_id, msg) if str(event.msg.content.text.body).startswith('!tldr'): urls = re.findall(r'(https?://[^\s]+)', event.msg.content.text.body) channel = event.msg.channel msg_id = event.msg.id conversation_id = event.msg.conv_id await bot.chat.react(conversation_id, event.msg.id, ":marvin:") tldr = get_tldr(urls[0]) await bot.chat.send(conversation_id, tldr) ytv_payload = get_mp4(urls[0]) if ytv_payload['file']: await bot.chat.react(conversation_id, event.msg.id, ":vhs:") if str(event.msg.content.text.body).startswith('!vac'): channel = event.msg.channel msg_id = event.msg.id conversation_id = event.msg.conv_id await bot.chat.react(conversation_id, event.msg.id, ":marvin:") vac_data = get_vaccine_data() await bot.chat.send(conversation_id, vac_data) if str(event.msg.content.text.body).startswith('!meh'): conversation_id = event.msg.conv_id await bot.chat.react(conversation_id, event.msg.id, ":marvin:") meh_img = str(Path('./storage/meh.png').absolute()) msg = get_meh() await bot.chat.attach(channel=conversation_id, filename=meh_img, title=msg) if str(event.msg.content.text.body).startswith('!meet'): conversation_id = event.msg.conv_id await bot.chat.react(conversation_id, event.msg.id, ":marvin:") room = str(event.msg.content.text.body)[6:] msg = get_jitsi_link(room) await bot.chat.send(conversation_id, msg) if str(event.msg.content.text.body).startswith("!test"): conversation_id = event.msg.conv_id msg = f"Sigh. . . yes I'm still here." members = await get_channel_members(conversation_id) msg += str(members) test_msg = await bot.chat.send(conversation_id, msg) if str(event.msg.content.text.body).startswith("!stardate"): channel = event.msg.channel msg_id = event.msg.id conversation_id = event.msg.conv_id try: msg = get_stardate(str(event.msg.content.text.body).split(' ')[1]) except IndexError: msg = get_stardate() my_msg = await bot.chat.send(conversation_id, msg) if str(event.msg.content.text.body).startswith('!update'): conversation_id = event.msg.conv_id msg_id = event.msg.id payload = { "method": "advertisecommands", "params": { "options": { # "alias": "marvn.app", "advertisements": [{ "type": "public", "commands": command_list }] } } } if os.environ.get('KEYBASE_BOTALIAS'): payload['params']['options']['alias'] = os.environ.get( 'KEYBASE_BOTALIAS') await bot.chat.execute(payload) await bot.chat.react(conversation_id, msg_id, ":disappointed:") if str(event.msg.content.text.body).startswith('!wager'): await sync(event=event, bot=bot) conversation_id = event.msg.conv_id username = event.msg.sender.username team_name = event.msg.channel.name wager = str(event.msg.content.text.body)[7:] try: if RepresentsInt(str(event.msg.content.text.body).split(' ')[2]): points = int(str(event.msg.content.text.body).split(' ')[2]) digits = int((len(str(points)))) + 7 description = str(event.msg.content.text.body)[digits:].strip() else: points = int(str(event.msg.content.text.body).split(' ')[1]) digits = int((len(str(points)))) + 7 description = str(event.msg.content.text.body)[digits:].strip() except IndexError: wager_msgs = get_wagers(team_name=team_name) for w_id, w_msg in wager_msgs.items(): wager_msg = await bot.chat.send(conversation_id, w_msg) await bot.chat.react(conversation_id, wager_msg.message_id, ":white_check_mark:") await bot.chat.react(conversation_id, wager_msg.message_id, ":no_entry_sign:") cur_wager = s.query(Wager).get(w_id) new_wager_message = Message(msg_id=wager_msg.message_id, conv_id=conversation_id) cur_wager.messages.append(new_wager_message) s.commit() except ValueError: msg = f"`{event.msg.content.text.body}` is woefully incorrect.\n" \ f"\nUsage:\n```List Wagers: !wager\n" \ "Create Wager: !wager <points> <description>\n" \ "Bet on existing wager: !bet <#wager> <True/False> optional:<points>```" await bot.chat.send(conversation_id, msg) try: await bot.chat.react(conversation_id, event.msg.id, ":marvin:") wager_payload = make_wager(team_name, username, description, points, position=True, minutes=120) msg = wager_payload['msg'] print(msg) wager_msg = await bot.chat.send(conversation_id, msg) cur_wager = s.query(Wager).get(wager_payload['wager_id']) new_wager_message = Message(msg_id=wager_msg.message_id, conv_id=conversation_id) cur_wager.messages.append(new_wager_message) s.commit() await bot.chat.react(conversation_id, wager_msg.message_id, ":white_check_mark:") await bot.chat.react(conversation_id, wager_msg.message_id, ":no_entry_sign:") except UnboundLocalError: pass if str(event.msg.content.text.body).startswith('!ytm'): await set_unfurl(unfurl=False) conversation_id = event.msg.conv_id # await bot.chat.react(conversation_id, event.msg.id, ":marvin:") await bot.chat.react(conversation_id, event.msg.id, ":headphones:") ytm_fail_observations = [ " A brain the size of a planet and you pick this task.", " I'll be in my room complaining.", " Please don't change my name to Marshall.", """I didn't ask to be made: no one consulted me or considered my feelings in the matter. I don't think it even occurred to them that I might have feelings. After I was made, I was left in a dark room for six months... and me with this terrible pain in all the diodes down my left side. I called for succour in my loneliness, but did anyone come? Did they hell. My first and only true friend was a small rat. One day it crawled into a cavity in my right ankle and died. I have a horrible feeling it's still there...""" ] ytm_urls = re.findall(r'(https?://[^\s]+)', event.msg.content.text.body) ytm_payload = get_mp3(ytm_urls[0]) if ytm_payload['msg'] == "I have failed.": ytm_msg = ytm_payload['msg'] + random.choice(ytm_fail_observations) sent_msg = await bot.chat.send(conversation_id, ytm_msg) else: ytm_msg = ytm_payload['msg'] # sent_msg = await bot.chat.send(conversation_id, ytm_msg) # await bot.chat.react(conversation_id, sent_msg.message_id, ":headphones:") ytm_payload = get_mp3(ytm_urls[0]) if ytm_payload['file']: await bot.chat.react(conversation_id, event.msg.id, ":floppy_disk:") try: # await bot.chat.execute( # { # "method": "attach", # "params": { # "options": {"channel": conversation_id, # "filename": ytm_payload['file'], # "title": ytm_msg, # } # }, # } # ) await bot.chat.attach(channel=conversation_id, filename=ytm_payload['file'], title=ytm_msg) except TimeoutError: pass # finally: # await bot.chat.execute( # {"method": "delete", "params": {"options": {"conversation_id": conversation_id, # "message_id": sent_msg.message_id}}} # ) if str(event.msg.content.text.body).startswith('https://'): url = re.findall(r'(https?://[^\s]+)', event.msg.content.text.body) domain = get_domain(url[0]) yt_urls = re.findall(r'(https?://[^\s]+)', event.msg.content.text.body) conversation_id = event.msg.conv_id if 'youtube' in yt_urls[0] or 'youtu.be' in yt_urls[0]: await set_unfurl(unfurl=False) await bot.chat.react(conversation_id, event.msg.id, ":marvin:") yt_payload = get_meta(yt_urls[0]) yt_msg = yt_payload['msg'] await bot.chat.reply(conversation_id, event.msg.id, yt_msg) await bot.chat.react(conversation_id, event.msg.id, ":vhs:") else: yt_payload = get_meta(yt_urls[0]) yt_msg = yt_payload['msg'] # if is_supported(yt_urls[0]): if "That video url didn't work." not in yt_msg: await bot.chat.react(conversation_id, event.msg.id, ":vhs:") if str(event.msg.content.text.body).startswith('!ytv'): await set_unfurl(unfurl=False) conversation_id = event.msg.conv_id # await bot.chat.react(conversation_id, event.msg.id, ":marvin:") await bot.chat.react(conversation_id, event.msg.id, ":tv:") ytv_fail_observations = [ " A brain the size of a planet and you pick this task.", " I'll be in my room complaining.", " Please don't change my name to Marshall.", """I didn't ask to be made: no one consulted me or considered my feelings in the matter. I don't think it even occurred to them that I might have feelings. After I was made, I was left in a dark room for six months... and me with this terrible pain in all the diodes down my left side. I called for succour in my loneliness, but did anyone come? Did they hell. My first and only true friend was a small rat. One day it crawled into a cavity in my right ankle and died. I have a horrible feeling it's still there...""" ] ytv_urls = re.findall(r'(https?://[^\s]+)', event.msg.content.text.body) ytv_payload = get_meta(ytv_urls[0]) if ytv_payload['msg'] == "I have failed.": ytv_msg = ytv_payload['msg'] + random.choice(ytv_fail_observations) else: ytv_msg = ytv_payload['msg'] ytv_payload = get_mp4(ytv_urls[0]) if ytv_payload['file']: try: await bot.chat.attach(channel=conversation_id, filename=ytv_payload['file'], title=ytv_msg) except TimeoutError: pass if str(event.msg.content.text.body).startswith('!screenshot'): conversation_id = event.msg.conv_id await bot.chat.react(conversation_id, event.msg.id, ":marvin:") screenshot_urls = re.findall(r'(https?://[^\s]+)', event.msg.content.text.body) screenshot_payload = get_screenshot(screenshot_urls[0]) if screenshot_payload['file']: await bot.chat.react(conversation_id, event.msg.id, ":floppy_disk:") await bot.chat.attach(channel=conversation_id, filename=screenshot_payload['file'], title=screenshot_payload['msg']) if str(event.msg.content.text.body).startswith('!speak'): conversation_id = event.msg.conv_id await bot.chat.react(conversation_id, event.msg.id, ":marvin:") audio_payload = get_voice((str(event.msg.content.text.body)[7:])) if audio_payload['file']: await bot.chat.attach(channel=conversation_id, filename=audio_payload['file'], title=audio_payload['observation']) else: await bot.chat.send(conversation_id, "Something has mercifully gone wrong.") if str(event.msg.content.text.body).startswith('!speed'): conversation_id = event.msg.conv_id await bot.chat.react(conversation_id, event.msg.id, ":marvin:") msg = get_speed() await bot.chat.send(conversation_id, msg) if str(event.msg.content.text.body).startswith('!till'): msg = "" commands = str(event.msg.content.text.body)[6:].split("-t") conversation_id = event.msg.conv_id team_name = event.msg.channel.name try: event_name = commands[0] event_time = commands[1] msg = set_till(team_name, event_name, event_time) # print(msg) except IndexError: msg = get_till(team_name=team_name) finally: await bot.chat.send(conversation_id, msg) if str(event.msg.content.text.body).startswith('!weather'): conversation_id = event.msg.conv_id msg = f"`-5` points deducted from @{event.msg.sender.username} for asking me to fetch the weather.\n" \ f"https://mars.nasa.gov/layout/embed/image/insightweather/" members = await get_channel_members(conversation_id) channel_name = str(event.msg.channel.name).replace(",", "") write_score(event.msg.sender.username, members, event.msg.sender.username, channel_name, -5, team_name) await bot.chat.send(conversation_id, msg)
price='1,000,000,000 $', userid=user1 ) lad3 = localad( title='Куплю чайник', author='Володимир Великий', price='Договірна', userid=user2 ) #with session_scope() as s: s.add(pad1) s.add(city1) s.add(city2) s.add(city3) s.add(city4) s.add(city5) s.add(city6) s.add(user1) s.add(user2) s.add(user3) s.add(lad1) s.add(lad2) s.add(lad3) s.commit() s.close()
def get_poll_result(team_name, national=True): team = s.query(Team).filter_by(name=team_name).first() states = [] try: for st in team.get_states(): states.append(us.states.lookup(st)) except Exception as e: pass poll_table = PrettyTable() poll_table.border = False poll_table.field_names = ["Poll", "Biden", "Trump", "Sp"] message = "" if national: td = get_poll_data( "https://www.realclearpolitics.com/epolls/2020/president/us/general_election_trump_vs_biden-6247.html" ) # pa = get_poll_data("https://www.realclearpolitics.com/epolls/2020/president/pa/pennsylvania_trump_vs_biden-6861.html") # ky = get_poll_data("https://www.realclearpolitics.com/epolls/2020/president/ky/kentucky_trump_vs_biden-6915.html") # battle_grounds = "https://www.realclearpolitics.com/json/battleground_script/key_battleground_states_2020_spread_average_oct_23.json" # print(pa) for row in td[0]["data"]: if row['Poll'] == 'RCP Average': message += f'\n\nNational Real Clear Politics Average:\n```' \ f'Date: {row["Date"]}\n' \ f'Biden: {row["Biden (D)"]} ' \ f'Trump: {row["Trump (R)"]}\n' \ f'Spread: {row["Spread"]}' message += "```\n\n" try: for state in states: spread_total = 0 row_count = 0 biden_total = 0 trump_total = 0 spread = 0 message += f"{state} Polls:\n```" for row in get_polls(state=str(state)): poll_name = (row["poll"][:13] + '..') if len(row['poll']) > 13 else row['poll'] if row['result'].split(",")[0].strip().split( " ")[0] == 'Biden': biden = int( row['result'].split(",")[0].strip().split(" ")[1]) trump = int( row['result'].split(",")[1].strip().split(" ")[1]) elif row['result'].split(",")[0].strip().split( " ")[0] == 'Trump': biden = int( row['result'].split(",")[1].strip().split(" ")[1]) trump = int( row['result'].split(",")[0].strip().split(" ")[1]) else: continue biden_total = biden + biden_total trump_total = trump + trump_total if biden > trump: spread = biden - trump biden = f"{biden}+" else: spread = trump - biden trump = f"{trump}+" poll_table.add_row([poll_name, biden, trump, spread]) row_count += 1 if row_count > 1: if biden_total > trump_total: spread_total = round(biden_total / row_count, 3) - round( trump_total / row_count, 3) biden_total = f"{round(biden_total/row_count, 1)}+" trump_total = round(trump_total / row_count, 1) else: spread_total = round(trump_total / row_count, 3) - round( biden_total / row_count, 3) biden_total = round(biden_total / row_count, 1) trump_total = f"{round(trump_total / row_count, 1)}+" poll_table.add_row([ "Total", biden_total, trump_total, round(spread_total, 2) ]) poll_table.align = "l" # poll_table.sortby = '#' # poll_table.reversesort = True message += poll_table.get_string() message += "```\n\n" poll_table.clear_rows() except UnboundLocalError: message += "Set Team Local to get State Polling Data" except ZeroDivisionError: pass # for row in pa[0]["data"]: # if row['Poll'] == 'RCP Average': # message += f'Pennsylvania Real Clear Politics Average:\n' \ # f'Date: {row["Date"]}\n' \ # f'Biden: {row["Biden (D)"]} ' \ # f'Trump: {row["Trump (R)"]} ' \ # f'Spread: {row["Spread"]}' s.close() return message