mrmp/hypertext.py

import html.parser
import requests

import msys


class Parser(html.parser.HTMLParser):
    def __init__(self):
        self.links = []
        super().__init__()

    def handle_starttag(self, tag, attributes):
        if tag == 'a':
            self.links.extend(
                [v for k, v in attributes if k == 'href'])


class HyperText:
    def __init__(self, location):
        self.location = location
        self.load()

    def load(self):
        hypertext = requests.get(self.location).content.decode(msys.CHARSET)
        parser = Parser()
        parser.feed(hypertext)
        self.links = parser.links