xhs-note-crawling/python/py/Lib/site-packages/pymsgbox/_native_win.py

# This module contains all the Windows-specific code to create native
# message boxes using the winapi.

# If you'd like to learn more about calling the winapi functions from
# Python, you can check out my other module "nicewin" to see nicely-documented
# examples. It is at https://github.com/asweigart/nicewin

# The documentation for the MessageBox winapi is at:
# https://docs.microsoft.com/en-us/windows/desktop/api/winuser/nf-winuser-messagebox
from __future__ import annotations
from typing import Optional, Any

import sys, ctypes
import pymsgbox

MB_OK = 0x0
MB_OKCANCEL = 0x1
MB_ABORTRETRYIGNORE = 0x2
MB_YESNOCANCEL = 0x3
MB_YESNO = 0x4
MB_RETRYCANCEL = 0x5
MB_CANCELTRYCONTINUE = 0x6

NO_ICON = 0
STOP = MB_ICONHAND = MB_ICONSTOP = MB_ICONERRPR = 0x10
QUESTION = MB_ICONQUESTION = 0x20
WARNING = MB_ICONEXCLAIMATION = 0x30
INFO = MB_ICONASTERISK = MB_ICONINFOMRAITON = 0x40

MB_DEFAULTBUTTON1 = 0x0
MB_DEFAULTBUTTON2 = 0x100
MB_DEFAULTBUTTON3 = 0x200
MB_DEFAULTBUTTON4 = 0x300

MB_SETFOREGROUND = 0x10000
MB_TOPMOST = 0x40000

IDABORT = 0x3
IDCANCEL = 0x2
IDCONTINUE = 0x11
IDIGNORE = 0x5
IDNO = 0x7
IDOK = 0x1
IDRETRY = 0x4
IDTRYAGAIN = 0x10
IDYES = 0x6

runningOnPython2 = sys.version_info[0] == 2
if runningOnPython2:
    messageBoxFunc = ctypes.windll.user32.MessageBoxA  # pyright: ignore[reportAttributeAccessIssue]
else:  # Python 3 functions.
    messageBoxFunc = ctypes.windll.user32.MessageBoxW  # pyright: ignore[reportAttributeAccessIssue]


def alert(
    text: str = "",
    title: str = "",
    button: str = pymsgbox.OK_TEXT,
    root: Optional[Any] = None,
    timeout: Optional[int] = None,
    icon: int = NO_ICON,
    _tkinter: bool = False,
) -> str:
    """Displays a simple message box with text and a single OK button. Returns the text of the button clicked on."""
    text = str(text)
    if (_tkinter) or (timeout is not None) or (button != pymsgbox.OK_TEXT):
        # Timeouts are not supported by Windows message boxes.
        # Call the original tkinter alert function, not this native one:
        return pymsgbox._alertTkinter(text, title, button, root, timeout)

    messageBoxFunc(0, text, title, MB_OK | MB_SETFOREGROUND | MB_TOPMOST | icon)
    return button


def confirm(
    text: str = "",
    title: str = "",
    buttons: tuple = (pymsgbox.OK_TEXT, pymsgbox.CANCEL_TEXT),
    root: Optional[Any] = None,
    timeout: Optional[int] = None,
    icon: int = QUESTION,
    _tkinter: bool = False,
) -> Optional[str]:
    """Displays a message box with OK and Cancel buttons. Number and text of buttons can be customized. Returns the text of the button clicked on, or None if the dialog box was closed."""
    text = str(text)
    buttonFlag = None
    if len(buttons) == 1:
        if buttons[0] == pymsgbox.OK_TEXT:
            buttonFlag = MB_OK
    elif len(buttons) == 2:
        if buttons[0] == pymsgbox.OK_TEXT and buttons[1] == pymsgbox.CANCEL_TEXT:
            buttonFlag = MB_OKCANCEL
        elif buttons[0] == pymsgbox.YES_TEXT and buttons[1] == pymsgbox.NO_TEXT:
            buttonFlag = MB_YESNO
        elif buttons[0] == pymsgbox.RETRY_TEXT and buttons[1] == pymsgbox.CANCEL_TEXT:
            buttonFlag = MB_RETRYCANCEL
    elif len(buttons) == 3:
        if (
            buttons[0] == pymsgbox.ABORT_TEXT
            and buttons[1] == pymsgbox.RETRY_TEXT
            and buttons[2] == pymsgbox.IGNORE_TEXT
        ):
            buttonFlag = MB_ABORTRETRYIGNORE
        elif (
            buttons[0] == pymsgbox.CANCEL_TEXT
            and buttons[1] == pymsgbox.TRY_AGAIN_TEXT
            and buttons[2] == pymsgbox.CONTINUE_TEXT
        ):
            buttonFlag = MB_CANCELTRYCONTINUE
        elif (
            buttons[0] == pymsgbox.YES_TEXT
            and buttons[1] == pymsgbox.NO_TEXT
            and buttons[2] == pymsgbox.CANCEL_TEXT
        ):
            buttonFlag = MB_YESNOCANCEL

    if (_tkinter) or (timeout is not None) or (buttonFlag is None):
        # Call the original tkinter confirm() function, not this native one:
        return pymsgbox._confirmTkinter(text, title, buttons, root, timeout)

    retVal = messageBoxFunc(
        0, text, title, buttonFlag | MB_SETFOREGROUND | MB_TOPMOST | icon
    )
    if retVal == IDOK or len(buttons) == 1:
        return pymsgbox.OK_TEXT
    elif retVal == IDCANCEL:
        return pymsgbox.CANCEL_TEXT
    elif retVal == IDYES:
        return pymsgbox.YES_TEXT
    elif retVal == IDNO:
        return pymsgbox.NO_TEXT
    elif retVal == IDTRYAGAIN:
        return pymsgbox.RETRY_TEXT
    elif retVal == IDRETRY:
        return pymsgbox.RETRY_TEXT
    elif retVal == IDIGNORE:
        return pymsgbox.IGNORE_TEXT
    elif retVal == IDCONTINUE:
        return pymsgbox.CONTINUE_TEXT
    elif retVal == IDABORT:
        return pymsgbox.ABORT_TEXT
    else:
        assert False, "Unexpected return value from MessageBox: %s" % (retVal)


'''
def prompt(text='', title='' , default=''):
    """Displays a message box with text input, and OK & Cancel buttons. Returns the text entered, or None if Cancel was clicked."""
    pass

def password(text='', title='', default='', mask='*'):
    """Displays a message box with text input, and OK & Cancel buttons. Typed characters appear as *. Returns the text entered, or None if Cancel was clicked."""
    pass

'''