commentparser.py

__author__ = 'Almenon'

from re import compile, search
import logging
logging.basicConfig(level="DEBUG") # set level to INFO to get rid of debug messages

regex = compile(r"(?:\/u\/the_episode_bot[ ,:]*(\S+(?:\ \S+\b)*)[ ,]+)?"  # bot's username followed by title (optional)
                r"(s|e)[a-s]*\ *?(\d+)[\: ,_\[\]\-x]*"  # season or episode followed by optional seperator
                r"(?:s|e)[a-s]*\ *?(\d+)")   # season/episode
# example match:  /u/the_episode_bot pokemon S01E06

class ParseError(Exception):
    pass


def parse(request):
    logging.info("request: " + request)

    # PARSE REQUEST
    request = request.lower()
    show_season_episode = search(regex,request)
    if show_season_episode is None:
        raise ParseError("request does not contain correct format")
    if show_season_episode.group(1) is not None: show = show_season_episode.group(1)
    else: show = None
    if show_season_episode.group(2) is 'e': # format is episode season
        episode = show_season_episode.group(3)
        season = show_season_episode.group(4)
    else:  # format is season episode
        season = show_season_episode.group(3)
        episode = show_season_episode.group(4)

    return show, season, episode

# testing

# try:
#     answer = parse("1 :: Beers in S01E05")
#     print(answer)
# except ParseError as e:
#     print(e)