← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/info.py

  • Committer: John Arbash Meinel
  • Date: 2009-04-23 00:58:30 UTC
  • mfrom: (4300.2.1 1.15-gc-python)
  • mto: This revision was merged to the branch mainline in revision 4301.
  • Revision ID: john@arbash-meinel.com-20090423005830-kkdc31tqjetbj2f0
Bring in the other test cases.
Also, remove the assert statements.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/info.py
1
 
# Copyright (C) 2004, 2005 by Martin Pool
2
 
# Copyright (C) 2005 by Canonical Ltd
3
 
 
4
 
 
 
1
# Copyright (C) 2005, 2006, 2007 Canonical Ltd
 
2
#
5
3
# This program is free software; you can redistribute it and/or modify
6
4
# it under the terms of the GNU General Public License as published by
7
5
# the Free Software Foundation; either version 2 of the License, or
8
6
# (at your option) any later version.
9
 
 
 
7
#
10
8
# This program is distributed in the hope that it will be useful,
11
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
12
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
11
# GNU General Public License for more details.
14
 
 
 
12
#
15
13
# You should have received a copy of the GNU General Public License
16
14
# along with this program; if not, write to the Free Software
17
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18
 
 
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
 
 
17
__all__ = ['show_bzrdir_info']
 
18
 
 
19
import os
19
20
import time
20
 
 
21
 
from osutils import format_date
22
 
 
23
 
 
24
 
def _countiter(it):
25
 
    # surely there's a builtin for this?
26
 
    i = 0
27
 
    for j in it:
28
 
        i += 1
29
 
    return i        
30
 
 
31
 
 
32
 
 
33
 
def show_info(b):
34
 
    import diff
35
 
    
36
 
    print 'branch format:', b.controlfile('branch-format', 'r').readline().rstrip('\n')
37
 
 
38
 
    def plural(n, base='', pl=None):
39
 
        if n == 1:
40
 
            return base
41
 
        elif pl != None:
42
 
            return pl
43
 
        else:
44
 
            return 's'
45
 
 
46
 
    count_version_dirs = 0
47
 
 
48
 
    basis = b.basis_tree()
49
 
    working = b.working_tree()
50
 
    work_inv = working.inventory
51
 
    delta = diff.compare_trees(basis, working, want_unchanged=True)
52
 
    
53
 
    print
54
 
    print 'in the working tree:'
55
 
    print '  %8s unchanged' % len(delta.unchanged)
56
 
    print '  %8d modified' % len(delta.modified)
57
 
    print '  %8d added' % len(delta.added)
58
 
    print '  %8d removed' % len(delta.removed)
59
 
    print '  %8d renamed' % len(delta.renamed)
 
21
import sys
 
22
 
 
23
from bzrlib import (
 
24
    bzrdir,
 
25
    diff,
 
26
    errors,
 
27
    osutils,
 
28
    urlutils,
 
29
    )
 
30
from bzrlib.errors import (NoWorkingTree, NotBranchError,
 
31
                           NoRepositoryPresent, NotLocalUrl)
 
32
from bzrlib.missing import find_unmerged
 
33
 
 
34
 
 
35
def plural(n, base='', pl=None):
 
36
    if n == 1:
 
37
        return base
 
38
    elif pl is not None:
 
39
        return pl
 
40
    else:
 
41
        return 's'
 
42
 
 
43
 
 
44
class LocationList(object):
 
45
 
 
46
    def __init__(self, base_path):
 
47
        self.locs = []
 
48
        self.base_path = base_path
 
49
 
 
50
    def add_url(self, label, url):
 
51
        """Add a URL to the list, converting it to a path if possible"""
 
52
        if url is None:
 
53
            return
 
54
        try:
 
55
            path = urlutils.local_path_from_url(url)
 
56
        except errors.InvalidURL:
 
57
            self.locs.append((label, url))
 
58
        else:
 
59
            self.add_path(label, path)
 
60
 
 
61
    def add_path(self, label, path):
 
62
        """Add a path, converting it to a relative path if possible"""
 
63
        try:
 
64
            path = osutils.relpath(self.base_path, path)
 
65
        except errors.PathNotChild:
 
66
            pass
 
67
        else:
 
68
            if path == '':
 
69
                path = '.'
 
70
        if path != '/':
 
71
            path = path.rstrip('/')
 
72
        self.locs.append((label, path))
 
73
 
 
74
    def get_lines(self):
 
75
        max_len = max(len(l) for l, u in self.locs)
 
76
        return ["  %*s: %s\n" % (max_len, l, u) for l, u in self.locs ]
 
77
 
 
78
 
 
79
def gather_location_info(repository, branch=None, working=None):
 
80
    locs = {}
 
81
    repository_path = repository.bzrdir.root_transport.base
 
82
    if branch is not None:
 
83
        branch_path = branch.bzrdir.root_transport.base
 
84
        master_path = branch.get_bound_location()
 
85
        if master_path is None:
 
86
            master_path = branch_path
 
87
    else:
 
88
        branch_path = None
 
89
        master_path = None
 
90
    if working:
 
91
        working_path = working.bzrdir.root_transport.base
 
92
        if working_path != branch_path:
 
93
            locs['light checkout root'] = working_path
 
94
        if master_path != branch_path:
 
95
            if repository.is_shared():
 
96
                locs['repository checkout root'] = branch_path
 
97
            else:
 
98
                locs['checkout root'] = branch_path
 
99
        if working_path != master_path:
 
100
            locs['checkout of branch'] = master_path
 
101
        elif repository.is_shared():
 
102
            locs['repository branch'] = branch_path
 
103
        elif branch_path is not None:
 
104
            # standalone
 
105
            locs['branch root'] = branch_path
 
106
    else:
 
107
        working_path = None
 
108
        if repository.is_shared():
 
109
            # lightweight checkout of branch in shared repository
 
110
            if branch_path is not None:
 
111
                locs['repository branch'] = branch_path
 
112
        elif branch_path is not None:
 
113
            # standalone
 
114
            locs['branch root'] = branch_path
 
115
            if master_path != branch_path:
 
116
                locs['bound to branch'] = master_path
 
117
        else:
 
118
            locs['repository'] = repository_path
 
119
    if repository.is_shared():
 
120
        # lightweight checkout of branch in shared repository
 
121
        locs['shared repository'] = repository_path
 
122
    order = ['light checkout root', 'repository checkout root',
 
123
             'checkout root', 'checkout of branch', 'shared repository',
 
124
             'repository', 'repository branch', 'branch root',
 
125
             'bound to branch']
 
126
    return [(n, locs[n]) for n in order if n in locs]
 
127
 
 
128
 
 
129
def _show_location_info(locs, outfile):
 
130
    """Show known locations for working, branch and repository."""
 
131
    outfile.write('Location:\n')
 
132
    path_list = LocationList(osutils.getcwd())
 
133
    for name, loc in locs:
 
134
        path_list.add_url(name, loc)
 
135
    outfile.writelines(path_list.get_lines())
 
136
 
 
137
 
 
138
def _gather_related_branches(branch):
 
139
    locs = LocationList(osutils.getcwd())
 
140
    locs.add_url('public branch', branch.get_public_branch())
 
141
    locs.add_url('push branch', branch.get_push_location())
 
142
    locs.add_url('parent branch', branch.get_parent())
 
143
    locs.add_url('submit branch', branch.get_submit_branch())
 
144
    try:
 
145
        locs.add_url('stacked on', branch.get_stacked_on_url())
 
146
    except (errors.UnstackableBranchFormat, errors.UnstackableRepositoryFormat,
 
147
        errors.NotStacked):
 
148
        pass
 
149
    return locs
 
150
 
 
151
 
 
152
def _show_related_info(branch, outfile):
 
153
    """Show parent and push location of branch."""
 
154
    locs = _gather_related_branches(branch)
 
155
    if len(locs.locs) > 0:
 
156
        outfile.write('\n')
 
157
        outfile.write('Related branches:\n')
 
158
        outfile.writelines(locs.get_lines())
 
159
 
 
160
 
 
161
def _show_format_info(control=None, repository=None, branch=None,
 
162
                      working=None, outfile=None):
 
163
    """Show known formats for control, working, branch and repository."""
 
164
    outfile.write('\n')
 
165
    outfile.write('Format:\n')
 
166
    if control:
 
167
        outfile.write('       control: %s\n' %
 
168
            control._format.get_format_description())
 
169
    if working:
 
170
        outfile.write('  working tree: %s\n' %
 
171
            working._format.get_format_description())
 
172
    if branch:
 
173
        outfile.write('        branch: %s\n' %
 
174
            branch._format.get_format_description())
 
175
    if repository:
 
176
        outfile.write('    repository: %s\n' %
 
177
            repository._format.get_format_description())
 
178
 
 
179
 
 
180
def _show_locking_info(repository, branch=None, working=None, outfile=None):
 
181
    """Show locking status of working, branch and repository."""
 
182
    if (repository.get_physical_lock_status() or
 
183
        (branch and branch.get_physical_lock_status()) or
 
184
        (working and working.get_physical_lock_status())):
 
185
        outfile.write('\n')
 
186
        outfile.write('Lock status:\n')
 
187
        if working:
 
188
            if working.get_physical_lock_status():
 
189
                status = 'locked'
 
190
            else:
 
191
                status = 'unlocked'
 
192
            outfile.write('  working tree: %s\n' % status)
 
193
        if branch:
 
194
            if branch.get_physical_lock_status():
 
195
                status = 'locked'
 
196
            else:
 
197
                status = 'unlocked'
 
198
            outfile.write('        branch: %s\n' % status)
 
199
        if repository:
 
200
            if repository.get_physical_lock_status():
 
201
                status = 'locked'
 
202
            else:
 
203
                status = 'unlocked'
 
204
            outfile.write('    repository: %s\n' % status)
 
205
 
 
206
 
 
207
def _show_missing_revisions_branch(branch, outfile):
 
208
    """Show missing master revisions in branch."""
 
209
    # Try with inaccessible branch ?
 
210
    master = branch.get_master_branch()
 
211
    if master:
 
212
        local_extra, remote_extra = find_unmerged(branch, master)
 
213
        if remote_extra:
 
214
            outfile.write('\n')
 
215
            outfile.write(('Branch is out of date: missing %d '
 
216
                'revision%s.\n') % (len(remote_extra),
 
217
                plural(len(remote_extra))))
 
218
 
 
219
 
 
220
def _show_missing_revisions_working(working, outfile):
 
221
    """Show missing revisions in working tree."""
 
222
    branch = working.branch
 
223
    basis = working.basis_tree()
 
224
    work_inv = working.inventory
 
225
    branch_revno, branch_last_revision = branch.last_revision_info()
 
226
    try:
 
227
        tree_last_id = working.get_parent_ids()[0]
 
228
    except IndexError:
 
229
        tree_last_id = None
 
230
 
 
231
    if branch_revno and tree_last_id != branch_last_revision:
 
232
        tree_last_revno = branch.revision_id_to_revno(tree_last_id)
 
233
        missing_count = branch_revno - tree_last_revno
 
234
        outfile.write('\n')
 
235
        outfile.write(('Working tree is out of date: missing %d '
 
236
            'revision%s.\n') % (missing_count, plural(missing_count)))
 
237
 
 
238
 
 
239
def _show_working_stats(working, outfile):
 
240
    """Show statistics about a working tree."""
 
241
    basis = working.basis_tree()
 
242
    work_inv = working.inventory
 
243
    delta = working.changes_from(basis, want_unchanged=True)
 
244
 
 
245
    outfile.write('\n')
 
246
    outfile.write('In the working tree:\n')
 
247
    outfile.write('  %8s unchanged\n' % len(delta.unchanged))
 
248
    outfile.write('  %8d modified\n' % len(delta.modified))
 
249
    outfile.write('  %8d added\n' % len(delta.added))
 
250
    outfile.write('  %8d removed\n' % len(delta.removed))
 
251
    outfile.write('  %8d renamed\n' % len(delta.renamed))
60
252
 
61
253
    ignore_cnt = unknown_cnt = 0
62
254
    for path in working.extras():
64
256
            ignore_cnt += 1
65
257
        else:
66
258
            unknown_cnt += 1
67
 
 
68
 
    print '  %8d unknown' % unknown_cnt
69
 
    print '  %8d ignored' % ignore_cnt
 
259
    outfile.write('  %8d unknown\n' % unknown_cnt)
 
260
    outfile.write('  %8d ignored\n' % ignore_cnt)
70
261
 
71
262
    dir_cnt = 0
72
263
    for file_id in work_inv:
73
 
        if work_inv.get_file_kind(file_id) == 'directory':
 
264
        if (work_inv.get_file_kind(file_id) == 'directory' and
 
265
            not work_inv.is_root(file_id)):
74
266
            dir_cnt += 1
75
 
    print '  %8d versioned %s' \
76
 
          % (dir_cnt,
77
 
             plural(dir_cnt, 'subdirectory', 'subdirectories'))
78
 
 
79
 
    print
80
 
    print 'branch history:'
81
 
    history = b.revision_history()
82
 
    revno = len(history)
83
 
    print '  %8d revision%s' % (revno, plural(revno))
84
 
    committers = {}
85
 
    for rev in history:
86
 
        committers[b.get_revision(rev).committer] = True
87
 
    print '  %8d committer%s' % (len(committers), plural(len(committers)))
88
 
    if revno > 0:
89
 
        firstrev = b.get_revision(history[0])
90
 
        age = int((time.time() - firstrev.timestamp) / 3600 / 24)
91
 
        print '  %8d day%s old' % (age, plural(age))
92
 
        print '   first revision: %s' % format_date(firstrev.timestamp,
93
 
                                                    firstrev.timezone)
94
 
 
95
 
        lastrev = b.get_revision(history[-1])
96
 
        print '  latest revision: %s' % format_date(lastrev.timestamp,
97
 
                                                    lastrev.timezone)
98
 
 
99
 
    print
100
 
    print 'text store:'
101
 
    c, t = b.text_store.total_size()
102
 
    print '  %8d file texts' % c
103
 
    print '  %8d kB' % (t/1024)
104
 
 
105
 
    print
106
 
    print 'revision store:'
107
 
    c, t = b.revision_store.total_size()
108
 
    print '  %8d revisions' % c
109
 
    print '  %8d kB' % (t/1024)
110
 
 
111
 
 
112
 
    print
113
 
    print 'inventory store:'
114
 
    c, t = b.inventory_store.total_size()
115
 
    print '  %8d inventories' % c
116
 
    print '  %8d kB' % (t/1024)
117
 
 
 
267
    outfile.write('  %8d versioned %s\n' % (dir_cnt,
 
268
        plural(dir_cnt, 'subdirectory', 'subdirectories')))
 
269
 
 
270
 
 
271
def _show_branch_stats(branch, verbose, outfile):
 
272
    """Show statistics about a branch."""
 
273
    revno, head = branch.last_revision_info()
 
274
    outfile.write('\n')
 
275
    outfile.write('Branch history:\n')
 
276
    outfile.write('  %8d revision%s\n' % (revno, plural(revno)))
 
277
    stats = branch.repository.gather_stats(head, committers=verbose)
 
278
    if verbose:
 
279
        committers = stats['committers']
 
280
        outfile.write('  %8d committer%s\n' % (committers,
 
281
            plural(committers)))
 
282
    if revno:
 
283
        timestamp, timezone = stats['firstrev']
 
284
        age = int((time.time() - timestamp) / 3600 / 24)
 
285
        outfile.write('  %8d day%s old\n' % (age, plural(age)))
 
286
        outfile.write('   first revision: %s\n' %
 
287
            osutils.format_date(timestamp, timezone))
 
288
        timestamp, timezone = stats['latestrev']
 
289
        outfile.write('  latest revision: %s\n' %
 
290
            osutils.format_date(timestamp, timezone))
 
291
    return stats
 
292
 
 
293
 
 
294
def _show_repository_info(repository, outfile):
 
295
    """Show settings of a repository."""
 
296
    if repository.make_working_trees():
 
297
        outfile.write('\n')
 
298
        outfile.write('Create working tree for new branches inside '
 
299
            'the repository.\n')
 
300
 
 
301
 
 
302
def _show_repository_stats(stats, outfile):
 
303
    """Show statistics about a repository."""
 
304
    if 'revisions' in stats or 'size' in stats:
 
305
        outfile.write('\n')
 
306
        outfile.write('Repository:\n')
 
307
    if 'revisions' in stats:
 
308
        revisions = stats['revisions']
 
309
        outfile.write('  %8d revision%s\n' % (revisions, plural(revisions)))
 
310
    if 'size' in stats:
 
311
        outfile.write('  %8d KiB\n' % (stats['size']/1024))
 
312
 
 
313
 
 
314
def show_bzrdir_info(a_bzrdir, verbose=False, outfile=None):
 
315
    """Output to stdout the 'info' for a_bzrdir."""
 
316
    if outfile is None:
 
317
        outfile = sys.stdout
 
318
    try:
 
319
        tree = a_bzrdir.open_workingtree(
 
320
            recommend_upgrade=False)
 
321
    except (NoWorkingTree, NotLocalUrl):
 
322
        tree = None
 
323
        try:
 
324
            branch = a_bzrdir.open_branch()
 
325
        except NotBranchError:
 
326
            branch = None
 
327
            try:
 
328
                repository = a_bzrdir.open_repository()
 
329
            except NoRepositoryPresent:
 
330
                # Return silently; cmd_info already returned NotBranchError
 
331
                # if no bzrdir could be opened.
 
332
                return
 
333
            else:
 
334
                lockable = repository
 
335
        else:
 
336
            repository = branch.repository
 
337
            lockable = branch
 
338
    else:
 
339
        branch = tree.branch
 
340
        repository = branch.repository
 
341
        lockable = tree
 
342
 
 
343
    lockable.lock_read()
 
344
    try:
 
345
        show_component_info(a_bzrdir, repository, branch, tree, verbose,
 
346
                            outfile)
 
347
    finally:
 
348
        lockable.unlock()
 
349
 
 
350
 
 
351
def show_component_info(control, repository, branch=None, working=None,
 
352
    verbose=1, outfile=None):
 
353
    """Write info about all bzrdir components to stdout"""
 
354
    if outfile is None:
 
355
        outfile = sys.stdout
 
356
    if verbose is False:
 
357
        verbose = 1
 
358
    if verbose is True:
 
359
        verbose = 2
 
360
    layout = describe_layout(repository, branch, working)
 
361
    format = describe_format(control, repository, branch, working)
 
362
    outfile.write("%s (format: %s)\n" % (layout, format))
 
363
    _show_location_info(gather_location_info(repository, branch, working),
 
364
                        outfile)
 
365
    if branch is not None:
 
366
        _show_related_info(branch, outfile)
 
367
    if verbose == 0:
 
368
        return
 
369
    _show_format_info(control, repository, branch, working, outfile)
 
370
    _show_locking_info(repository, branch, working, outfile)
 
371
    if branch is not None:
 
372
        _show_missing_revisions_branch(branch, outfile)
 
373
    if working is not None:
 
374
        _show_missing_revisions_working(working, outfile)
 
375
        _show_working_stats(working, outfile)
 
376
    elif branch is not None:
 
377
        _show_missing_revisions_branch(branch, outfile)
 
378
    if branch is not None:
 
379
        show_committers = verbose >= 2
 
380
        stats = _show_branch_stats(branch, show_committers, outfile)
 
381
    else:
 
382
        stats = repository.gather_stats()
 
383
    if branch is None and working is None:
 
384
        _show_repository_info(repository, outfile)
 
385
    _show_repository_stats(stats, outfile)
 
386
 
 
387
 
 
388
def describe_layout(repository=None, branch=None, tree=None):
 
389
    """Convert a control directory layout into a user-understandable term
 
390
 
 
391
    Common outputs include "Standalone tree", "Repository branch" and
 
392
    "Checkout".  Uncommon outputs include "Unshared repository with trees"
 
393
    and "Empty control directory"
 
394
    """
 
395
    if repository is None:
 
396
        return 'Empty control directory'
 
397
    if branch is None and tree is None:
 
398
        if repository.is_shared():
 
399
            phrase = 'Shared repository'
 
400
        else:
 
401
            phrase = 'Unshared repository'
 
402
        if repository.make_working_trees():
 
403
            phrase += ' with trees'
 
404
        return phrase
 
405
    else:
 
406
        if repository.is_shared():
 
407
            independence = "Repository "
 
408
        else:
 
409
            independence = "Standalone "
 
410
        if tree is not None:
 
411
            phrase = "tree"
 
412
        else:
 
413
            phrase = "branch"
 
414
        if branch is None and tree is not None:
 
415
            phrase = "branchless tree"
 
416
        else:
 
417
            if (tree is not None and tree.bzrdir.root_transport.base !=
 
418
                branch.bzrdir.root_transport.base):
 
419
                independence = ''
 
420
                phrase = "Lightweight checkout"
 
421
            elif branch.get_bound_location() is not None:
 
422
                if independence == 'Standalone ':
 
423
                    independence = ''
 
424
                if tree is None:
 
425
                    phrase = "Bound branch"
 
426
                else:
 
427
                    phrase = "Checkout"
 
428
        if independence != "":
 
429
            phrase = phrase.lower()
 
430
        return "%s%s" % (independence, phrase)
 
431
 
 
432
 
 
433
def describe_format(control, repository, branch, tree):
 
434
    """Determine the format of an existing control directory
 
435
 
 
436
    Several candidates may be found.  If so, the names are returned as a
 
437
    single string, separated by ' or '.
 
438
 
 
439
    If no matching candidate is found, "unnamed" is returned.
 
440
    """
 
441
    candidates  = []
 
442
    if (branch is not None and tree is not None and
 
443
        branch.bzrdir.root_transport.base !=
 
444
        tree.bzrdir.root_transport.base):
 
445
        branch = None
 
446
        repository = None
 
447
    non_aliases = set(bzrdir.format_registry.keys())
 
448
    non_aliases.difference_update(bzrdir.format_registry.aliases())
 
449
    for key in non_aliases:
 
450
        format = bzrdir.format_registry.make_bzrdir(key)
 
451
        if isinstance(format, bzrdir.BzrDirMetaFormat1):
 
452
            if (tree and format.workingtree_format !=
 
453
                tree._format):
 
454
                continue
 
455
            if (branch and format.get_branch_format() !=
 
456
                branch._format):
 
457
                continue
 
458
            if (repository and format.repository_format !=
 
459
                repository._format):
 
460
                continue
 
461
        if format.__class__ is not control._format.__class__:
 
462
            continue
 
463
        candidates.append(key)
 
464
    if len(candidates) == 0:
 
465
        return 'unnamed'
 
466
    candidates.sort()
 
467
    new_candidates = [c for c in candidates if not
 
468
        bzrdir.format_registry.get_info(c).hidden]
 
469
    if len(new_candidates) > 0:
 
470
        # If there are any non-hidden formats that match, only return those to
 
471
        # avoid listing hidden formats except when only a hidden format will
 
472
        # do.
 
473
        candidates = new_candidates
 
474
    return ' or '.join(candidates)