PyAudioCodingTools/taskbar_utils.py at main · Crysisjim/PyAudioCodingTools · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
# taskbar_utils.py
import sys
import os
import ctypes
from ctypes import wintypes

# --- DÉFINITIONS CONSTANTES WINDOWS ---
CLSID_TaskbarList = "{56FDF344-FD6D-11d0-958A-006097C9A090}"
IID_ITaskbarList3 = "{EA1AFB91-9E28-4B86-90E9-9E9F8A5EEFAF}"

TBPF_NOPROGRESS = 0
TBPF_INDETERMINATE = 1
TBPF_NORMAL = 2
TBPF_ERROR = 4
TBPF_PAUSED = 8

class GUID(ctypes.Structure):
    _fields_ = [
        ("Data1", ctypes.c_ulong),
        ("Data2", ctypes.c_ushort),
        ("Data3", ctypes.c_ushort),
        ("Data4", ctypes.c_ubyte * 8)
    ]

class TaskbarController:
    def __init__(self, root):
        self.root = root
        self.taskbar_interface = None
        self.hwnd = 0
        self.ptr = None

        if os.name == 'nt':
            try:
                # 1. Initialiser COM
                ctypes.windll.ole32.CoInitialize(0)

                # 2. Créer l'objet TaskbarList
                clsid = GUID()
                iid = GUID()
                ctypes.windll.ole32.CLSIDFromString(ctypes.c_wchar_p(CLSID_TaskbarList), ctypes.byref(clsid))
                ctypes.windll.ole32.CLSIDFromString(ctypes.c_wchar_p(IID_ITaskbarList3), ctypes.byref(iid))

                self.ptr = ctypes.c_void_p()
                ctypes.windll.ole32.CoCreateInstance(
                    ctypes.byref(clsid),
                    0,
                    1, # CLSCTX_INPROC_SERVER
                    ctypes.byref(iid),
                    ctypes.byref(self.ptr)
                )

                # 3. Récupérer le HWND (Handle de fenêtre)
                try:
                    self.hwnd = ctypes.windll.user32.GetParent(self.root.winfo_id())
                    if not self.hwnd:
                        self.hwnd = self.root.winfo_id()
                except Exception as e:
                    print(f"Avertissement: Impossible de récupérer le HWND: {e}")
                    self.hwnd = 0

            except Exception as e:
                print(f"Erreur init Taskbar: {e}")
                self.ptr = None

    def _call_method(self, method_index, *args):
        """Appelle une méthode COM par son index dans la VTable"""
        if not self.ptr or not self.hwnd:
            return
        try:
            # Récupération de la VTable (table des méthodes virtuelles)
            vtable = ctypes.cast(self.ptr, ctypes.POINTER(ctypes.c_void_p)).contents
            func_ptr = ctypes.cast(vtable.value + (method_index * ctypes.sizeof(ctypes.c_void_p)), ctypes.POINTER(ctypes.c_void_p)).contents

            # Définition de la signature de la fonction (WINFUNCTYPE pour stdcall)
            arg_types = [ctypes.c_void_p, ctypes.c_void_p] + [ctypes.c_uint64 for _ in args]

            functype = ctypes.WINFUNCTYPE(ctypes.c_long, *arg_types)
            func = functype(func_ptr.value)

            func(self.ptr, ctypes.c_void_p(self.hwnd), *args)
        except Exception as e:
            print(f"Erreur appel COM méthode {method_index}: {e}")

    def set_progress(self, current, total):
        # SetProgressValue est la méthode index 9 dans ITaskbarList3
        self._call_method(9, int(current), int(total))

    def set_state_normal(self):
        # SetProgressState est la méthode index 10
        self._call_method(10, TBPF_NORMAL)

    def set_state_paused(self):
        self._call_method(10, TBPF_PAUSED)

    def set_state_error(self):
        self._call_method(10, TBPF_ERROR)

    def set_state_off(self):
        self._call_method(10, TBPF_NOPROGRESS)