/b-gtk/fix-viz

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/b-gtk/fix-viz

« back to all changes in this revision

Viewing changes to branchview/treemodel.py

  • Committer: Daniel Schierbeck
  • Date: 2008-03-15 20:42:00 UTC
  • mto: (450.1.11 trunk) (399.1.17 signatures)
  • mto: This revision was merged to the branch mainline in revision 458.
  • Revision ID: daniel.schierbeck@gmail.com-20080315204200-xtbkppagzh434n38
Added signature icons.

Show diffs side-by-side

added added

removed removed

Lines of Context:
branchview/treemodel.py
12
12
import pango
13
13
import re
14
14
from xml.sax.saxutils import escape
15
 
from bzrlib.revision import NULL_REVISION
16
15
 
17
16
from time import (strftime, localtime)
18
17
 
28
27
REVISION = 9
29
28
PARENTS = 10
30
29
CHILDREN = 11
31
 
TAGS = 12
32
30
 
33
31
class TreeModel(gtk.GenericTreeModel):
34
32
 
35
33
    
36
 
    def __init__ (self, branch, line_graph_data):
 
34
    def __init__ (self, repository, line_graph_data):
37
35
        gtk.GenericTreeModel.__init__(self)
38
36
        self.revisions = {}
39
 
        self.branch = branch
40
 
        self.repository = branch.repository
 
37
        self.repository = repository
41
38
        self.line_graph_data = line_graph_data
42
 
 
43
 
        if self.branch.supports_tags():
44
 
            self.tags = self.branch.tags.get_reverse_tag_dict()
45
 
        else:
46
 
            self.tags = {}
47
 
 
48
 
    def add_tag(self, tag, revid):
49
 
        self.branch.tags.set_tag(tag, revid)
50
 
        try:
51
 
            self.tags[revid].append(tag)
52
 
        except KeyError:
53
 
            self.tags[revid] = [tag]
54
 
 
 
39
    
55
40
    def on_get_flags(self):
56
41
        return gtk.TREE_MODEL_LIST_ONLY
57
42
    
58
43
    def on_get_n_columns(self):
59
 
        return 13
 
44
        return 12
60
45
    
61
46
    def on_get_column_type(self, index):
62
47
        if index == REVID: return gobject.TYPE_STRING
71
56
        if index == REVISION: return gobject.TYPE_PYOBJECT
72
57
        if index == PARENTS: return gobject.TYPE_PYOBJECT
73
58
        if index == CHILDREN: return gobject.TYPE_PYOBJECT
74
 
        if index == TAGS: return gobject.TYPE_PYOBJECT
75
59
        
76
60
    def on_get_iter(self, path):
77
61
        return path[0]
98
82
            return []
99
83
        if column == REVNO: return ".".join(["%d" % (revno)
100
84
                                      for revno in revno_sequence])
101
 
 
102
 
        if column == TAGS: return self.tags.get(revid, [])
103
 
 
104
 
        if not revid or revid == NULL_REVISION:
 
85
        
 
86
        if revid is None:
105
87
            return None
106
88
        if revid not in self.revisions:
107
89
            revision = self.repository.get_revisions([revid])[0]