flairbot

import praw
import asyncio
import traceback

from datetime import timedelta
from time import time
from collections import OrderedDict


username = "UmbrellaCorpBot"
password = "ResidentEvil"
subreddit_name = "automodtest1243"


'''
`sleep time` : time (in seconds) the bot sleeps before performing a new check
`time_until_message` : time (in seconds) a person has to add flair before a initial message is sent
`time_until_remove` : time (in seconds) after a message is sent that a person has to add flair before the post is removed and they have to resubmit it
`h_time_intil_remove` : Human Readable Version of time_until_remove
`post_grab_limit` : how many new posts to check at a time.
`add_flair_subject_line`, `add_flair_message` : Initial Message that tells a user that they need to flair their post
`remove_post_subject_line`, `remove_post_message`: Second message telling them to resubmit their post since they have not flaired in time
`no_flair` : Posts that still have a grace period to add a flair`
'''

sleep_time = 10
time_until_message = 300
time_until_remove = 1800
h_time_until_remove = str(timedelta(seconds=time_until_remove))
post_grab_limit = 30
post_memory_limit = 300
posts_to_forget = post_memory_limit - post_grab_limit

add_flair_subject_line = "You have not tagged your post."

add_flair_message = ("Hello {author}, **[your submission]({url}) in /r/Rainbow6 does not have any flair** and will soon be removed.\n\n"
                     "Please add flair to your post. "
                     "If you do not add flair within **" + h_time_until_remove + "**, you will have to resubmit your post. "
                     "Don't know how to flair your post? Click [here](http://imgur.com/a/m3FI3) to view this helpful guide on how to flair your post. "
                     "If you are using the mobile version of the site, feel free to reply to this message and I'll flair it for you!")

remove_post_subject_line = "You have not tagged your post within the allotted amount of time."
remove_post_message = "Hello {author}, **[your submission]({url}) in /r/Rainbow6 has been removed for it still does not have any flair.** It will remain removed but feel free to resubmit your post and remember to flair it once it is posted."

no_flair = OrderedDict()
user_agent = ("/r/Rainbow6 Flair Bot")
session = praw.Reddit(user_agent=username)
session.login(username=username, password=password, disable_warning=True)
subreddit = session.get_subreddit(subreddit_name)


@asyncio.coroutine
def get_subreddit_settings(name):
    raise NotImplementedError("TODO: Subreddit settings")


@asyncio.coroutine
def refresh_sesison():
    '''Re-logs in every n seconds'''
    while True:
        try:
            yield from asyncio.sleep(300)
            session.login(username=UmbrellaCorpBot, password=ResidentEvil, disable_warning=True)
            print("Session refreshed")
        except Exception as e:
            print(traceback.format_exc())
            print("{0}: {1}".format(type(e).__name__, str(e)))

    yield from refresh_sesison()


@asyncio.coroutine
def inbox_stuff():
    # For lack of a better name
    '''Looks to see if users have replied to the bot's message with a selected flair
    Refreshes every n seconds
    '''
    while True:
        try:
            for message in session.get_unread():
                if message.parent_id:
                    if message.parent_id[3:] in no_flair:
                        flaired = False
                        post = session.get_submission(submission_id=no_flair[message.parent_id[3:]])
                        choices = post.get_flair_choices()['choices']
                        for ch in choices:
                            if message.body == ch['flair_text']:
                                new_flair = ch['flair_text']
                                post.set_flair(new_flair)
                                flaired = True
                                break
                        if flaired:
                            message.reply("Set Flair: **{}**".format(new_flair))
                    message.mark_as_read()

        except Exception as e:
            print(traceback.format_exc())
            print("{0}: {1}".format(type(e).__name__, str(e)))

        yield from asyncio.sleep(sleep_time)

    await inbox_stuff()


@asyncio.coroutine
def main():
    '''
    Checks to see if a post has a flair, sends the user a message after
    `time_until_message seconds`, and removes it if there is no flair after
    `time_until_remove` seonds. Approves post if a flair is added. Refreshes every n seconds.
    '''
    while True:
        # Checks to see if storing too much messages
        if len(no_flair) >= post_memory_limit:
            i = 0
            while i < posts_to_forget:
                no_flair.popitem(0)
                i += 1

        try:
            for submission in subreddit.get_new(limit=post_grab_limit):
                # If message has no flair
                if (submission.link_flair_text is None):
                    if((time() - submission.created_utc) > time_until_message) and submission.id not in no_flair.values():
                        final_add_flair_message = add_flair_message.format(post_url=submission.short_link)
                        print("Sent Message to : {}".format(submission.author))
                        session.send_message(submission.author, add_flair_subject_line, final_add_flair_message)
                        for msg in session.get_sent():
                            if msg.body == final_add_flair_message:
                                no_flair[msg.id] = submission.id
                                continue

                    if((time() - submission.created_utc) > time_until_remove):
                        final_remove_post_message = remove_post_message.format(post_url=submission.short_link)
                        session.send_message(submission.author, remove_post_subject_line, final_remove_post_message)
                        print("Removed {0.short_link} of {0.author}'s".format(submission))
                        for k in list(no_flair.keys()):
                            if no_flair[k] == submission.id:
                                no_flair.pop(k)
                        submission.remove()
                        continue
                        # Keeps track of how many posts the bot removed
                        f = open('NumberRemoved','a')
                        f.write('1\n')
                        f.close()
                #
                if submission.id in no_flair.values() and submission.link_flair_text:
                    submission.approve()
                    print("Approved {0.short_link} of {0.author}'s".format(submission))
                    for k in list(no_flair.keys()):
                        if no_flair[k] == submission.id:
                            no_flair.pop(k)
                    continue
        except Exception as e:
            print(traceback.format_exc())
            print("{0}: {1}".format(type(e).__name__, str(e)))

        yield from asyncio.sleep(sleep_time)

    yield from main()

if __name__ == "__main__":
    # Puts main func into a loop and runs forever
    loop = asyncio.get_event_loop()

    print("Registering session refresh\n")
    asyncio.ensure_future(refresh_sesison())

    print("Registering Mod Invites\n")
    asyncio.ensure_future(inbox_stuff())

    userAgent = "/r/Rainbow6 Flair Bot"
    clientId = '***********'
    clientSecret = '********************'
    r = praw.Reddit(user_agent=userAgent, client_id=clientId, client_secret=clientSecret)

    print("Registering Main\n")
    asyncio.ensure_future(main())

    print("\nStarting...\n")
    loop.run_forever()

    loop.close()