Beispiel #1
0
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
Beispiel #2
0
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
Beispiel #3
0
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
Beispiel #4
0
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."
Beispiel #5
0
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."
Beispiel #6
0
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
Beispiel #7
0
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)
Beispiel #8
0
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)
Beispiel #10
0
    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