theycallmemac/dcurooms

import sys
from mechanicalsoup import StatefulBrowser
from bs4 import BeautifulSoup
import utils
if sys.version_info[0] < 3:
    from cookielib import LWPCookieJar
else:
    from http.cookiejar import LWPCookieJar

__author__ = "theycallmemac"
__version__ = '2.0.0'
__copyright__ = 'Copyright (c) 2018 theycallmemac'
__license__ = 'GPL-3.0'


class LookUp(object):
    week = ""
    day = ""
    hour = ""

    def __init__(self, week, day, hour):
        self.week = week
        self.day = day
        self.hour = hour

    def check_time(self, times):
        if self.hour not in times:
            print("Outside scheduled timetables. Please try again at 08:00.")
            sys.exit()
        for k, v in times.items():
            if k == self.hour:
                self.hour = v
                break
            else:
                pass

    def building_option(self, lst):
        for room in lst:
            status = LookUp.build_timetable(self, room)
            if len(status) <= 9:
                print("\033[1;92m{0}\033[00m".format(room) +
                      ": " + "\033[1;97m{0}\033[00m".format(status))
            else:
                print("\033[1;91m{0}\033[00m".format(room) +
                      ": " + "\033[1;90m{0}\033[00m".format(status))

    def room_option(self, room):
        status = LookUp.build_timetable(self, room)
        print(room + ": " + status)

    def build_timetable(self, room):
        browser = StatefulBrowser()
        cookie_jar = LWPCookieJar()
        browser.set_cookiejar(cookie_jar)
        browser.user_agent = [
            ("User-Agent",
             """Mozilla/5.0 (Windows NT 10.0; Win64; x64)
             AppleWebKit/537.36 (KHTML, like Gecko)
             Chrome/58.0.3029.110
             Safari/537.36""")]
        url = "http://www101.dcu.ie/timetables/feed.php?room=GLA." + \
            room + "&week1=" + self.week + \
            "&hour=" + str(self.hour) + \
            "&day=" + self.day + "&template=location"
        browser.open(url, verify=False)
        result = utils.check_room(url)
        return result