view tests/heredoctest.py @ 44003:cb2e2b095dc9

rust-core: updated copyright notice

The intention here is to put an email address that actually
works (I did have full personal copyrights with my
former affiliation).

It is also an opportunity to acknowledge that I've not been
the only one to work on this file.

Differential Revision: https://phab.mercurial-scm.org/D7781

from __future__ import absolute_import, print_function

import sys


def flush():
    sys.stdout.flush()
    sys.stderr.flush()


globalvars = {}
lines = sys.stdin.readlines()
while lines:
    l = lines.pop(0)
    if l.startswith('SALT'):
        print(l[:-1])
    elif l.startswith('>>> '):
        snippet = l[4:]
        while lines and lines[0].startswith('... '):
            l = lines.pop(0)
            snippet += l[4:]
        c = compile(snippet, '<heredoc>', 'single')
        try:
            flush()
            exec(c, globalvars)
            flush()
        except Exception as inst:
            flush()
            print(repr(inst))