SickGear/lib/sg_futures/base.py
2023-02-09 13:41:15 +00:00

22 lines
608 B
Python

import re
import sys
import threading
if 2 == sys.version_info[0]:
# noinspection PyProtectedMember
from .futures.thread import _WorkItem
else:
# noinspection PyCompatibility,PyProtectedMember
from concurrent.futures.thread import _WorkItem
class GenericWorkItem(_WorkItem):
number_regex = re.compile(r'(_\d+)$')
def _set_thread_name(self):
try:
ct = threading.current_thread()
ct.name = '%s^WEB%s' % (self.args[0].__class__.__name__.upper(), self.number_regex.search(ct.name).group(1))
except (BaseException, Exception):
pass