platformio/platformio-core
1
0
Fork 1
mirror of https://github.com/platformio/platformio-core.git synced 2025-07-29 17:47:14 +02:00
Code Issues Packages Projects Releases Wiki Activity

Ensure buffer is created for thread stream

Browse Source
This commit is contained in:
Ivan Kravets
2019-07-02 22:35:35 +03:00
parent 6c47c7506e
commit fd9ca0cd15
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
platformio/commands/home/rpc/handlers/piocore.py
View File

@ -38,14 +38,16 @@ class MultiThreadingStdStream(object):
def __getattr__(self, name):
thread_id = thread_get_ident()
if thread_id not in self._buffers:
raise AttributeError(name)
self._ensure_thread_buffer(thread_id)
return getattr(self._buffers[thread_id], name)
def _ensure_thread_buffer(self, thread_id):
if thread_id not in self._buffers:
self._buffers[thread_id] = BytesIO() if PY2 else StringIO()
def write(self, value):
thread_id = thread_get_ident()
if thread_id not in self._buffers:
self._buffers[thread_id] = BytesIO() if PY2 else StringIO()
self._ensure_thread_buffer(thread_id)
return self._buffers[thread_id].write(
value.decode() if is_bytes(value) else value)