/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: Aaron Bentley
  • Date: 2008-02-23 04:55:56 UTC
  • mfrom: (423.1.16 trunk)
  • Revision ID: aaron@aaronbentley.com-20080223045556-xl3kf2v61crquq83
MergeĀ fromĀ mainline

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
 
MESSAGE = 5
24
 
COMMITER = 6
25
 
TIMESTAMP = 7
26
 
REVISION = 8
27
 
PARENTS = 9
28
 
CHILDREN = 10
 
23
SUMMARY = 5
 
24
MESSAGE = 6
 
25
COMMITTER = 7
 
26
TIMESTAMP = 8
 
27
REVISION = 9
 
28
PARENTS = 10
 
29
CHILDREN = 11
 
30
TAGS = 12
29
31
 
30
32
class TreeModel(gtk.GenericTreeModel):
31
33
 
32
34
    
33
 
    def __init__ (self, repository, line_graph_data):
 
35
    def __init__ (self, branch, line_graph_data):
34
36
        gtk.GenericTreeModel.__init__(self)
35
37
        self.revisions = {}
36
 
        self.repository = repository
 
38
        self.branch = branch
 
39
        self.repository = branch.repository
37
40
        self.line_graph_data = line_graph_data
38
 
    
 
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
 
39
54
    def on_get_flags(self):
40
55
        return gtk.TREE_MODEL_LIST_ONLY
41
56
    
42
57
    def on_get_n_columns(self):
43
 
        return 11
 
58
        return 13
44
59
    
45
60
    def on_get_column_type(self, index):
46
61
        if index == REVID: return gobject.TYPE_STRING
48
63
        if index == LINES: return gobject.TYPE_PYOBJECT
49
64
        if index == LAST_LINES: return gobject.TYPE_PYOBJECT
50
65
        if index == REVNO: return gobject.TYPE_STRING
 
66
        if index == SUMMARY: return gobject.TYPE_STRING
51
67
        if index == MESSAGE: return gobject.TYPE_STRING
52
 
        if index == COMMITER: return gobject.TYPE_STRING
 
68
        if index == COMMITTER: return gobject.TYPE_STRING
53
69
        if index == TIMESTAMP: return gobject.TYPE_STRING
54
70
        if index == REVISION: return gobject.TYPE_PYOBJECT
55
71
        if index == PARENTS: return gobject.TYPE_PYOBJECT
56
72
        if index == CHILDREN: return gobject.TYPE_PYOBJECT
 
73
        if index == TAGS: return gobject.TYPE_PYOBJECT
57
74
        
58
75
    def on_get_iter(self, path):
59
76
        return path[0]
80
97
            return []
81
98
        if column == REVNO: return ".".join(["%d" % (revno)
82
99
                                      for revno in revno_sequence])
83
 
        
 
100
 
 
101
        if column == TAGS: return self.tags.get(revid, [])
 
102
 
84
103
        if revid is None:
85
104
            return None
86
105
        if revid not in self.revisions:
90
109
            revision = self.revisions[revid]
91
110
        
92
111
        if column == REVISION: return revision
93
 
        if column == MESSAGE: return escape(revision.get_summary())
94
 
        if column == COMMITER: return re.sub('<.*@.*>', '', 
 
112
        if column == SUMMARY: return escape(revision.get_summary())
 
113
        if column == MESSAGE: return escape(revision.message)
 
114
        if column == COMMITTER: return re.sub('<.*@.*>', '', 
95
115
                                             revision.committer).strip(' ')
96
116
        if column == TIMESTAMP: 
97
117
            return strftime("%Y-%m-%d %H:%M", localtime(revision.timestamp))