Kiln » TortoiseHg » TortoiseHg
Clone URL:  
Pushed to one repository · View In Graph Contained in 0.8, 0.8.1, and 0.8.2

move win32/taskbar.py to .. and fix imports

Changeset 683ef5cad6af…

Parent 002bf6b4b1d9…

by Adrian Buehlmann

Changes to 4 files · Browse files at 683ef5cad6af Showing diff from parent 002bf6b4b1d9 Diff from another changeset...

Change 1 of 1 Show Entire File taskbar.py Stacked
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
@@ -0,0 +1,139 @@
+# Creates a task-bar icon. Run from Python.exe to see the +# messages printed. + +from win32 import rpcserver +from thgutil import thread2 +from win32api import * +from win32gui import * +import win32ui +import win32pipe +import win32con +import pywintypes +import sys, os + +APP_TITLE = "TortoiseHg RPC server" + +class MainWindow: + def __init__(self): + msg_TaskbarRestart = RegisterWindowMessage("TaskbarCreated"); + message_map = { + msg_TaskbarRestart: self.OnRestart, + win32con.WM_DESTROY: self.OnDestroy, + win32con.WM_COMMAND: self.OnCommand, + win32con.WM_USER+20 : self.OnTaskbarNotify, + } + # Register the Window class. + wc = WNDCLASS() + hinst = wc.hInstance = GetModuleHandle(None) + wc.lpszClassName = "THgRpcServer" + wc.style = win32con.CS_VREDRAW | win32con.CS_HREDRAW; + wc.hCursor = LoadCursor( 0, win32con.IDC_ARROW ) + wc.hbrBackground = win32con.COLOR_WINDOW + wc.lpfnWndProc = message_map # could also specify a wndproc. + classAtom = RegisterClass(wc) + # Create the Window. + style = win32con.WS_OVERLAPPED | win32con.WS_SYSMENU + self.hwnd = CreateWindow( classAtom, APP_TITLE, style, \ + 0, 0, win32con.CW_USEDEFAULT, win32con.CW_USEDEFAULT, \ + 0, 0, hinst, None) + UpdateWindow(self.hwnd) + self._DoCreateIcons() + + def _DoCreateIcons(self): + # Try and find a custom icon + hinst = GetModuleHandle(None) + from thgutil.paths import get_tortoise_icon + iconPathName = get_tortoise_icon("hg.ico") + if os.path.isfile(iconPathName): + icon_flags = win32con.LR_LOADFROMFILE | win32con.LR_DEFAULTSIZE + hicon = LoadImage(hinst, iconPathName, win32con.IMAGE_ICON, 0, 0, icon_flags) + else: + print "Can't find a Python icon file - using default" + hicon = LoadIcon(0, win32con.IDI_APPLICATION) + + flags = NIF_ICON | NIF_MESSAGE | NIF_TIP + nid = (self.hwnd, 0, flags, win32con.WM_USER+20, hicon, APP_TITLE) + try: + Shell_NotifyIcon(NIM_ADD, nid) + except error: + # This is common when windows is starting, and this code is hit + # before the taskbar has been created. + print "Failed to add the taskbar icon - is explorer running?" + # but keep running anyway - when explorer starts, we get the + # TaskbarCreated message. + + # start namepipe server for hg status + self.start_pipe_server() + + def OnRestart(self, hwnd, msg, wparam, lparam): + self._DoCreateIcons() + + def OnDestroy(self, hwnd, msg, wparam, lparam): + nid = (self.hwnd, 0) + Shell_NotifyIcon(NIM_DELETE, nid) + PostQuitMessage(0) # Terminate the app. + + def OnTaskbarNotify(self, hwnd, msg, wparam, lparam): + if lparam==win32con.WM_RBUTTONUP or lparam==win32con.WM_LBUTTONUP: + menu = CreatePopupMenu() + AppendMenu(menu, win32con.MF_STRING, 1023, 'Options...') + AppendMenu(menu, win32con.MF_SEPARATOR, 0, '') + AppendMenu(menu, win32con.MF_STRING, 1025, 'Exit' ) + pos = GetCursorPos() + # See http://msdn.microsoft.com/library/default.asp?url=/library/en-us/winui/menus_0hdi.asp + SetForegroundWindow(self.hwnd) + TrackPopupMenu(menu, win32con.TPM_LEFTALIGN, pos[0], pos[1], 0, self.hwnd, None) + PostMessage(self.hwnd, win32con.WM_NULL, 0, 0) + return 1 + + def OnCommand(self, hwnd, msg, wparam, lparam): + id = LOWORD(wparam) + if id == 1023: + # place holder for options dialog + msg = "TortoiseHG options dialog in construction" + win32ui.MessageBox(msg, 'TortoiseHG options...', win32con.MB_OK) + elif id == 1025: + self.exit_application() + else: + print "Unknown command -", id + + def exit_application(self): + print "stopping pipe server..." + if self.stop_pipe_server(): + DestroyWindow(self.hwnd) + print "\n\nGoodbye" + + def stop_pipe_server(self): + max_try = 10 + cnt = 1 + while cnt <= max_try and self.pipethread.isAlive(): + print "testing pipe [try %d] ..." % cnt + try: + self.pipethread.terminate() + win32pipe.CallNamedPipe(rpcserver.PIPENAME, '', + rpcserver.PIPEBUFSIZE, 0) + except: + pass + cnt += 1 + + if self.pipethread.isAlive(): + print "WARNING: unable to stop server after %d trys." % max_try + return False + else: + return True + + + def start_pipe_server(self): + def servepipe(): + self.svc = rpcserver.PipeServer() + self.svc.SvcDoRun() + + self.pipethread = thread2.Thread(target=servepipe) + self.pipethread.start() + +def main(): + w=MainWindow() + PumpMessages() + +if __name__=='__main__': + main()
Change 1 of 1 Show Entire File win32/​__init__.py Stacked
 
 
 
1
@@ -0,0 +1,1 @@
+#placeholder
Change 1 of 3 Show Entire File win32/​rpcserver.py Stacked
 
5
6
7
8
 
9
10
11
 
100
101
102
103
 
104
105
106
 
119
120
121
122
 
123
124
125
 
5
6
7
 
8
9
10
11
 
100
101
102
 
103
104
105
106
 
119
120
121
 
122
123
124
125
@@ -5,7 +5,7 @@
 import _winreg  from mercurial import hg, cmdutil, util  from mercurial import repo as _repo -import thgutil +from thgutil import paths  import sys  import win32serviceutil  import win32service @@ -100,7 +100,7 @@
  else:   print "find new root"   cache_pdir = pdir - cache_root = root = thgutil.find_root(pdir) + cache_root = root = paths.find_root(pdir)   print "_get_state: root = ", root   if root is None:   print "_get_state: not in repo" @@ -119,7 +119,7 @@
    print "%s: global overlayicons = " % path, global_opts   print "%s: repo overlayicons = " % path, repo_opts - is_netdrive = thgutil.netdrive_status(path) is not None + is_netdrive = paths.netdrive_status(path) is not None   if (is_netdrive and 'localdisks' in global_opts) \   or 'False' in repo_opts:   print "%s: overlayicons disabled" % path
Change 1 of 1 Show Entire File win32/​taskbar.py Stacked
 
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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
@@ -1,139 +0,0 @@
-# Creates a task-bar icon. Run from Python.exe to see the -# messages printed. - -import rpcserver -import thread2 -from win32api import * -from win32gui import * -import win32ui -import win32pipe -import win32con -import pywintypes -import sys, os - -APP_TITLE = "TortoiseHg RPC server" - -class MainWindow: - def __init__(self): - msg_TaskbarRestart = RegisterWindowMessage("TaskbarCreated"); - message_map = { - msg_TaskbarRestart: self.OnRestart, - win32con.WM_DESTROY: self.OnDestroy, - win32con.WM_COMMAND: self.OnCommand, - win32con.WM_USER+20 : self.OnTaskbarNotify, - } - # Register the Window class. - wc = WNDCLASS() - hinst = wc.hInstance = GetModuleHandle(None) - wc.lpszClassName = "THgRpcServer" - wc.style = win32con.CS_VREDRAW | win32con.CS_HREDRAW; - wc.hCursor = LoadCursor( 0, win32con.IDC_ARROW ) - wc.hbrBackground = win32con.COLOR_WINDOW - wc.lpfnWndProc = message_map # could also specify a wndproc. - classAtom = RegisterClass(wc) - # Create the Window. - style = win32con.WS_OVERLAPPED | win32con.WS_SYSMENU - self.hwnd = CreateWindow( classAtom, APP_TITLE, style, \ - 0, 0, win32con.CW_USEDEFAULT, win32con.CW_USEDEFAULT, \ - 0, 0, hinst, None) - UpdateWindow(self.hwnd) - self._DoCreateIcons() - - def _DoCreateIcons(self): - # Try and find a custom icon - hinst = GetModuleHandle(None) - from thgutil import get_tortoise_icon - iconPathName = get_tortoise_icon("hg.ico") - if os.path.isfile(iconPathName): - icon_flags = win32con.LR_LOADFROMFILE | win32con.LR_DEFAULTSIZE - hicon = LoadImage(hinst, iconPathName, win32con.IMAGE_ICON, 0, 0, icon_flags) - else: - print "Can't find a Python icon file - using default" - hicon = LoadIcon(0, win32con.IDI_APPLICATION) - - flags = NIF_ICON | NIF_MESSAGE | NIF_TIP - nid = (self.hwnd, 0, flags, win32con.WM_USER+20, hicon, APP_TITLE) - try: - Shell_NotifyIcon(NIM_ADD, nid) - except error: - # This is common when windows is starting, and this code is hit - # before the taskbar has been created. - print "Failed to add the taskbar icon - is explorer running?" - # but keep running anyway - when explorer starts, we get the - # TaskbarCreated message. - - # start namepipe server for hg status - self.start_pipe_server() - - def OnRestart(self, hwnd, msg, wparam, lparam): - self._DoCreateIcons() - - def OnDestroy(self, hwnd, msg, wparam, lparam): - nid = (self.hwnd, 0) - Shell_NotifyIcon(NIM_DELETE, nid) - PostQuitMessage(0) # Terminate the app. - - def OnTaskbarNotify(self, hwnd, msg, wparam, lparam): - if lparam==win32con.WM_RBUTTONUP or lparam==win32con.WM_LBUTTONUP: - menu = CreatePopupMenu() - AppendMenu(menu, win32con.MF_STRING, 1023, 'Options...') - AppendMenu(menu, win32con.MF_SEPARATOR, 0, '') - AppendMenu(menu, win32con.MF_STRING, 1025, 'Exit' ) - pos = GetCursorPos() - # See http://msdn.microsoft.com/library/default.asp?url=/library/en-us/winui/menus_0hdi.asp - SetForegroundWindow(self.hwnd) - TrackPopupMenu(menu, win32con.TPM_LEFTALIGN, pos[0], pos[1], 0, self.hwnd, None) - PostMessage(self.hwnd, win32con.WM_NULL, 0, 0) - return 1 - - def OnCommand(self, hwnd, msg, wparam, lparam): - id = LOWORD(wparam) - if id == 1023: - # place holder for options dialog - msg = "TortoiseHG options dialog in construction" - win32ui.MessageBox(msg, 'TortoiseHG options...', win32con.MB_OK) - elif id == 1025: - self.exit_application() - else: - print "Unknown command -", id - - def exit_application(self): - print "stopping pipe server..." - if self.stop_pipe_server(): - DestroyWindow(self.hwnd) - print "\n\nGoodbye" - - def stop_pipe_server(self): - max_try = 10 - cnt = 1 - while cnt <= max_try and self.pipethread.isAlive(): - print "testing pipe [try %d] ..." % cnt - try: - self.pipethread.terminate() - win32pipe.CallNamedPipe(rpcserver.PIPENAME, '', - rpcserver.PIPEBUFSIZE, 0) - except: - pass - cnt += 1 - - if self.pipethread.isAlive(): - print "WARNING: unable to stop server after %d trys." % max_try - return False - else: - return True - - - def start_pipe_server(self): - def servepipe(): - self.svc = rpcserver.PipeServer() - self.svc.SvcDoRun() - - self.pipethread = thread2.Thread(target=servepipe) - self.pipethread.start() - -def main(): - w=MainWindow() - PumpMessages() - -if __name__=='__main__': - main()