# Exception handling

import atexit

import sys

import strings

import term

import util

def raise_(e): raise e

def add_msg(e, msg): e.args = (strings.ensure_newl(str(e))+msg,)

def repl_msg(e, **repls): e.args = (str(e) % repls,)

class ExceptionWithCause(Exception):

    def __init__(self, msg, cause=None):

        Exception.__init__(self, msg)

        if cause != None: add_msg(self, 'cause: '+str_(cause))

def str_(e, with_name=True):

    msg = strings.ensure_newl(str(e))

    if with_name: msg = util.type_name(e)+': '+msg

    return msg

def print_ex(e, **format):

    emph = format.pop('emph', True)

    msg = str_(e, **format)

    if emph:

        first_line, nl, rest = msg.partition('\n')

        msg = term.error(first_line)+nl+rest

    sys.stderr.write(msg)

class ExTracker:

    def __init__(self):

        self.e_ct = 0

        def at_exit():

            if self.e_ct > 0:

                raise SystemExit('Encountered '+str(self.e_ct)+' error(s)')

        atexit.register(at_exit)

    def track(self, e, **format):

        self.e_ct += 1

        print_ex(e, **format)