SignalHandler

Complexity

Total Complexity

6

Size/Duplication

Total Lines	25
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	block_signals()	0	9	2
A	__init__()	0	5	1
A	run_callback()	0	3	1
A	on_signal()	0	4	2

#!/usr/bin/env python

import signal
from contextlib import contextmanager

class SignalHandler:
    def __init__(self, callback, what=signal.SIGUSR2):
        self.callback = callback
        self.received = False
        self.working  = False
        signal.signal(what, self.on_signal)

    def on_signal(self, signal, frame):
        self.received = True
        if not self.working:
            self.run_callback()

    def run_callback(self):
        self.received = False
        self.callback()

    @contextmanager
    def block_signals(self):
        self.working = True
        try:
            yield
        finally:
            self.working = False
            if self.received:
                self.run_callback()