Difficult-Rocket/Difficult_Rocket/api/new_thread.py
2021-09-05 00:50:05 +08:00

54 lines
1.6 KiB
Python

import functools
import inspect
import threading
import time
from typing import Optional, Callable
"""
This part of code come from MCDReforged(https://github.com/Fallen_Breath/MCDReforged)
Very thanks to Fallen_Breath and other coder who helped MCDR worked better
https://www.gnu.org/licenses/gpl-3.0.en.html
"""
def new_thread(thread_name: Optional[str or Callable] = None):
"""
Use a new thread to execute the decorated function
The function return value will be set to the thread instance that executes this function
The name of the thread can be specified in parameter
"""
def wrapper(func):
@functools.wraps(func) # to preserve the origin function information
def wrap(*args, **kwargs):
thread = threading.Thread(target=func, args=args, kwargs=kwargs, name=thread_name)
thread.setDaemon(False)
thread.start()
return thread
# bring the signature of the func to the wrap function
# so inspect.getfullargspec(func) works correctly
# https://stackoverflow.com/questions/39926567/python-create-decorator-preserving-function-arguments
wrap.__signature__ = inspect.signature(func)
return wrap
# Directly use @on_new_thread without ending brackets case
if isinstance(thread_name, Callable):
this_is_a_function = thread_name
thread_name = None
return wrapper(this_is_a_function)
# Use @on_new_thread with ending brackets case
return wrapper
@new_thread()
def a(name):
print('ah')
print(name)
time.sleep(2)
print('done')
if __name__ == '__main__':
a('shenjack')