49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
# Copyright (c) 2019 OpenStack Foundation.
|
|
# All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
from threading import Thread
|
|
|
|
|
|
class PropagatingThread(Thread):
|
|
def run(self):
|
|
self.exc = None
|
|
try:
|
|
if hasattr(self, '_Thread__target'):
|
|
# Thread uses name mangling prior to Python 3.
|
|
self.ret = self._Thread__target(*self._Thread__args,
|
|
**self._Thread__kwargs)
|
|
else:
|
|
self.ret = self._target(*self._args, **self._kwargs)
|
|
except BaseException as e:
|
|
self.exc = e
|
|
|
|
def join(self):
|
|
super(PropagatingThread, self).join()
|
|
if self.exc:
|
|
raise self.exc
|
|
return self.ret
|
|
|
|
|
|
def run_async(func):
|
|
from functools import wraps
|
|
|
|
@wraps(func)
|
|
def async_func(*args, **kwargs):
|
|
thread = PropagatingThread(target=func, args=args, kwargs=kwargs)
|
|
thread.start()
|
|
return thread
|
|
|
|
return async_func
|