← Back to branch summary

~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to heads.py

  • Committer: Aaron Bentley
  • Date: 2008-11-12 17:09:36 UTC
  • Revision ID: aaron@aaronbentley.com-20081112170936-3dybsv7il8gxhchz
Use bzrlib's getchar

Show diffs side-by-side

added added

removed removed

Lines of Context:
heads.py
 
1
# Written by Alexander Belchenko
 
2
# based on Robert Collins code
 
3
#
 
4
#    This program is free software; you can redistribute it and/or modify
 
5
#    it under the terms of the GNU General Public License as published by
 
6
#    the Free Software Foundation; either version 2 of the License, or
 
7
#    (at your option) any later version.
 
8
#
 
9
#    This program is distributed in the hope that it will be useful,
 
10
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
 
11
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
12
#    GNU General Public License for more details.
 
13
#
 
14
#    You should have received a copy of the GNU General Public License
 
15
#    along with this program; if not, write to the Free Software
 
16
#    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
17
 
 
18
"""Show all 'heads' in a repository"""
 
19
 
 
20
 
 
21
import os
 
22
import time
 
23
 
 
24
import bzrlib
 
25
from bzrlib.commands import Command, display_command, register_command
 
26
from bzrlib import errors
 
27
from bzrlib.option import Option
 
28
from bzrlib.urlutils import unescape_for_display
 
29
 
 
30
 
 
31
class cmd_heads(Command):
 
32
    """Show all revisions in a repository not having descendants.
 
33
    """
 
34
    takes_options = [Option('by-date',
 
35
                     help='Sort heads by date (descending).'),
 
36
                     Option('all', help='Show all heads (dead and alive).'),
 
37
                     Option('dead-only', help='Show only dead heads.'),
 
38
                     Option('tips', help='Show tips of all branches.'),
 
39
                     Option('debug-time',
 
40
                     help='Enable debug print of operations times.'),
 
41
                    ]
 
42
 
 
43
    encoding_type = "replace"
 
44
    takes_args = ['location?']
 
45
 
 
46
    @display_command
 
47
    def run(self, by_date=False, all=False, dead_only=False, tips=False,
 
48
            debug_time=False, location='.'):
 
49
        import bzrlib.branch
 
50
        from bzrlib.osutils import format_date
 
51
        import bzrlib.repository
 
52
 
 
53
        self._init_elapsed_time(debug_time)
 
54
 
 
55
        to_file = self.outf
 
56
 
 
57
        branch = None
 
58
        try:
 
59
            branch = bzrlib.branch.Branch.open_containing(location)[0]
 
60
            repo = branch.repository
 
61
        except errors.NotBranchError:
 
62
            try:
 
63
                repo = bzrlib.repository.Repository.open(location)
 
64
            except errors.NotBranchError:
 
65
                print >>to_file, \
 
66
                      ("You need to run this command "
 
67
                       "either from the root of a shared repository,\n"
 
68
                       "or from a branch.")
 
69
                return 3
 
70
        repo.lock_read()
 
71
        try:
 
72
            possible_heads = set(repo.all_revision_ids())
 
73
            g = repo.get_graph().get_parent_map(possible_heads)
 
74
            not_heads = set()
 
75
            for parents in g.values():
 
76
                not_heads.update(set(parents))
 
77
 
 
78
            self.heads = possible_heads.difference(not_heads)
 
79
 
 
80
            # TODO: use different sorting schemes instead of alphabetical sort
 
81
            self.heads = list(self.heads)
 
82
 
 
83
            self._print_elapsed_time('get heads:')
 
84
 
 
85
            ## mark heads as dead or alive
 
86
            # mark all heads as dead
 
87
            self.head_mark = {}
 
88
            self.tips = {}
 
89
            for head in self.heads:
 
90
                self.head_mark[head] = 'dead'
 
91
            # give the list of live branches in repository or current branch
 
92
            # tip
 
93
            self._iter_branches_update_marks(repo, self.outf.encoding)
 
94
            self._print_elapsed_time('make head marks:')
 
95
 
 
96
            if tips:
 
97
                heads_tips = set(self.heads)
 
98
                heads_tips.update(set(self.tips.keys()))
 
99
                self.heads = list(heads_tips)
 
100
            head_revisions = dict(zip(self.heads,
 
101
                                  repo.get_revisions(self.heads)))
 
102
 
 
103
            # sorting by date
 
104
            if by_date:
 
105
                dates = {}
 
106
                for head in self.heads:
 
107
                    rev = head_revisions[head]
 
108
                    timestamp = rev.timestamp
 
109
                    dates[timestamp] = head
 
110
                keys = dates.keys()
 
111
                keys.sort()
 
112
                keys.reverse()
 
113
                self.heads = []
 
114
                for k in keys:
 
115
                    self.heads.append(dates[k])
 
116
                self._print_elapsed_time('sort by date:')
 
117
 
 
118
 
 
119
            # show time
 
120
            indent = ' '*2
 
121
            show_timezone = 'original'
 
122
 
 
123
            for head in self.heads:
 
124
 
 
125
                mark = self.head_mark[head]
 
126
                if not all:
 
127
                    if dead_only:
 
128
                        if mark != 'dead':
 
129
                            continue
 
130
                    else:
 
131
                        if mark != 'alive':
 
132
                            if not tips or mark != 'tip':
 
133
                                continue
 
134
 
 
135
                # tips
 
136
                if mark in ('alive', 'tip'):
 
137
                    t = self.tips[head]
 
138
                    if len(t) > 1:
 
139
                        print >>to_file, 'TIP of branches:',
 
140
                        print >>to_file, '[', ', '.join(t), ']'
 
141
                    else:
 
142
                        print >>to_file, 'TIP of branch:', t[0]
 
143
 
 
144
                if mark in ('alive', 'dead'):
 
145
                    print >>to_file, 'HEAD:',
 
146
 
 
147
                print >>to_file, "revision-id:", head,
 
148
                if mark == 'dead':  print >>to_file, '(dead)'
 
149
                else:               print >>to_file
 
150
 
 
151
                rev = head_revisions[head]
 
152
                # borrowed from LongLogFormatter
 
153
                print >>to_file,  indent+'committer:', rev.committer
 
154
                try:
 
155
                    print >>to_file, indent+'branch nick: %s' % \
 
156
                        rev.properties['branch-nick']
 
157
                except KeyError:
 
158
                    pass
 
159
                date_str = format_date(rev.timestamp,
 
160
                                       rev.timezone or 0,
 
161
                                       show_timezone)
 
162
                print >>to_file,  indent+'timestamp: %s' % date_str
 
163
 
 
164
                print >>to_file,  indent+'message:'
 
165
                if not rev.message:
 
166
                    print >>to_file,  indent+'  (no message)'
 
167
                else:
 
168
                    message = rev.message.rstrip('\r\n')
 
169
                    for l in message.split('\n'):
 
170
                        print >>to_file,  indent+'  ' + l
 
171
                self._print_elapsed_time('print head:')
 
172
                print
 
173
 
 
174
            if not self.heads:
 
175
                print >>to_file, 'No heads found'
 
176
                return 1
 
177
        finally:
 
178
            repo.unlock()
 
179
 
 
180
    def _iter_branches_update_marks(self, repo, encoding):
 
181
        for b in repo.find_branches(using=True):
 
182
            last_revid = b.last_revision()
 
183
            if last_revid in self.heads:
 
184
                self.head_mark[last_revid] = 'alive'
 
185
            else:
 
186
                self.head_mark[last_revid] = 'tip'
 
187
            escaped = unescape_for_display(b.base, encoding)
 
188
            self.tips.setdefault(last_revid, []).append(escaped)
 
189
 
 
190
    def _init_elapsed_time(self, debug_time=False):
 
191
        self.debug_time = debug_time
 
192
        if debug_time:
 
193
            self._time = time.time()
 
194
 
 
195
    def _print_elapsed_time(self, msg):
 
196
        if self.debug_time:
 
197
            last_time = time.time()
 
198
            print msg, last_time - self._time
 
199
            self._time = last_time
 
200
#/class cmd_heads