LiberTEM/LiberTEM

import sys
import msvcrt
import ctypes
from ctypes import windll, wintypes, byref
import logging
import warnings


logger = logging.getLogger(__name__)


def get_owner_name_dummy(full_path, stat):
    return "<Unknown>"


try:
    import win32security
    import pywintypes

    def get_owner_name(full_path, stat):
        try:
            s = win32security.GetFileSecurity(full_path, win32security.OWNER_SECURITY_INFORMATION)
            sid = s.GetSecurityDescriptorOwner()
            (name, domain, t) = win32security.LookupAccountSid(None, sid)
            return f"{domain}\\{name}"
        except pywintypes.error as e:
            raise OSError(e)

except ModuleNotFoundError:
    warnings.warn("Install package pywin32 to enable detecting file owners on Windows.")

    get_owner_name = get_owner_name_dummy

except ImportError as e:
    warnings.warn(
        f"Importing modules from pywin32 failed. {e} "
        "Possibly the post installation script of pywin32 has to be run. More information: "
        "https://github.com/mhammond/pywin32#installing-via-pip"
    )

    get_owner_name = get_owner_name_dummy


ENABLE_QUICK_EDIT = 0x0040
ENABLE_EXTENDED_FLAGS = 0x0080


def get_console_mode(stream=sys.stdin):
    file_handle = msvcrt.get_osfhandle(stream.fileno())
    getConsoleMode = windll.kernel32.GetConsoleMode
    getConsoleMode.argtypes, getConsoleMode.restype = ([wintypes.HANDLE, wintypes.LPDWORD],
                                                       wintypes.BOOL)
    mode = wintypes.DWORD(0)
    if getConsoleMode(file_handle, byref(mode)):
        return mode.value
    else:
        err = ctypes.get_last_error()
        raise ctypes.WinError(err)


def set_console_mode(mode, stream=sys.stdin):
    file_handle = msvcrt.get_osfhandle(stream.fileno())
    setConsoleMode = windll.kernel32.SetConsoleMode
    setConsoleMode.argtypes, setConsoleMode.restype = ([wintypes.HANDLE, wintypes.DWORD],
                                                       wintypes.BOOL)
    if setConsoleMode(file_handle, mode):
        return
    else:
        err = ctypes.get_last_error()
        raise ctypes.WinError(err)


def disable_quickedit():
    try:
        mode = get_console_mode()
        mode &= ~ENABLE_QUICK_EDIT
        mode |= ENABLE_EXTENDED_FLAGS
        set_console_mode(mode)
    except OSError:
        logger.info("Quick Edit couldn't be disabled. Console probably doesn't support Quick Edit.")


if __name__ == "__main__":
    disable_quickedit()