Adastral Hub Menu

adastral.py

#!/usr/bin/env python

"""
Find the Adastral Park Hub Menu
"""

from __future__ import print_function

import re
import webbrowser
from urllib2 import urlopen
from BeautifulSoup import BeautifulSoup as bs


def ask_open_in_browser(link):
    """Ask whether the given link should be opened in a browser"""
    keep_asking = True
    while keep_asking:
        answer = raw_input("[?] Open in browser? [yn] ")
        if answer != '':
            answer = answer[0].lower()
            if answer in ['y', 'n']:
                keep_asking = False
            if answer == 'y':
                webbrowser.open(link)
            elif answer == 'n':
                pass
            else:
                print("[!] Invalid option, try again!")


def main():
    """Download their main website, find and display the PDF link"""

    soup = bs(urlopen('https://atadastral.co.uk/'))

    for pdf in soup.findAll(href=re.compile(r'menu.*\.pdf$')):
        link = pdf['href']
        print("[+] Here is your menu good sir.")
        print("[+] {0}".format(link))
        ask_open_in_browser(link)


if __name__ == '__main__':
    main()