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.
## Data Mining dialog for TortoiseHg and Mercurial## Copyright (C) 2008 Steve Borho <steve@borho.org>importgtkimportgobjectimportosimportpangoimportQueueimportthreading,thread2from mercurial import hg, ui, util, revlog
from mercurial.i18n import _
from hglib import hgcmd_toq, toutf, fromutf, gettabwidth, displaytime, LookupError, rootpath
-from gdialog import GDialog
+from gdialog import GDialog, Promptfrom vis import treemodel
from vis.colormap import AnnotateColorMap, AnnotateColorSaturation
from vis.treeview import TreeView
importgtklibclassDataMineDialog(GDialog):COL_REVID=0COL_TEXT=1COL_TOOLTIP=2COL_PATH=3COL_COLOR=4COL_USER=5COL_LINENUM=6defget_title(self):return_('DataMining')+' - '+os.path.basename(self.repo.root)defget_icon(self):return'menurepobrowse.ico'defparse_opts(self):passdefget_tbbuttons(self):self.stop_button=self.make_toolbutton(gtk.STOCK_STOP,_('Stop'),self._stop_current_search,tip=_('Stop operation on current tab'))return[self.make_toolbutton(gtk.STOCK_FIND,_('New Search'),self._search_clicked,tip=_('Open new search tab')),self.stop_button]defprepare_display(self):root=self.repo.rootcf=[]forfinself.pats:ifos.path.isfile(f):cf.append(util.canonpath(root,self.cwd,f))elifos.path.isdir(f):Prompt(_('Invalid path'),_('Cannot annotate directory: %s')%f,None).run()forfincf:self.add_annotate_page(f,'.')ifnotself.notebook.get_n_pages():self.add_search_page()os.chdir(root)defsave_settings(self):settings=GDialog.save_settings(self)settings['datamine']=()returnsettingsdefload_settings(self,settings):GDialog.load_settings(self,settings)self.connect('thg-close',self._close_current_page)self.tabwidth=gettabwidth(self.ui)# settings['datamine']defget_body(self):""" Initialize the Dialog. """self.grep_cmenu=self.grep_context_menu()self.changedesc={}self.newpagecount=1vbox=gtk.VBox()notebook=gtk.Notebook()notebook.set_tab_pos(gtk.POS_TOP)notebook.set_scrollable(True)notebook.popup_enable()notebook.show()self.notebook=notebookvbox.pack_start(self.notebook,True,True,2)self.stbar=gtklib.StatusBar()self.stbar.sttext.set_property('use-markup',True)vbox.pack_start(self.stbar,False,False,2)self.stop_button.set_sensitive(False)returnvboxdef_destroying(self,gtkobj):self._stop_all_searches()GDialog._destroying(self,gtkobj)defann_header_context_menu(self,treeview):_menu=gtk.Menu()_button=gtk.CheckMenuItem(_('Filename'))_button.connect('toggled',self.toggle_annatate_columns,treeview,2)_menu.append(_button)_button=gtk.CheckMenuItem(_('User'))_button.connect('toggled',self.toggle_annatate_columns,treeview,3)_menu.append(_button)_menu.show_all()return_menudefgrep_context_menu(self):_menu=gtk.Menu()_menu.append(create_menu(_('di_splay change'),self._cmenu_display))_menu.append(create_menu(_('_annotate file'),self._cmenu_annotate))_menu.append(create_menu(_('_file history'),self._cmenu_file_log))_menu.show_all()return_menudefannotate_context_menu(self,opts):_menu=gtk.Menu()_menu.append(create_menu(_('di_splay change'),self._cmenu_display))_menu.append(create_menu(_('_annotate parent'),self._annotate_parent,opts))_menu.show_all()return_menudef_annotate_parent(self,menuitem,objs):ifnotself.currev:returnparent=self.repo[self.currev].parents()[0].rev()self.trigger_annotate(parent,objs)def_cmenu_display(self,menuitem):fromchangesetimportChangeSetstatopts={'rev':[self.currev]}dialog=ChangeSet(self.ui,self.repo,self.cwd,[],statopts)dialog.display()def_cmenu_annotate(self,menuitem):self.add_annotate_page(self.curpath,self.currev)def_cmenu_file_log(self,menuitem):fromhistoryimportGLogdialog=GLog(self.ui,self.repo,self.cwd,[self.repo.root],{})dialog.open_with_file(self.curpath)dialog.display()def_grep_button_release(self,widget,event):ifevent.button==3andnot(event.state&(gtk.gdk.SHIFT_MASK|gtk.gdk.CONTROL_MASK)):self._grep_popup_menu(widget,event.button,event.time)returnFalsedef_grep_popup_menu(self,treeview,button=0,time=0):self.grep_cmenu.popup(None,None,None,button,time)returnTruedef_grep_row_act(self,tree,path,column):"""Default action is the first entry in the context menu """self.grep_cmenu.get_children()[0].activate()returnTruedefget_rev_desc(self,rev):ifrevinself.changedesc:returnself.changedesc[rev]ctx=self.repo.changectx(rev)author=util.shortuser(ctx.user())summary=ctx.description().replace('\0','')summary=gobject.markup_escape_text(summary.split('\n')[0])date=displaytime(ctx.date())desc=author+'@'+str(rev)+' '+date+' "'+summary+'"'self.changedesc[rev]=(desc,author)return(desc,author)def_search_clicked(self,button,data):self.add_search_page()defcreate_tab_close_button(self):button=gtk.Button()iconBox=gtk.HBox(False,0)image=gtk.Image()image.set_from_stock(gtk.STOCK_CLOSE,gtk.ICON_SIZE_MENU)gtk.Button.set_relief(button,gtk.RELIEF_NONE)settings=gtk.Widget.get_settings(button)(w,h)=gtk.icon_size_lookup_for_settings(settings,gtk.ICON_SIZE_MENU)gtk.Widget.set_size_request(button,w+4,h+4)image.show()iconBox.pack_start(image,True,False,0)button.add(iconBox)iconBox.show()returnbuttondefadd_search_page(self):frame=gtk.Frame()frame.set_border_width(10)vbox=gtk.VBox()search_hbox=gtk.HBox()regexp=gtk.Entry()includes=gtk.Entry()ifself.cwd.startswith(self.repo.root):includes.set_text(util.canonpath(self.repo.root,self.cwd,'.'))excludes=gtk.Entry()search=gtk.Button(_('Search'))search_hbox.pack_start(gtk.Label(_('Regexp:')),False,False,4)search_hbox.pack_start(regexp,True,True,4)search_hbox.pack_start(gtk.Label(_('Includes:')),False,False,4)search_hbox.pack_start(includes,True,True,4)search_hbox.pack_start(gtk.Label(_('Excludes:')),False,False,4)search_hbox.pack_start(excludes,True,True,4)search_hbox.pack_start(search,False,False)self.tooltips.set_tip(search,_('Start this search'))self.tooltips.set_tip(regexp,_('Regular expression search pattern'))self.tooltips.set_tip(includes,_('Comma separated list of'' inclusion patterns. By default, the entire repository'' is searched.'))self.tooltips.set_tip(excludes,_('Comma separated list of'' exclusion patterns. Exclusion patterns are applied'' after inclusion patterns.'))vbox.pack_start(search_hbox,False,False,4)hbox=gtk.HBox()follow=gtk.CheckButton(_('Follow copies and renames'))ignorecase=gtk.CheckButton(_('Ignore case'))linenum=gtk.CheckButton(_('Show line numbers'))showall=gtk.CheckButton(_('Show all matching revisions'))hbox.pack_start(follow,False,False,4)hbox.pack_start(ignorecase,False,False,4)hbox.pack_start(linenum,False,False,4)hbox.pack_start(showall,False,False,4)vbox.pack_start(hbox,False,False,4)treeview=gtk.TreeView()treeview.get_selection().set_mode(gtk.SELECTION_SINGLE)treeview.set_rules_hint(True)treeview.set_property('fixed-height-mode',True)treeview.connect("cursor-changed",self._grep_selection_changed)treeview.connect('button-release-event',self._grep_button_release)treeview.connect('popup-menu',self._grep_popup_menu)treeview.connect('row-activated',self._grep_row_act)results=gtk.ListStore(str,str,str,str)treeview.set_model(results)treeview.set_search_equal_func(self.search_in_grep)fortitle,width,col,emodein((_('Rev'),10,self.COL_REVID,pango.ELLIPSIZE_NONE),(_('File'),25,self.COL_PATH,pango.ELLIPSIZE_START),(_('Matches'),80,self.COL_TEXT,pango.ELLIPSIZE_END)):cell=gtk.CellRendererText()cell.set_property('width-chars',width)cell.set_property('ellipsize',emode)cell.set_property('family','Monospace')column=gtk.TreeViewColumn(title)column.set_resizable(True)column.set_sizing(gtk.TREE_VIEW_COLUMN_FIXED)column.set_fixed_width(cell.get_size(treeview)[2])column.pack_start(cell,expand=True)column.add_attribute(cell,'text',col)treeview.append_column(column)ifhasattr(treeview,'set_tooltip_column'):treeview.set_tooltip_column(self.COL_TOOLTIP)scroller=gtk.ScrolledWindow()scroller.set_policy(gtk.POLICY_AUTOMATIC,gtk.POLICY_AUTOMATIC)scroller.add(treeview)vbox.pack_start(scroller,True,True)frame.add(vbox)frame.show_all()hbox=gtk.HBox()lbl=gtk.Label(_('Search %d')%self.newpagecount)close=self.create_tab_close_button()close.connect('clicked',self.close_page,frame)hbox.pack_start(lbl,True,True,2)hbox.pack_start(close,False,False)hbox.show_all()num=self.notebook.append_page(frame,hbox)self.newpagecount+=1objs=(treeview.get_model(),frame,regexp,follow,ignorecase,excludes,includes,linenum,showall,search_hbox)# Clicking 'search' or hitting Enter in any text entry triggers searchsearch.connect('clicked',self.trigger_search,objs)regexp.connect('activate',self.trigger_search,objs)includes.connect('activate',self.trigger_search,objs)excludes.connect('activate',self.trigger_search,objs)# Includes/excludes must disable following copiesobjs=(includes,excludes,follow)includes.connect('changed',self._update_following_possible,objs)excludes.connect('changed',self._update_following_possible,objs)ifhasattr(self.notebook,'set_tab_reorderable'):self.notebook.set_tab_reorderable(frame,True)self.notebook.set_current_page(num)regexp.grab_focus()defsearch_in_grep(self,model,column,key,iter):"""Searches all fields shown in the tree when the user hits crtr+f, not just the ones that are set via tree.set_search_column. Case insensitive """key=key.lower()forcolin(self.COL_PATH,self.COL_TEXT):ifkeyinmodel.get_value(iter,col).lower():returnFalsereturnTruedeftrigger_search(self,button,objs):(model,frame,regexp,follow,ignorecase,excludes,includes,linenum,showall,search_hbox)=objsretext=regexp.get_text()ifnotretext:Prompt(_('No regular expression given'),_('You must provide a search expression'),self).run()regexp.grab_focus()returnq=Queue.Queue()args=[self.repo.root,q,'grep']iffollow.get_active():args.append('--follow')ifignorecase.get_active():args.append('--ignore-case')iflinenum.get_active():args.append('--line-number')ifshowall.get_active():args.append('--all')incs=[x.strip()forxinincludes.get_text().split(',')]excs=[x.strip()forxinexcludes.get_text().split(',')]foriinincs:ifi:args.extend(['-I',i])forxinexcs:ifx:args.extend(['-X',x])args.append(retext)thread=thread2.Thread(target=hgcmd_toq,args=args)thread.start()frame._mythread=threadself.stop_button.set_sensitive(True)model.clear()search_hbox.set_sensitive(False)self.stbar.begin()self.stbar.set_status_text('hg '+' '.join(args[2:]))hbox=gtk.HBox()lbl=gtk.Label(_('Search "%s"')%retext.split()[0])close=self.create_tab_close_button()close.connect('clicked',self.close_page,frame)hbox.pack_start(lbl,True,True,2)hbox.pack_start(close,False,False)hbox.show_all()self.notebook.set_tab_label(frame,hbox)gobject.timeout_add(50,self.grep_wait,thread,q,model,search_hbox,regexp,frame)defgrep_wait(self,thread,q,model,search_hbox,regexp,frame):""" Handle all the messages currently in the queue (if any). """whileq.qsize():line=q.get(0).rstrip('\r\n')try:(path,revid,text)=line.split(':',2)exceptValueError:continuetip,user=self.get_rev_desc(long(revid))ifself.tabwidth:text=text.expandtabs(self.tabwidth)model.append((revid,toutf(text[:128]),tip,toutf(path)))ifthread.isAlive():returnTrueelse:ifthreading.activeCount()==1:self.stop_button.set_sensitive(False)frame._mythread=Nonesearch_hbox.set_sensitive(True)regexp.grab_focus()self.stbar.end()returnFalsedef_grep_selection_changed(self,treeview):""" Callback for when the user selects grep output. """(path,focus)=treeview.get_cursor()model=treeview.get_model()ifpathisnotNoneandmodelisnotNone:paths=model.get_iter(path)self.currev=model[paths][self.COL_REVID]self.curpath=fromutf(model[paths][self.COL_PATH])self.stbar.set_status_text(toutf(model[paths][self.COL_TOOLTIP]))def_close_current_page(self,window):num=self.notebook.get_current_page()ifnum!=-1andself.notebook.get_n_pages():self.notebook.remove_page(num)self.emit_stop_by_name('thg-close')def_stop_current_search(self,button,widget):num=self.notebook.get_current_page()frame=self.notebook.get_nth_page(num)self._stop_search(frame)def_stop_all_searches(self):fornuminxrange(self.notebook.get_n_pages()):frame=self.notebook.get_nth_page(num)self._stop_search(frame)def_stop_search(self,frame):ifhasattr(frame,'_mythread')andframe._mythread:frame._mythread.terminate()frame._mythread.join()frame._mythread=Nonedefclose_page(self,button,widget):'''Close page button has been pressed'''num=self.notebook.page_num(widget)ifnum!=-1andself.notebook.get_n_pages()>1:self.notebook.remove_page(num)def_add_header_context_menu(self,col,menu):lb=gtk.Label(col.get_title())lb.show()col.set_widget(lb)wgt=lb.get_parent()whilewgt:iftype(wgt)==gtk.Button:wgt.connect("button-press-event",self._tree_header_button_press,menu)breakwgt=wgt.get_parent()def_tree_header_button_press(self,widget,event,menu):ifevent.button==3:menu.popup(None,None,None,event.button,event.time)returnTruereturnFalsedef_update_following_possible(self,widget,objs):(includes,excludes,follow)=objsallow=notincludes.get_text()andnotexcludes.get_text()ifnotallow:follow.set_active(False)follow.set_sensitive(allow)defadd_annotate_page(self,path,revid):''' Add new annotation page to notebook. Start scan of file 'path' revision history, start annotate of supplied revision 'revid'. '''ifrevid=='.':ctx=self.repo.changectx(None).parents()[0]try:fctx=ctx.filectx(path)exceptLookupError:Prompt(_('File is unrevisioned'),_('Unable to annotate ')+path,self).run()returnrev=fctx.filelog().linkrev(fctx.filerev())revid=str(rev)else:rev=long(revid)frame=gtk.Frame()frame.set_border_width(10)vbox=gtk.VBox()# File log revision graphgraphview=TreeView(self.repo,5000,self.stbar)graphview.connect('revisions-loaded',self.revisions_loaded,rev)graphview.refresh(True,None,{'filehist':path,'filerev':rev})graphview.set_property('rev-column-visible',True)graphview.set_property('date-column-visible',True)hbox=gtk.HBox()followlabel=gtk.Label('')follow=gtk.Button(_('Follow'))follow.connect('clicked',self.follow_rename)follow.hide()follow.set_sensitive(False)hbox.pack_start(gtk.Label(''),True,True)hbox.pack_start(followlabel,False,False)hbox.pack_start(follow,False,False)# Annotation text tree viewtreeview=gtk.TreeView()treeview.get_selection().set_mode(gtk.SELECTION_SINGLE)treeview.set_property('fixed-height-mode',True)treeview.set_border_width(0)results=gtk.ListStore(str,str,str,str,str,str,str)treeview.set_model(results)treeview.set_search_equal_func(self.search_in_file)context_menu=self.ann_header_context_menu(treeview)fortitle,width,col,emode,visiblein((_('Line'),8,self.COL_LINENUM,pango.ELLIPSIZE_NONE,True),(_('Rev'),10,self.COL_REVID,pango.ELLIPSIZE_NONE,True),(_('File'),15,self.COL_PATH,pango.ELLIPSIZE_START,False),(_('User'),15,self.COL_USER,pango.ELLIPSIZE_END,False),(_('Source'),80,self.COL_TEXT,pango.ELLIPSIZE_END,True)):cell=gtk.CellRendererText()cell.set_property('width-chars',width)cell.set_property('ellipsize',emode)cell.set_property('family','Monospace')column=gtk.TreeViewColumn(title)column.set_resizable(True)column.set_sizing(gtk.TREE_VIEW_COLUMN_FIXED)column.set_fixed_width(cell.get_size(treeview)[2])column.pack_start(cell,expand=True)column.add_attribute(cell,'text',col)column.add_attribute(cell,'background',self.COL_COLOR)column.set_visible(visible)treeview.append_column(column)self._add_header_context_menu(column,context_menu)treeview.set_headers_clickable(True)ifhasattr(treeview,'set_tooltip_column'):treeview.set_tooltip_column(self.COL_TOOLTIP)results.path=pathresults.rev=revidscroller=gtk.ScrolledWindow()scroller.set_policy(gtk.POLICY_AUTOMATIC,gtk.POLICY_AUTOMATIC)scroller.add(treeview)vpaned=gtk.VPaned()vpaned.pack1(graphview,True,True)vpaned.pack2(scroller,True,True)vbox.pack_start(vpaned,True,True)vbox.pack_start(hbox,False,False)frame.add(vbox)frame.show_all()hbox=gtk.HBox()lbl=gtk.Label(toutf(os.path.basename(path)+'@'+revid))close=self.create_tab_close_button()close.connect('clicked',self.close_page,frame)hbox.pack_start(lbl,True,True,2)hbox.pack_start(close,False,False)hbox.show_all()num=self.notebook.append_page_menu(frame,hbox,gtk.Label(toutf(path+'@'+revid)))ifhasattr(self.notebook,'set_tab_reorderable'):self.notebook.set_tab_reorderable(frame,True)self.notebook.set_current_page(num)graphview.connect('revision-selected',self.log_selection_changed,path,followlabel,follow)objs=(frame,treeview.get_model(),path)graphview.treeview.connect('row-activated',self.log_activate,objs)graphview.treeview.connect('button-release-event',self._ann_button_release,objs)graphview.treeview.connect('popup-menu',self._ann_popup_menu,objs)treeview.connect("cursor-changed",self._ann_selection_changed)treeview.connect('button-release-event',self._ann_button_release,objs)treeview.connect('popup-menu',self._ann_popup_menu,objs)treeview.connect('row-activated',self._ann_row_act,objs)defsearch_in_file(self,model,column,key,iter):"""Searches all fields shown in the tree when the user hits crtr+f, not just the ones that are set via tree.set_search_column. Case insensitive """key=key.lower()forcolin(self.COL_USER,self.COL_TEXT):ifkeyinmodel.get_value(iter,col).lower():returnFalsereturnTruedeftoggle_annatate_columns(self,button,treeview,col):b=button.get_active()treeview.get_column(col).set_visible(b)deflog_selection_changed(self,graphview,path,label,button):row=graphview.get_revision()rev=row[treemodel.REVID]self.currev=str(rev)ctx=self.repo.changectx(rev)filectx=ctx.filectx(path)info=filectx.renamed()ifinfo:(rpath,node)=infofl=self.repo.file(rpath)frev=fl.linkrev(fl.rev(node))button.set_label(toutf('%s@%s'%(rpath,frev)))button.show()button.set_sensitive(True)label.set_text(_('Follow Rename:'))else:button.hide()button.set_sensitive(False)label.set_text('')deffollow_rename(self,button):path,rev=button.get_label().rsplit('@',1)self.add_annotate_page(path,rev)deflog_activate(self,treeview,path,column,objs):model=treeview.get_model()logiter=model.get_iter(path)rev=model.get_value(logiter,treemodel.REVID)self.trigger_annotate(rev,objs)defrevisions_loaded(self,graphview,rev):graphview.set_revision_id(rev)treeview=graphview.treeviewpath,column=treeview.get_cursor()# It's possible that the requested change was not found in the# file's filelog history. In that case, no row will be# selected.ifpath!=None:treeview.row_activated(path,column)deftrigger_annotate(self,rev,objs):''' User has selected a file revision to annotate. Trigger a background thread to perform the annotation. Disable the select button until this operation is complete. '''(frame,model,path)=objsq=Queue.Queue()args=[self.repo.root,q,'annotate','--follow','--number','--rev',str(rev),path]thread=threading.Thread(target=hgcmd_toq,args=args)thread.start()frame._mythread=threadself.stop_button.set_sensitive(True)# date of selected revisionctx=self.repo.changectx(long(rev))curdate=ctx.date()[0]# date of initial revisionfctx=self.repo.filectx(path,fileid=0)basedate=fctx.date()[0]agedays=(curdate-basedate)/(24*60*60)colormap=AnnotateColorSaturation(agedays)model.clear()self.stbar.begin()self.stbar.set_status_text(toutf('hg '+' '.join(args[2:])))hbox=gtk.HBox()lbl=gtk.Label(toutf(os.path.basename(path)+'@'+str(rev)))close=self.create_tab_close_button()close.connect('clicked',self.close_page,frame)hbox.pack_start(lbl,True,True,2)hbox.pack_start(close,False,False)hbox.show_all()self.notebook.set_tab_label(frame,hbox)gobject.timeout_add(50,self.annotate_wait,thread,q,model,curdate,colormap,frame)defannotate_wait(self,thread,q,model,curdate,colormap,frame):""" Handle all the messages currently in the queue (if any). """whileq.qsize():line=q.get(0).rstrip('\r\n')try:(revpath,text)=line.split(':',1)revid,path=revpath.lstrip().split(' ',1)rowrev=long(revid)exceptValueError:continuetip,user=self.get_rev_desc(rowrev)ctx=self.repo.changectx(rowrev)color=colormap.get_color(ctx,curdate)ifself.tabwidth:text=text.expandtabs(self.tabwidth)model.append((revid,toutf(text[:128]),tip,toutf(path.strip()),color,toutf(user),len(model)+1))ifthread.isAlive():returnTrueelse:ifthreading.activeCount()==1:self.stop_button.set_sensitive(False)frame._mythread=Noneself.stbar.end()returnFalsedef_ann_selection_changed(self,treeview):""" User selected line of annotate output, describe revision responsible for this line in the status bar """(path,focus)=treeview.get_cursor()model=treeview.get_model()ifpathisnotNoneandmodelisnotNone:anniter=model.get_iter(path)self.currev=model[anniter][self.COL_REVID]self.path=model.pathself.stbar.set_status_text(model[anniter][self.COL_TOOLTIP])def_ann_button_release(self,widget,event,objs):ifevent.button==3andnot(event.state&(gtk.gdk.SHIFT_MASK|gtk.gdk.CONTROL_MASK)):self._ann_popup_menu(widget,event.button,event.time,objs)returnFalsedef_ann_popup_menu(self,treeview,button,time,objs):ann_cmenu=self.annotate_context_menu(objs)ann_cmenu.popup(None,None,None,button,time)returnTruedef_ann_row_act(self,tree,path,column,objs):ann_cmenu=self.annotate_context_menu(objs)ann_cmenu.get_children()[0].activate()defcreate_menu(label,callback,*args):menuitem=gtk.MenuItem(label,True)menuitem.connect('activate',callback,*args)menuitem.set_border_width(1)returnmenuitemdefrun(ui,*pats,**opts):cmdoptions={'follow':False,'follow-first':False,'copies':False,'keyword':[],'limit':0,'rev':[],'removed':False,'no_merges':False,'date':None,'only_merges':None,'prune':[],'git':False,'verbose':False,'include':[],'exclude':[]}returnDataMineDialog(ui,None,None,pats,cmdoptions)
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.