/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 viz/treemodel.py

  • Committer: Daniel Schierbeck
  • Date: 2007-12-06 23:37:06 UTC
  • mto: This revision was merged to the branch mainline in revision 417.
  • Revision ID: daniel.schierbeck@gmail.com-20071206233706-eeinks66w86r3gfm
Fixed bug in gmissing.

Show diffs side-by-side

added added

removed removed

Lines of Context:
20
20
LINES = 2
21
21
LAST_LINES = 3
22
22
REVNO = 4
23
 
SUMMARY = 5
24
 
MESSAGE = 6
25
 
COMMITTER = 7
26
 
TIMESTAMP = 8
27
 
REVISION = 9
28
 
PARENTS = 10
29
 
CHILDREN = 11
30
 
TAGS = 12
 
23
MESSAGE = 5
 
24
COMMITER = 6
 
25
TIMESTAMP = 7
 
26
REVISION = 8
 
27
PARENTS = 9
 
28
CHILDREN = 10
31
29
 
32
30
class TreeModel(gtk.GenericTreeModel):
33
31
 
34
32
    
35
 
    def __init__ (self, branch, line_graph_data):
 
33
    def __init__ (self, repository, line_graph_data):
36
34
        gtk.GenericTreeModel.__init__(self)
37
35
        self.revisions = {}
38
 
        self.branch = branch
39
 
        self.repository = branch.repository
 
36
        self.repository = repository
40
37
        self.line_graph_data = line_graph_data
41
 
 
42
 
        if self.branch.supports_tags():
43
 
            self.tags = self.branch.tags.get_reverse_tag_dict()
44
 
        else:
45
 
            self.tags = {}
46
 
 
47
 
    def add_tag(self, tag, revid):
48
 
        self.branch.tags.set_tag(tag, revid)
49
 
        try:
50
 
            self.tags[revid].append(tag)
51
 
        except KeyError:
52
 
            self.tags[revid] = [tag]
53
 
 
 
38
    
54
39
    def on_get_flags(self):
55
40
        return gtk.TREE_MODEL_LIST_ONLY
56
41
    
57
42
    def on_get_n_columns(self):
58
 
        return 13
 
43
        return 11
59
44
    
60
45
    def on_get_column_type(self, index):
61
46
        if index == REVID: return gobject.TYPE_STRING
63
48
        if index == LINES: return gobject.TYPE_PYOBJECT
64
49
        if index == LAST_LINES: return gobject.TYPE_PYOBJECT
65
50
        if index == REVNO: return gobject.TYPE_STRING
66
 
        if index == SUMMARY: return gobject.TYPE_STRING
67
51
        if index == MESSAGE: return gobject.TYPE_STRING
68
 
        if index == COMMITTER: return gobject.TYPE_STRING
 
52
        if index == COMMITER: return gobject.TYPE_STRING
69
53
        if index == TIMESTAMP: return gobject.TYPE_STRING
70
54
        if index == REVISION: return gobject.TYPE_PYOBJECT
71
55
        if index == PARENTS: return gobject.TYPE_PYOBJECT
72
56
        if index == CHILDREN: return gobject.TYPE_PYOBJECT
73
 
        if index == TAGS: return gobject.TYPE_PYOBJECT
74
57
        
75
58
    def on_get_iter(self, path):
76
59
        return path[0]
79
62
        return rowref
80
63
    
81
64
    def on_get_value(self, rowref, column):
82
 
        if len(self.line_graph_data) > 0:
83
 
            (revid, node, lines, parents,
84
 
             children, revno_sequence) = self.line_graph_data[rowref]
85
 
        else:
86
 
            (revid, node, lines, parents,
87
 
             children, revno_sequence) = (None, (0, 0), (), (),
88
 
                                          (), ())
 
65
        (revid, node, lines, parents,
 
66
         children, revno_sequence) = self.line_graph_data[rowref]
89
67
        if column == REVID: return revid
90
68
        if column == NODE: return node
91
69
        if column == LINES: return lines
97
75
            return []
98
76
        if column == REVNO: return ".".join(["%d" % (revno)
99
77
                                      for revno in revno_sequence])
100
 
 
101
 
        if column == TAGS: return self.tags.get(revid, [])
102
 
 
103
 
        if revid is None:
104
 
            return None
 
78
        
105
79
        if revid not in self.revisions:
106
80
            revision = self.repository.get_revisions([revid])[0]
107
81
            self.revisions[revid] = revision
109
83
            revision = self.revisions[revid]
110
84
        
111
85
        if column == REVISION: return revision
112
 
        if column == SUMMARY: return escape(revision.get_summary())
113
 
        if column == MESSAGE: return escape(revision.message)
114
 
        if column == COMMITTER: return re.sub('<.*@.*>', '', 
 
86
        if column == MESSAGE: return escape(revision.get_summary())
 
87
        if column == COMMITER: return re.sub('<.*@.*>', '', 
115
88
                                             revision.committer).strip(' ')
116
89
        if column == TIMESTAMP: 
117
90
            return strftime("%Y-%m-%d %H:%M", localtime(revision.timestamp))