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*importwin32pipeimportwin32conimportwin32eventimportwin32fileimportwinerrorimportpywintypesimportwin32securityfrommercurialimportdemandimportdemandimport.ignore.append('win32com.shell')demandimport.enable()frommercurialimportuifromthgutilimportthread2,paths,shlibifhasattr(sys,"frozen"):# Insert PATH to binary installer gtk directoryfromthgutilimportpathsgtkpath=os.path.join(paths.bin_path,'gtk')os.environ['PATH']=os.pathsep.join([gtkpath,os.environ['PATH']])# Give stdout/stderr closed attributes to prevent ui.py errorssys.stdout.closed=Truesys.stderr.closed=TrueAPP_TITLE="TortoiseHg RPC server"SHOWLOG_CMD=1023EXIT_CMD=1025defSetIcon(hwnd,name,add=False):# Try and find a custom iconprint"SetIcon(%s)"%namehinst=GetModuleHandle(None)fromthgutil.pathsimportget_tortoise_iconiconPathName=get_tortoise_icon(name)ificonPathNameandos.path.isfile(iconPathName):icon_flags=win32con.LR_LOADFROMFILE|win32con.LR_DEFAULTSIZEhicon=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_TIPnid=(hwnd,0,flags,win32con.WM_USER+20,hicon,APP_TITLE)action=NIM_MODIFYifadd:action=NIM_ADDtry:Shell_NotifyIcon(action,nid)excepterror:# 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=Noneself._DoCreateIcons()def_DoCreateIcons(self):SetIcon(self.hwnd,"hg.ico",add=True)# start namepipe server for hg statusself.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()defstart_pipe_server(self):defservepipe():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)except(IOError,OSError):print"IOError or OSError 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)defget_config():hgighlight_taskbaricon=Truetry:from_winregimportHKEY_CURRENT_USER,OpenKey,QueryValueExhkey=OpenKey(HKEY_CURRENT_USER,r'Software\TortoiseHg')t=('1','True')try:hgighlight_taskbaricon=QueryValueEx(hkey,'HighlightTaskbarIcon')[0]intexceptEnvironmentError:passexcept(ImportError,WindowsError):passreturnhgighlight_taskbaricondefupdate(args,hwnd):batch=[]r=args[0]print"got update request %s (first in batch)"%rbatch.append(r)print"wait a bit for additional requests..."highlight=get_config()ifhighlight: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"printmsg%len(batch)update_batch(batch)ifhighlight:SetIcon(hwnd,"hg.ico")defremove(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:passifnotifypaths:shlib.shell_notify(list(notifypaths))defdispatch(req,cmd,args,hwnd):print"dispatch(%s)"%reqifcmd=='update':update(args,hwnd)elifcmd=='remove':remove(args)else:logger.msg("Error: unknown request '%s'"%req)classUpdater(threading.Thread):def__init__(self,hwnd):threading.Thread.__init__(self)self.hwnd=hwnddefrun(self):whileTrue:req=requests.get()s=req.split('|')cmd,args=s[0],s[1:]ifcmd=='terminate':logger.msg('Updater thread terminating')returndispatch(req,cmd,args,self.hwnd)gc.collect()classPipeServer:def__init__(self,hwnd):self.updater=Updater(hwnd)self.updater.start()# 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 exit
self.SvcStop()
-MUTEXNAME = 'thgtaskbar'
+INSTALLMUTEXNAME = 'thgtaskbar'
+RUNMUTEXNAME = 'thgtaskbar-' + GetUserName()def main():
sa = win32security.SECURITY_ATTRIBUTES()
- sa.SetSecurityDescriptorDacl(1, None, 0) # allow full access
-mutex1 = win32event.CreateMutex(sa, 1, MUTEXNAME)
+ sa.SetSecurityDescriptorDacl(1, None, 0) # allow full access
+runmutex = win32event.CreateMutex(sa, 1, RUNMUTEXNAME)
if GetLastError() == winerror.ERROR_ALREADY_EXISTS:
print "another instance is already running"
return
# see http://www.jrsoftware.org/iskb.php?mutexsessions
-mutex2 = win32event.CreateMutex(sa, 1, 'Global\\' + MUTEXNAME)
+installmutex1 = win32event.CreateMutex(sa, 1, INSTALLMUTEXNAME)+ installmutex2 = win32event.CreateMutex(sa, 1, 'Global\\' + INSTALLMUTEXNAME)
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.