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

rpcserver: remove __main__ function

This file is not intended to be used standalone.

Changeset 0d855d77d8d1…

Parent 7eb3166afc88…

by Steve Borho

Changes to one file · Browse files at 0d855d77d8d1 Showing diff from parent 7eb3166afc88 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
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 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"  PIPEBUFSIZE = 4096      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):   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"    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'   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() - -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] -