/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: Curtis Hovey
  • Date: 2011-09-03 01:25:04 UTC
  • mto: This revision was merged to the branch mainline in revision 741.
  • Revision ID: sinzui.is@verizon.net-20110903012504-0jr4diz9033g5df2
Menu fixes.

Show diffs side-by-side

added added

removed removed

Lines of Context:
branchview/treemodel.py
 
1
# -*- coding: UTF-8 -*-
 
2
"""Tree model.
 
3
 
 
4
"""
 
5
 
 
6
__copyright__ = "Copyright � 2005 Canonical Ltd."
 
7
__author__    = "Gary van der Merwe <garyvdm@gmail.com>"
 
8
 
 
9
 
 
10
from gi.repository import Gtk
 
11
from gi.repository import GObject
 
12
from xml.sax.saxutils import escape
 
13
 
 
14
from bzrlib.config import parse_username
 
15
from bzrlib.revision import NULL_REVISION
 
16
 
 
17
from time import (
 
18
    strftime,
 
19
    localtime,
 
20
    )
 
21
 
 
22
REVID = 0
 
23
NODE = 1
 
24
LINES = 2
 
25
LAST_LINES = 3
 
26
REVNO = 4
 
27
SUMMARY = 5
 
28
MESSAGE = 6
 
29
COMMITTER = 7
 
30
TIMESTAMP = 8
 
31
REVISION = 9
 
32
PARENTS = 10
 
33
CHILDREN = 11
 
34
TAGS = 12
 
35
AUTHORS = 13
 
36
 
 
37
class TreeModel(Gtk.TreeStore):
 
38
 
 
39
    def __init__ (self, branch, line_graph_data):
 
40
        Gtk.TreeStore.__init__(self)
 
41
        self.revisions = {}
 
42
        self.branch = branch
 
43
        self.repository = branch.repository
 
44
        self.line_graph_data = line_graph_data
 
45
 
 
46
        if self.branch.supports_tags():
 
47
            self.tags = self.branch.tags.get_reverse_tag_dict()
 
48
        else:
 
49
            self.tags = {}
 
50
 
 
51
    def add_tag(self, tag, revid):
 
52
        self.branch.tags.set_tag(tag, revid)
 
53
        try:
 
54
            self.tags[revid].append(tag)
 
55
        except KeyError:
 
56
            self.tags[revid] = [tag]
 
57
 
 
58
    def on_get_flags(self):
 
59
        return Gtk.TREE_MODEL_LIST_ONLY
 
60
 
 
61
    def on_get_n_columns(self):
 
62
        return 14
 
63
 
 
64
    def on_get_column_type(self, index):
 
65
        if index == REVID: return GObject.TYPE_STRING
 
66
        if index == NODE: return GObject.TYPE_PYOBJECT
 
67
        if index == LINES: return GObject.TYPE_PYOBJECT
 
68
        if index == LAST_LINES: return GObject.TYPE_PYOBJECT
 
69
        if index == REVNO: return GObject.TYPE_STRING
 
70
        if index == SUMMARY: return GObject.TYPE_STRING
 
71
        if index == MESSAGE: return GObject.TYPE_STRING
 
72
        if index == COMMITTER: return GObject.TYPE_STRING
 
73
        if index == TIMESTAMP: return GObject.TYPE_STRING
 
74
        if index == REVISION: return GObject.TYPE_PYOBJECT
 
75
        if index == PARENTS: return GObject.TYPE_PYOBJECT
 
76
        if index == CHILDREN: return GObject.TYPE_PYOBJECT
 
77
        if index == TAGS: return GObject.TYPE_PYOBJECT
 
78
        if index == AUTHORS: return GObject.TYPE_STRING
 
79
 
 
80
    def on_get_iter(self, path):
 
81
        # XXX sinzui 2011-08-12: maybe path.get_indices()[0]?
 
82
        return path[0]
 
83
 
 
84
    def on_get_path(self, rowref):
 
85
        return rowref
 
86
 
 
87
    def on_get_value(self, rowref, column):
 
88
        if len(self.line_graph_data) > 0:
 
89
            (revid, node, lines, parents,
 
90
             children, revno_sequence) = self.line_graph_data[rowref]
 
91
        else:
 
92
            (revid, node, lines, parents,
 
93
             children, revno_sequence) = (None, (0, 0), (), (),
 
94
                                          (), ())
 
95
        if column == REVID: return revid
 
96
        if column == NODE: return node
 
97
        if column == LINES: return lines
 
98
        if column == PARENTS: return parents
 
99
        if column == CHILDREN: return children
 
100
        if column == LAST_LINES:
 
101
            if rowref>0:
 
102
                return self.line_graph_data[rowref-1][2]
 
103
            return []
 
104
        if column == REVNO: return ".".join(["%d" % (revno)
 
105
                                      for revno in revno_sequence])
 
106
 
 
107
        if column == TAGS: return self.tags.get(revid, [])
 
108
 
 
109
        if not revid or revid == NULL_REVISION:
 
110
            return None
 
111
        if revid not in self.revisions:
 
112
            revision = self.repository.get_revisions([revid])[0]
 
113
            self.revisions[revid] = revision
 
114
        else:
 
115
            revision = self.revisions[revid]
 
116
 
 
117
        if column == REVISION: return revision
 
118
        if column == SUMMARY: return escape(revision.get_summary())
 
119
        if column == MESSAGE: return escape(revision.message)
 
120
        if column == COMMITTER: return parse_username(revision.committer)[0]
 
121
        if column == TIMESTAMP:
 
122
            return strftime("%Y-%m-%d %H:%M", localtime(revision.timestamp))
 
123
        if column == AUTHORS:
 
124
            return ", ".join([
 
125
                parse_username(author)[0] for author in revision.get_apparent_authors()])
 
126
 
 
127
    def on_iter_next(self, rowref):
 
128
        if rowref < len(self.line_graph_data) - 1:
 
129
            return rowref+1
 
130
        return None
 
131
 
 
132
    def on_iter_children(self, parent):
 
133
        if parent is None: return 0
 
134
        return None
 
135
 
 
136
    def on_iter_has_child(self, rowref):
 
137
        return False
 
138
 
 
139
    def on_iter_n_children(self, rowref):
 
140
        if rowref is None: return len(self.line_graph_data)
 
141
        return 0
 
142
 
 
143
    def on_iter_nth_child(self, parent, n):
 
144
        if parent is None: return n
 
145
        return None
 
146
 
 
147
    def on_iter_parent(self, child):
 
148
        return None