Mercurial and Git clients can push and pull from this alias URL to interact with this repository. You can change to which repository an alias points by going to the Aliases link on the project page.
# Creates a task-bar icon. Run from Python.exe to see the# messages printed.importgcimportosimportsysimporttimeimportthreadingimportQueuefromwin32apiimport*fromwin32guiimport*importwin32pipeimportwin32conimportwin32eventimportwin32fileimportwinerrorimportpywintypesfrommercurialimportdemandimportdemandimport.ignore.append('win32com.shell')demandimport.enable()frommercurialimportuifromthgutilimportthread2,paths,shlibifhasattr(sys,"frozen"):sys.stdout.closed=Truesys.stderr.closed=TrueAPP_TITLE="TortoiseHg RPC server"SHOWLOG_CMD = 1023
EXIT_CMD = 1025
-def SetIcon(hwnd, name):
+def SetIcon(hwnd, name, add=False):
# Try and find a custom icon
+ print "SetIcon(%s)" % name hinst = GetModuleHandle(None)
from thgutil.paths import get_tortoise_icon
- iconPathName = get_tortoise_icon("hgB.ico")
+ iconPathName = get_tortoise_icon(name)
if iconPathName and 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 = (hwnd, 0, flags, win32con.WM_USER+20, hicon, APP_TITLE)
+ action = NIM_MODIFY+ if add:+ action = NIM_ADD try:
- Shell_NotifyIcon(NIM_ADD, nid)
+ Shell_NotifyIcon(action, 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.classMainWindow: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_WINDOWwc.lpfnWndProc=message_map# could also specify a wndproc.classAtom=RegisterClass(wc)# Create the Window.style=win32con.WS_OVERLAPPED|win32con.WS_SYSMENUself.hwnd=CreateWindow(classAtom,APP_TITLE,style, \
0,0,win32con.CW_USEDEFAULT,win32con.CW_USEDEFAULT, \
0,0,hinst,None)UpdateWindow(self.hwnd)self.guithread=None self._DoCreateIcons()
def _DoCreateIcons(self):
- SetIcon(self.hwnd, "hg.ico")
+ SetIcon(self.hwnd, "hg.ico", add=True)
# start namepipe server for hg status
self.start_pipe_server()
defOnRestart(self,hwnd,msg,wparam,lparam):self._DoCreateIcons()defOnDestroy(self,hwnd,msg,wparam,lparam):nid=(self.hwnd,0)Shell_NotifyIcon(NIM_DELETE,nid)PostQuitMessage(0)# Terminate the app.defOnTaskbarNotify(self,hwnd,msg,wparam,lparam):iflparam==win32con.WM_RBUTTONUPorlparam==win32con.WM_LBUTTONUP:menu=CreatePopupMenu()AppendMenu(menu,win32con.MF_STRING,SHOWLOG_CMD,'Options...')AppendMenu(menu,win32con.MF_SEPARATOR,0,'')AppendMenu(menu,win32con.MF_STRING,EXIT_CMD,'Exit')pos=GetCursorPos()# See http://msdn.microsoft.com/library/default.asp?url=/library/en-us/winui/menus_0hdi.aspSetForegroundWindow(self.hwnd)TrackPopupMenu(menu,win32con.TPM_LEFTALIGN,pos[0],pos[1],0,self.hwnd,None)PostMessage(self.hwnd,win32con.WM_NULL,0,0)return1defOnCommand(self,hwnd,msg,wparam,lparam):id=LOWORD(wparam)ifid==SHOWLOG_CMD:ifnotself.guithreadornotself.guithread.isAlive():self.launchgui()else:print"TortoiseHG options dialog already running"elifid==EXIT_CMD:self.exit_application()else:print"Unknown command -",iddefexit_application(self):ifself.stop_pipe_server():DestroyWindow(self.hwnd)ifself.guithreadandself.guithread.isAlive():importgobjectgobject.idle_add(self.dialog.destroy)print"Goodbye"defstop_pipe_server(self):print"Stopping pipe server..."ifnotself.pipethread.isAlive():returnTrue# Try the nice way firstself.svc.SvcStop()max_try=10cnt=1whilecnt<=max_tryandself.pipethread.isAlive():print"testing pipe [try %d] ..."%cnttry:self.pipethread.terminate()win32pipe.CallNamedPipe(PIPENAME,'',PIPEBUFSIZE,0)except:passcnt+=1ifself.pipethread.isAlive():print"WARNING: unable to stop server after %d trys."%max_tryreturnFalseelse:returnTruedeflaunchgui(self):deflaunch():importgtkfromhggtkimporttaskbarui,hgtkdlg=taskbarui.TaskBarUI(logger.getqueue(),requests)dlg.show_all()dlg.connect('destroy',gtk.main_quit)self.dialog=dlggtk.gdk.threads_init()gtk.gdk.threads_enter()gtk.main()gtk.gdk.threads_leave()logger.reset()self.guithread=thread2.Thread(target=launch)self.guithread.start() def start_pipe_server(self):
def servepipe():
- self.svc = PipeServer()
+ self.svc = PipeServer(self.hwnd)
self.svc.SvcDoRun()
self.pipethread = thread2.Thread(target=servepipe)
self.pipethread.start()PIPENAME=r"\\.\pipe\TortoiseHgRpcServer-bc0c27107423-"PIPENAME+=GetUserName()PIPEBUFSIZE=4096classLogger():def__init__(self):self.q=Nonedefgetqueue(self):self.q=Queue.Queue()returnself.qdefreset(self):self.q=Nonedefmsg(self,msg):ts='[%s] '%time.strftime('%c')ifself.q:self.q.put(ts+msg)print'L'+ts+msgelse:printts+msglogger=Logger()defgetrepos(batch):roots=set()notifypaths=set()forpathinbatch:r=paths.find_root(path)ifrisNone:forninos.listdir(path):r=paths.find_root(os.path.join(path,n))if(risnotNone):roots.add(r)notifypaths.add(r)else:roots.add(r);notifypaths.add(path)returnroots,notifypathsdefupdate_batch(batch):'''updates thgstatus for all paths in batch'''roots,notifypaths=getrepos(batch)ifroots:_ui=ui.ui();failedroots=set()forrinsorted(roots):try:shlib.update_thgstatus(_ui,r,wait=False)shlib.shell_notify([r])logger.msg('Updated '+r)exceptIOError:print"IOError on updating %s (check permissions)"%rlogger.msg('Failed updating %s (check permissions)'%r)failedroots.add(r)notifypaths-=failedrootsifnotifypaths:time.sleep(2)shlib.shell_notify(list(notifypaths))logger.msg('Shell notified')requests = Queue.Queue(0)
-def update(args):
+def update(args, hwnd):
batch = []
r = args[0]
print "got update request %s (first in batch)" % r
batch.append(r)
print "wait a bit for additional requests..."
+ SetIcon(hwnd, "hgB.ico") time.sleep(0.2)
deferred_requests = []
try:
whileTrue:req=requests.get_nowait()s=req.split('|')cmd,args=s[0],s[1:]ifcmd=='update':print"got update request %s"%reqbatch.append(args[0])else:deferred_requests.append(req)exceptQueue.Empty:passforreqindeferred_requests:requests.put(req) msg = "processing batch with %i update requests"
print msg % len(batch)
update_batch(batch)
+ SetIcon(hwnd, "hg.ico")def remove(args):
path = args[0]
logger.msg('Removing '+path)roots,notifypaths=getrepos([path])ifroots:forrinsorted(roots):try:os.remove(os.path.join(r,'.hg','thgstatus'))exceptOSError:pass if notifypaths:
shlib.shell_notify(list(notifypaths))
-def dispatch(req, cmd, args):
+def dispatch(req, cmd, args, hwnd):
print "dispatch(%s)" % req
if cmd == 'update':
- update(args)
+ update(args, hwnd)
elif cmd == 'remove':
remove(args)
else:
logger.msg("Error: unknown request '%s'" % req)
class Updater(threading.Thread):
+ def __init__(self, hwnd):+ threading.Thread.__init__(self)+ self.hwnd = hwnd+ def run(self):
while True:
req = requests.get()
s=req.split('|')cmd,args=s[0],s[1:] if cmd == 'terminate':
logger.msg('Updater thread terminating')
return
- dispatch(req, cmd, args)
+ dispatch(req, cmd, args, self.hwnd)
gc.collect()
-Updater().start()
+class PipeServer:+ def __init__(self, hwnd):+ self.updater = Updater(hwnd)
+ self.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)defSvcStop(self):print'PipeServer thread terminating'win32event.SetEvent(self.hWaitStop)requests.put('terminate')defSvcDoRun(self):# We create our named pipe.pipeName=PIPENAMEopenMode=win32pipe.PIPE_ACCESS_DUPLEX|win32file.FILE_FLAG_OVERLAPPEDpipeMode=win32pipe.PIPE_TYPE_MESSAGE# When running as a service, we must use special security for the pipesa=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 connectionswhileTrue:try:hr=win32pipe.ConnectNamedPipe(pipeHandle,self.overlapped)exceptpywintypes.error,inst:print"Error connecting pipe: ",instpipeHandle.Close()breakifhr==winerror.ERROR_PIPE_CONNECTED:# Client is fast, and already connected - signal eventwin32event.SetEvent(self.overlapped.hEvent)# Wait for either a connection, or a service stop request.timeout=win32event.INFINITEwaitHandles=self.hWaitStop,self.overlapped.hEventrc=win32event.WaitForMultipleObjects(waitHandles,0,timeout)ifrc==win32event.WAIT_OBJECT_0:# Stop eventreturnelse:# read pipe and process requesttry:hr,data=win32file.ReadFile(pipeHandle,PIPEBUFSIZE)ifnotdata:raiseSystemExit# signal by dispatch terminatewin32pipe.DisconnectNamedPipe(pipeHandle)exceptwin32file.error:# Client disconnected without sending data# or before reading the response.# Thats OK - just get the next connectioncontinuetry:requests.put(data)exceptSystemExit:raiseSystemExit# interrupted by thread2.terminate()except:importtracebackprint"WARNING: something went wrong in requests.put"printtraceback.format_exc()status="ERROR"# Clean up when we exitself.SvcStop()defmain():w=MainWindow()PumpMessages()if__name__=='__main__':main()
Attach a Trello Card
Add a tag
Your session has expired
You are no longer logged in. Please log in and try your request again.
Filter RSS Feed
This RSS feed URL allows you to see the contents of your current filter using any feed reader.
This link includes a special authentication token. If you share the URL with anyone else, they can see this RSS feed's activity. You can disable these tokens when needed.
Your current filter is unsaved; changing it won't affect this RSS feed.