Simplify on_stop

This commit is contained in:
Kristóf Tóth 2019-06-23 18:44:34 +02:00
parent 1155011b48
commit ed62c1d01c

View File

@ -1,5 +1,4 @@
from threading import Thread, Event from threading import Thread, Event
from typing import Callable
from .pipe_reader_thread import PipeReaderThread from .pipe_reader_thread import PipeReaderThread
from .pipe_writer_thread import PipeWriterThread from .pipe_writer_thread import PipeWriterThread
@ -13,7 +12,6 @@ class PipeIOServer(Thread):
self._in_pipe, self._out_pipe = in_pipe, out_pipe self._in_pipe, self._out_pipe = in_pipe, out_pipe
self._create_pipes(permissions) self._create_pipes(permissions)
self._stop_event = Event() self._stop_event = Event()
self._on_stop = lambda: None
self._reader_thread, self._writer_thread = None, None self._reader_thread, self._writer_thread = None, None
self._io_threads = None self._io_threads = None
@ -66,14 +64,10 @@ class PipeIOServer(Thread):
thread.stop() thread.stop()
Pipe(self.in_pipe).remove() Pipe(self.in_pipe).remove()
Pipe(self.out_pipe).remove() Pipe(self.out_pipe).remove()
self._on_stop() self.on_stop()
def _set_on_stop(self, value): def on_stop(self):
if not isinstance(value, Callable): pass
raise ValueError("Supplied object is not callable!")
self._on_stop = value
on_stop = property(fset=_set_on_stop)
def wait(self): def wait(self):
self._stop_event.wait() self._stop_event.wait()