SickGear/lib/sg_futures/base.py

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