/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

Commit messages never contain config options

Show diffs side-by-side

added added

removed removed

Lines of Context:
branchview/treemodel.py
 
1
# -*- coding: UTF-8 -*-
 
2
"""BranchTreeModel."""
 
3
 
 
4
__copyright__ = "Copyright � 2005 Canonical Ltd."
 
5
__author__ = "Gary van der Merwe <garyvdm@gmail.com>"
 
6
 
 
7
 
 
8
from gi.repository import Gtk
 
9
from gi.repository import GObject
 
10
from xml.sax.saxutils import escape
 
11
 
 
12
from bzrlib.config import parse_username
 
13
from bzrlib.revision import NULL_REVISION
 
14
 
 
15
from time import (
 
16
    strftime,
 
17
    localtime,
 
18
    )
 
19
 
 
20
 
 
21
REVID = 0
 
22
NODE = 1
 
23
LINES = 2
 
24
LAST_LINES = 3
 
25
REVNO = 4
 
26
SUMMARY = 5
 
27
MESSAGE = 6
 
28
COMMITTER = 7
 
29
TIMESTAMP = 8
 
30
REVISION = 9
 
31
PARENTS = 10
 
32
CHILDREN = 11
 
33
TAGS = 12
 
34
AUTHORS = 13
 
35
 
 
36
 
 
37
class BranchTreeModel(Gtk.ListStore):
 
38
    """A model of branch's merge history."""
 
39
 
 
40
    def __init__(self, branch, line_graph_data):
 
41
        super(BranchTreeModel, self).__init__(
 
42
            GObject.TYPE_STRING,
 
43
            GObject.TYPE_PYOBJECT,
 
44
            GObject.TYPE_PYOBJECT,
 
45
            GObject.TYPE_PYOBJECT,
 
46
            GObject.TYPE_STRING,
 
47
            GObject.TYPE_STRING,
 
48
            GObject.TYPE_STRING,
 
49
            GObject.TYPE_STRING,
 
50
            GObject.TYPE_STRING,
 
51
            GObject.TYPE_PYOBJECT,
 
52
            GObject.TYPE_PYOBJECT,
 
53
            GObject.TYPE_PYOBJECT,
 
54
            GObject.TYPE_PYOBJECT,
 
55
            GObject.TYPE_STRING)
 
56
        self.revisions = {}
 
57
        self.branch = branch
 
58
        self.repository = branch.repository
 
59
        if self.branch.supports_tags():
 
60
            self.tags = self.branch.tags.get_reverse_tag_dict()
 
61
        else:
 
62
            self.tags = {}
 
63
        self.set_line_graph_data(line_graph_data)
 
64
 
 
65
    def add_tag(self, tag, revid):
 
66
        self.branch.tags.set_tag(tag, revid)
 
67
        try:
 
68
            self.tags[revid].append(tag)
 
69
        except KeyError:
 
70
            self.tags[revid] = [tag]
 
71
 
 
72
    def _line_graph_item_to_model_row(self, rowref, data):
 
73
        revid, node, lines, parents, children, revno_sequence = data
 
74
        if rowref > 0:
 
75
            last_lines = self.line_graph_data[rowref - 1][2]
 
76
        else:
 
77
            last_lines = []
 
78
        revno = ".".join(["%d" % (revno) for revno in revno_sequence])
 
79
        tags = self.tags.get(revid, [])
 
80
        if not revid or revid == NULL_REVISION:
 
81
            revision = None
 
82
        elif revid not in self.revisions:
 
83
            revision = self.repository.get_revisions([revid])[0]
 
84
            self.revisions[revid] = revision
 
85
        else:
 
86
            revision = self.revisions[revid]
 
87
        if revision is None:
 
88
            summary = message = committer = timestamp = authors = None
 
89
        else:
 
90
            summary = escape(revision.get_summary())
 
91
            message = escape(revision.message)
 
92
            committer = parse_username(revision.committer)[0]
 
93
            timestamp = strftime(
 
94
                "%Y-%m-%d %H:%M", localtime(revision.timestamp))
 
95
            authors = ", ".join([
 
96
                parse_username(author)[0]
 
97
                for author in revision.get_apparent_authors()])
 
98
        return (revid, node, lines, last_lines, revno, summary, message,
 
99
                committer, timestamp, revision, parents, children, tags,
 
100
                authors)
 
101
 
 
102
    def set_line_graph_data(self, line_graph_data):
 
103
        self.clear()
 
104
        self.line_graph_data = line_graph_data
 
105
        for rowref, data in enumerate(self.line_graph_data):
 
106
            row = self._line_graph_item_to_model_row(rowref, data)
 
107
            self.append(row)