← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/clean_tree.py

(gz) Backslash escape selftest output when printing to non-unicode consoles
 (Martin [gz])

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/clean_tree.py
1
 
# Copyright (C) 2005 Canonical Ltd
 
1
# Copyright (C) 2009, 2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
18
18
import errno
19
19
import os
20
20
import shutil
21
 
import sys
22
21
 
23
 
from bzrlib.osutils import has_symlinks, isdir
 
22
from bzrlib import (
 
23
    bzrdir,
 
24
    errors,
 
25
    ui,
 
26
    )
 
27
from bzrlib.osutils import isdir
24
28
from bzrlib.trace import note
25
29
from bzrlib.workingtree import WorkingTree
26
30
 
53
57
    try:
54
58
        deletables = list(iter_deletables(tree, unknown=unknown,
55
59
            ignored=ignored, detritus=detritus))
 
60
        deletables = _filter_out_nested_bzrdirs(deletables)
56
61
        if len(deletables) == 0:
57
62
            note('Nothing to delete.')
58
63
            return 0
59
64
        if not no_prompt:
60
65
            for path, subp in deletables:
 
66
                # FIXME using print is very bad idea
 
67
                # clean_tree should accept to_file argument to write the output
61
68
                print subp
62
69
            val = raw_input('Are you sure you wish to delete these [y/N]?')
63
70
            if val.lower() not in ('y', 'yes'):
68
75
        tree.unlock()
69
76
 
70
77
 
 
78
def _filter_out_nested_bzrdirs(deletables):
 
79
    result = []
 
80
    for path, subp in deletables:
 
81
        # bzr won't recurse into unknowns/ignored directories by default
 
82
        # so we don't pay a penalty for checking subdirs of path for nested
 
83
        # bzrdir.
 
84
        # That said we won't detect the branch in the subdir of non-branch
 
85
        # directory and therefore delete it. (worth to FIXME?)
 
86
        if isdir(path):
 
87
            try:
 
88
                bzrdir.BzrDir.open(path)
 
89
            except errors.NotBranchError:
 
90
                result.append((path,subp))
 
91
            else:
 
92
                # TODO may be we need to notify user about skipped directories?
 
93
                pass
 
94
        else:
 
95
            result.append((path,subp))
 
96
    return result
 
97
 
 
98
 
71
99
def delete_items(deletables, dry_run=False):
72
100
    """Delete files in the deletables iterable"""
 
101
    def onerror(function, path, excinfo):
 
102
        """Show warning for errors seen by rmtree.
 
103
        """
 
104
        # Handle only permission error while removing files.
 
105
        # Other errors are re-raised.
 
106
        if function is not os.remove or excinfo[1].errno != errno.EACCES:
 
107
            raise
 
108
        ui.ui_factory.show_warning('unable to remove %s' % path)
73
109
    has_deleted = False
74
110
    for path, subp in deletables:
75
111
        if not has_deleted:
76
112
            note("deleting paths:")
77
113
            has_deleted = True
78
 
        note('  ' + subp)
79
114
        if not dry_run:
80
115
            if isdir(path):
81
 
                shutil.rmtree(path)
 
116
                shutil.rmtree(path, onerror=onerror)
82
117
            else:
83
 
                os.unlink(path)
 
118
                try:
 
119
                    os.unlink(path)
 
120
                    note('  ' + subp)
 
121
                except OSError, e:
 
122
                    # We handle only permission error here
 
123
                    if e.errno != errno.EACCES:
 
124
                        raise e
 
125
                    ui.ui_factory.show_warning(
 
126
                        'unable to remove "%s": %s.' % (path, e.strerror))
 
127
        else:
 
128
            note('  ' + subp)
84
129
    if not has_deleted:
85
130
        note("No files deleted.")