import praw
import sys
import os
from time import gmtime, strftime
try:
    from flair_list import flairs
except ImportError as e:
    print("Flairs file can't be accessed\n")
    print(e)
    sys.exit()
except SyntaxError as e:
    print("There is a syntax error in the flair list\n")
    print(e)
    sys.exit()


class FlairBot:

    USER_AGENT = 'FlairBot'
    USER_NAME = ''
    PASSWD = ''
    SUBJECT = 'flair'
    TARGET_SUB = ''
    LOGGING = True

    # Class variable to hold the PRAW instance
    r = None

    # Class variable to hold the unread pms
    pms = None

    def init(self):
        if self.LOGGING:
            os.chdir(os.path.dirname(os.path.abspath(__file__)))
        self.login()

    def login(self):
        """ Log in to the account being used for the bot """
        try:
            self.r = praw.Reddit(user_agent=self.USER_AGENT)
            self.r.login(self.USER_NAME, self.PASSWD)
            self.fetch_pms()
        except:
            raise

    def fetch_pms(self):
        """ Get a listing of all unread PMs for the user account """
        self.pms = self.r.get_unread(limit=None)
        if self.pms is not None:
            self.process_pms()

    def process_pms(self):
        for pm in self.pms:
            if str(pm.subject) == self.SUBJECT:
                author = str(pm.author)  # Author of the PM
                content = str(pm.body)  # Content of the PM
                subreddit = self.r.get_subreddit(self.TARGET_SUB)
                if content in flairs:
                    # Get the flair text that corresponds with the class name
                    flair_text = str(flairs[content])
                    subreddit.set_flair(author, flair_text, content)
                    if self.LOGGING:
                        self.log(author, content, flair_text)
                pm.mark_as_read()  # Mark processed PM as read

    def log(self, author, content, flair_text):
        with open('log.txt', 'a') as logfile:
            time_now = strftime("%Y-%m-%d %H:%M:%S", gmtime())
            log_text = 'Added: ' + author + ' : ' \
                        + flair_text + ' : ' \
                        + content + ' @ ' + time_now + '\n'
            logfile.write(log_text)

FlairBot().init()