isotopic_logging.threadsafe_iter

Complexity

Total Complexity

5

Size/Duplication

Total Lines	19
Duplicated Lines	0 %

Test Coverage

Coverage

81.82%

4 Methods

Rating	Name	Duplication	Size	Complexity
A	__next__()	0	3	2
A	__iter__()	0	2	1
A	__init__()	0	3	1
A	next()	0	2	1

# -*- coding: utf-8 -*-

import threading


class threadsafe_iter(object):
    """
    Takes an iterator/generator and makes it thread-safe by serializing call to
    the `next` method of given iterator/generator.
    """

    def __init__(self, original):
        self.original = original
        self.lock = threading.Lock()

    def __iter__(self):
        return self

    def __next__(self):
        with self.lock:
            return next(self.original)

    def next(self):
        return self.__next__()