1#@!#!123s

: / opt / saltstack / salt / lib / python3.10 / site-packages / salt / utils /

Filename : winapi.py

back

import logging
import threading

try:
    import pythoncom

    HAS_LIBS = True
except ImportError:
    HAS_LIBS = False

log = logging.getLogger(__name__)


def __virtual__():
    """
    Only load if required libraries exist
    """
    if not HAS_LIBS:
        return False
    else:
        return True


class Com:
    def __init__(self):
        self.need_com_init = not self._is_main_thread()

    def _is_main_thread(self):
        return threading.current_thread().name == "MainThread"

    def __enter__(self):
        if self.need_com_init:
            log.debug("Initializing COM library")
            pythoncom.CoInitialize()

    def __exit__(self, exc_type, exc_value, traceback):
        if self.need_com_init:
            log.debug("Uninitializing COM library")
            pythoncom.CoUninitialize()