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

rpcserver: adjust print message

Changeset f70076f5bd35…

Parent 7b847237fb07…

by Adrian Buehlmann

Changes to one file · Browse files at f70076f5bd35 Showing diff from parent 7b847237fb07 Diff from another changeset...

Change 1 of 1 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
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
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
172
173
174
175
176
177
178
179
180
181
182
183
184
 import os  import win32api  import win32con    from win32com.shell import shell, shellcon  import _winreg    from mercurial import hg, cmdutil, util  from mercurial import repo as _repo    from thgutil import paths, shlib    import sys  import time  import Queue  import threading    import win32serviceutil  import win32service  import win32event  import win32pipe  import win32file  import pywintypes  import winerror      PIPENAME = "\\\\.\\pipe\\TortoiseHgRpcServer-bc0c27107423"  PIPEBUFSIZE = 4096      # FIXME: quick workaround traceback caused by missing "closed"  # attribute in win32trace.  from mercurial import ui  def write_err(self, *args):   for a in args:   sys.stderr.write(str(a))  ui.ui.write_err = write_err      def update_batch(batch):   '''updates thgstatus for all paths in batch'''   roots = []   notifypaths = []   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) and (r not in roots):   roots.append(r)   notifypaths.append(r)   elif r not in roots:   roots.append(r);   notifypaths.append(path)   if roots:   _ui = ui.ui();   for r in sorted(roots):   shlib.update_thgstatus(_ui, r, wait=False)   print "updated repo %s" % r   if notifypaths:   time.sleep(2)   shlib.shell_notify(notifypaths)   print "shell notified"    requests = Queue.Queue(0)    class Updater(threading.Thread):   def run(self):   n = 0   while True:   batch = []   r = requests.get()   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 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 1:   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   break   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 get_hg_state()" + print "WARNING: something went wrong in requests.put"   print traceback.format_exc()   status = "ERROR"    if __name__ == '__main__':   import sys   if '--server' in sys.argv:   svc = PipeServer()   svc.SvcDoRun()   elif '--client' in sys.argv:   for x in sys.argv[1:]:   if x.startswith('-'):   continue   path = os.path.abspath(x)   try:   status = win32pipe.CallNamedPipe(PIPENAME, path, PIPEBUFSIZE, 0)   except pywintypes.error, inst:   print "can't access named pipe '%s'" % PIPENAME   sys.exit()   print "%s = %s" % (path, status)   else:   print "usage:\n%s [--server|--client]" % sys.argv[0]