Source code for pyadi.trace

import time
import sys



class Stop(BaseException):
    pass




def decorator(tracecalls=False, verbose=False, verboseargs=False, **opts):

    data = {}

    if tracecalls:
        hist = []


    def inner(done, key, f):

        adfun = done(key)

        def run(*args, **kw):
            nonlocal hist

            if verbose:
                print(f'call to {f.__name__}() starts {time.time()} s')
            if verboseargs:
                print(f'call to {f.__name__}({args})')

            if tracecalls:
                data['cur'] = f.__qualname__
                hist += [f.__qualname__]

            if data and 'cmd' in data:
                while True:
                    cmd = data.get('cmd', '')
                    print(f'Command set: {cmd}')
                    if cmd == 'pause':
                        print(f'condition wait on {data["condition"]}')
                        sys.stdout.flush()
                        data['condition'].wait()
                        print('been notified')
                        sys.stdout.flush()
                    elif cmd == 'sleep':
                        time.sleep(data['timeout'])
                    else:
                        if cmd == 'stop':
                            raise Stop(f'Stopping because stop set: {data["msg"]}.')
                        elif cmd == 'raise':
                            raise data['exception']
                        else:
                            if cmd == 'call':
                                data['function'](data, hist, f, args, kw)
                        break

            res = adfun(*args, **kw)
            if verbose:
                print(f'call to {f.__name__}() ends {time.time()} s')
            if verboseargs:
                print(f'call to {f.__name__}(,,,) = {res}')

            return res

        return run

    def get(*args, **kw):
        nonlocal hist
        for name in args:
            del data[name]
        if kw.get('get') == 'hist':
            return hist
        elif kw.get('clear') == 'hist':
            hist = []
        elif kw.get('get'):
            return data[kw.get('get')]
        else:
            data.update(kw)
        return data

    return inner, get


# (c) 2023 AI & IT UG
# Author: Johannes Willkomm jwillkomm@ai-and-it.de