test262/tools/lint/lib/frontmatter.py

import re
import yaml

class Result(dict):
    def __init__(self, meta, events):
        self.parsing_events = events
        super(Result, self).__init__(**meta)

class MyLoader(yaml.SafeLoader):
    events = None

    def __init__(self, *args, **kwargs):
        MyLoader.events = []
        super(MyLoader, self).__init__(*args, **kwargs)

    def get_event(self):
        event = super(MyLoader, self).get_event()
        MyLoader.events.append(event)
        return event

def parse(src):
    '''Parse the YAML-formatted metadata found in a given string of source
    code. Tolerate missing or invalid metadata; those conditions are handled by
    a dedicated "Check" instance.'''

    match = re.search(r'/\*---(.*)---\*/', src, re.DOTALL)
    if not match:
        return None

    try:
        return Result(yaml.load(match.group(1), MyLoader), MyLoader.events)
    except (yaml.scanner.ScannerError, yaml.parser.ParserError):
        return None