/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/clean_tree.py

  • Committer: Jelmer Vernooij
  • Date: 2018-02-18 21:42:57 UTC
  • mto: This revision was merged to the branch mainline in revision 6859.
  • Revision ID: jelmer@jelmer.uk-20180218214257-jpevutp1wa30tz3v
Update TODO to reference Breezy, not Bazaar.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/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
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
 
17
from __future__ import absolute_import
17
18
 
 
19
import errno
18
20
import os
19
21
import shutil
20
22
 
21
 
from bzrlib import bzrdir, errors
22
 
from bzrlib.osutils import isdir
23
 
from bzrlib.trace import note
24
 
from bzrlib.workingtree import WorkingTree
25
 
 
 
23
from . import (
 
24
    controldir,
 
25
    errors,
 
26
    ui,
 
27
    )
 
28
from .osutils import isdir
 
29
from .trace import note
 
30
from .workingtree import WorkingTree
 
31
from .i18n import gettext
26
32
 
27
33
def is_detritus(subp):
28
34
    """Return True if the supplied path is detritus, False otherwise"""
48
54
               dry_run=False, no_prompt=False):
49
55
    """Remove files in the specified classes from the tree"""
50
56
    tree = WorkingTree.open_containing(directory)[0]
51
 
    tree.lock_read()
52
 
    try:
 
57
    with tree.lock_read():
53
58
        deletables = list(iter_deletables(tree, unknown=unknown,
54
59
            ignored=ignored, detritus=detritus))
55
 
        deletables = _filter_out_nested_bzrdirs(deletables)
 
60
        deletables = _filter_out_nested_controldirs(deletables)
56
61
        if len(deletables) == 0:
57
 
            note('Nothing to delete.')
 
62
            note(gettext('Nothing to delete.'))
58
63
            return 0
59
64
        if not no_prompt:
60
65
            for path, subp in deletables:
61
 
                # FIXME using print is very bad idea
62
 
                # clean_tree should accept to_file argument to write the output
63
 
                print subp
64
 
            val = raw_input('Are you sure you wish to delete these [y/N]?')
65
 
            if val.lower() not in ('y', 'yes'):
66
 
                print 'Canceled'
 
66
                ui.ui_factory.note(subp)
 
67
            prompt = gettext('Are you sure you wish to delete these')
 
68
            if not ui.ui_factory.get_boolean(prompt):
 
69
                ui.ui_factory.note(gettext('Canceled'))
67
70
                return 0
68
71
        delete_items(deletables, dry_run=dry_run)
69
 
    finally:
70
 
        tree.unlock()
71
 
 
72
 
 
73
 
def _filter_out_nested_bzrdirs(deletables):
 
72
 
 
73
 
 
74
def _filter_out_nested_controldirs(deletables):
74
75
    result = []
75
76
    for path, subp in deletables:
76
77
        # bzr won't recurse into unknowns/ignored directories by default
77
78
        # so we don't pay a penalty for checking subdirs of path for nested
78
 
        # bzrdir.
 
79
        # control dir.
79
80
        # That said we won't detect the branch in the subdir of non-branch
80
81
        # directory and therefore delete it. (worth to FIXME?)
81
82
        if isdir(path):
82
83
            try:
83
 
                bzrdir.BzrDir.open(path)
 
84
                controldir.ControlDir.open(path)
84
85
            except errors.NotBranchError:
85
 
                result.append((path,subp))
 
86
                result.append((path, subp))
86
87
            else:
87
88
                # TODO may be we need to notify user about skipped directories?
88
89
                pass
89
90
        else:
90
 
            result.append((path,subp))
 
91
            result.append((path, subp))
91
92
    return result
92
93
 
93
94
 
94
95
def delete_items(deletables, dry_run=False):
95
96
    """Delete files in the deletables iterable"""
 
97
    def onerror(function, path, excinfo):
 
98
        """Show warning for errors seen by rmtree.
 
99
        """
 
100
        # Handle only permission error while removing files.
 
101
        # Other errors are re-raised.
 
102
        if function is not os.remove or excinfo[1].errno != errno.EACCES:
 
103
            raise
 
104
        ui.ui_factory.show_warning(gettext('unable to remove %s') % path)
96
105
    has_deleted = False
97
106
    for path, subp in deletables:
98
107
        if not has_deleted:
99
 
            note("deleting paths:")
 
108
            note(gettext("deleting paths:"))
100
109
            has_deleted = True
101
 
        note('  ' + subp)
102
110
        if not dry_run:
103
111
            if isdir(path):
104
 
                shutil.rmtree(path)
 
112
                shutil.rmtree(path, onerror=onerror)
105
113
            else:
106
 
                os.unlink(path)
 
114
                try:
 
115
                    os.unlink(path)
 
116
                    note('  ' + subp)
 
117
                except OSError as e:
 
118
                    # We handle only permission error here
 
119
                    if e.errno != errno.EACCES:
 
120
                        raise e
 
121
                    ui.ui_factory.show_warning(gettext(
 
122
                        'unable to remove "{0}": {1}.').format(
 
123
                                                    path, e.strerror))
 
124
        else:
 
125
            note('  ' + subp)
107
126
    if not has_deleted:
108
 
        note("No files deleted.")
 
127
        note(gettext("No files deleted."))