/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: Jelmer Vernooij
  • Date: 2008-06-29 18:31:29 UTC
  • mto: This revision was merged to the branch mainline in revision 518.
  • Revision ID: jelmer@samba.org-20080629183129-syqvz3xm5gqagzsx
https://launchpad.net/bugs/244025
Fix use of smart_add.

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