Method for creating a Mock side_effect that only raises an exception for n calls

get_exception_side_effect.py

def get_exception_side_effect(num_of_exceptions, exception_class=Exception):
    def side_effect(*args, **kwargs):
        side_effect.calls += 1
        if side_effect.calls > num_of_exceptions:
            return Mock()
        else:
            raise exception_class
    side_effect.calls = 0
    
    return side_effect