File: flattenrec.py

package info (click to toggle)
pypy 7.3.3%2Bdfsg-2
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 113,660 kB
  • sloc: python: 1,419,707; ansic: 64,313; cpp: 3,290; sh: 2,763; makefile: 540; xml: 256; asm: 213; lisp: 45; awk: 4
file content (31 lines) | stat: -rw-r--r-- 811 bytes parent folder | download | duplicates (8)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
"""
A general way to flatten deeply recursive algorithms by delaying some
parts until later.
"""

try:
    from thread import _local as TlsClass
except ImportError:
    class TlsClass(object):
        pass


class FlattenRecursion(TlsClass):

    def __init__(self):
        self.later = None

    def __call__(self, func, *args, **kwds):
        """Call func(*args, **kwds), either now, or, if we're recursing,
        then the call will be done later by the first level.
        """
        if self.later is not None:
            self.later.append((func, args, kwds))
        else:
            self.later = lst = []
            try:
                func(*args, **kwds)
                for func, args, kwds in lst:
                    func(*args, **kwds)
            finally:
                self.later = None