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

taskbar: hook up taskbarui dialog

Changeset b179a5cc1e71…

Parent 9c45f864ab67…

by Steve Borho

Changes to 2 files · Browse files at b179a5cc1e71 Showing diff from parent 9c45f864ab67 Diff from another changeset...

Change 1 of 4 Show Entire File taskbar.py Stacked
 
39
40
41
 
42
43
44
 
91
92
93
94
95
96
 
 
 
 
 
97
98
99
 
102
103
104
105
 
 
 
 
106
107
108
 
130
131
132
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
133
134
135
 
39
40
41
42
43
44
45
 
92
93
94
 
 
 
95
96
97
98
99
100
101
102
 
105
106
107
 
108
109
110
111
112
113
114
 
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
@@ -39,6 +39,7 @@
  0, 0, win32con.CW_USEDEFAULT, win32con.CW_USEDEFAULT, \   0, 0, hinst, None)   UpdateWindow(self.hwnd) + self.guithread = None   self._DoCreateIcons()     def _DoCreateIcons(self): @@ -91,9 +92,11 @@
  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) + if not self.guithread or not self.guithread.isAlive(): + self.launchgui() + else: + msg = "TortoiseHG options dialog already running" + win32ui.MessageBox(msg, 'TortoiseHG options...', win32con.MB_OK)   elif id == 1025:   self.exit_application()   else: @@ -102,7 +105,10 @@
  def exit_application(self):   if self.stop_pipe_server():   DestroyWindow(self.hwnd) - print "Goodbye" + if self.guithread and self.guithread.isAlive(): + import gobject + gobject.idle_add(self.dialog.destroy) + print "Goodbye"     def stop_pipe_server(self):   print "Stopping pipe server..." @@ -130,6 +136,21 @@
  else:   return True   + def launchgui(self): + def launch(): + import gtk + from hggtk import taskbarui, hgtk + dlg = taskbarui.TaskBarUI(rpcserver.logq) + dlg.show_all() + dlg.connect('destroy', gtk.main_quit) + self.dialog = dlg + gtk.gdk.threads_init() + gtk.gdk.threads_enter() + gtk.main() + gtk.gdk.threads_leave() + + self.guithread = thread2.Thread(target=launch) + self.guithread.start()     def start_pipe_server(self):   def servepipe():
Change 1 of 3 Show Changes Only win32/​rpcserver.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
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
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
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
 import os  import win32api  import win32con    from win32com.shell import shell, shellcon  import _winreg    from mercurial import ui    from thgutil import paths, shlib    import sys  import time  import Queue  import threading    import win32event  import win32pipe  import win32file  import pywintypes  import winerror      PIPENAME = r"\\.\pipe\TortoiseHgRpcServer-bc0c27107423-"  PIPENAME += win32api.GetUserName()    PIPEBUFSIZE = 4096   +logq = Queue.Queue(0) +def logmsg(msg): + if logq.qsize() < 100: + ts = '[%s] ' % time.strftime('%c') + logq.put(ts + msg)    def update_batch(batch):   '''updates thgstatus for all paths in batch'''   roots = set()   notifypaths = set()   for path in batch:   r = paths.find_root(path)   if r is None:   for n in os.listdir(path):   r = paths.find_root(os.path.join(path, n))   if (r is not None):   roots.add(r)   notifypaths.add(r)   else:   roots.add(r);   notifypaths.add(path)   if roots:   _ui = ui.ui();   for r in sorted(roots): + logmsg('Updating ' + r)   shlib.update_thgstatus(_ui, r, wait=False) - print "updated repo %s" % r   if notifypaths:   time.sleep(2)   shlib.shell_notify(list(notifypaths)) - print "shell notified" + logmsg('Shell notified')    requests = Queue.Queue(0)  _abort_request = r';:;:; Quit ;:;:;' # any invalid path would do    class Updater(threading.Thread):   def run(self):   n = 0   while True:   batch = []   r = requests.get()   if r == _abort_request: - print 'Updater thread quiting' + logmsg('Updater thread quiting')   return   print "got request %s (first in batch)" % r   batch.append(r)   print "wait a bit for additional requests..."   time.sleep(0.2)   try:   while True:   r = requests.get_nowait()   print "got request %s" % r   batch.append(r)   except Queue.Empty:   pass   n += 1   msg = "--- processing batch %i with %i requests ---"   print msg % (n, len(batch))   update_batch(batch)    Updater().start()      class PipeServer:   def __init__(self):   # Create an event which we will use to wait on.   # The "service stop" request will set this event.   self.hWaitStop = win32event.CreateEvent(None, 0, 0, None)     # We need to use overlapped IO for this, so we dont block when   # waiting for a client to connect. This is the only effective way   # to handle either a client connection, or a service stop request.   self.overlapped = pywintypes.OVERLAPPED()     # And create an event to be used in the OVERLAPPED object.   self.overlapped.hEvent = win32event.CreateEvent(None,0,0,None)     def SvcStop(self):   print 'PipeServer thread quiting'   win32event.SetEvent(self.hWaitStop)   requests.put(_abort_request)     def SvcDoRun(self):   # We create our named pipe.   pipeName = PIPENAME   openMode = win32pipe.PIPE_ACCESS_DUPLEX | win32file.FILE_FLAG_OVERLAPPED   pipeMode = win32pipe.PIPE_TYPE_MESSAGE     # When running as a service, we must use special security for the pipe   sa = pywintypes.SECURITY_ATTRIBUTES()   # Say we do have a DACL, and it is empty   # (ie, allow full access!)   sa.SetSecurityDescriptorDacl ( 1, None, 0 )     pipeHandle = win32pipe.CreateNamedPipe(pipeName,   openMode,   pipeMode,   win32pipe.PIPE_UNLIMITED_INSTANCES,   0, 0, 6000, # default buffers, and 6 second timeout.   sa)     # Loop accepting and processing connections   while True:   try:   hr = win32pipe.ConnectNamedPipe(pipeHandle, self.overlapped)   except pywintypes.error, inst:   print "Error connecting pipe: ", inst   pipeHandle.Close()   break     if hr==winerror.ERROR_PIPE_CONNECTED:   # Client is fast, and already connected - signal event   win32event.SetEvent(self.overlapped.hEvent)   # Wait for either a connection, or a service stop request.   timeout = win32event.INFINITE   waitHandles = self.hWaitStop, self.overlapped.hEvent   rc = win32event.WaitForMultipleObjects(waitHandles, 0, timeout)   if rc==win32event.WAIT_OBJECT_0:   # Stop event   return   else:   # read pipe and process request   try:   hr, data = win32file.ReadFile(pipeHandle, PIPEBUFSIZE)   if not data:   raise SystemExit # signal by dispatch terminate   win32pipe.DisconnectNamedPipe(pipeHandle)   except win32file.error:   # Client disconnected without sending data   # or before reading the response.   # Thats OK - just get the next connection   continue     try:   requests.put(data)   except SystemExit:   raise SystemExit # interrupted by thread2.terminate()   except:   import traceback   print "WARNING: something went wrong in requests.put"   print traceback.format_exc()   status = "ERROR"   # Clean up when we exit   self.SvcStop()