~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/clean_tree.py

• browse files at revision 5200.3.5
• compare with another revision
• download diff
• download tarball
• view history from revision 5200.3.5

• reverse the comparison (5200.3.5 to 2592.3.231)
• stop comparing with revision 2592.3.231

Show diffs side-by-side

added

removed

bzrlib/clean_tree.py

 

1

# Copyright (C) 2005 Canonical Ltd

 

2

#

 

3

# This program is free software; you can redistribute it and/or modify

 

4

# it under the terms of the GNU General Public License as published by

 

5

# the Free Software Foundation; either version 2 of the License, or

 

6

# (at your option) any later version.

 

7

#

 

8

# This program is distributed in the hope that it will be useful,

 

9

# but WITHOUT ANY WARRANTY; without even the implied warranty of

 

10

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 

11

# GNU General Public License for more details.

 

12

#

 

13

# You should have received a copy of the GNU General Public License

 

14

# along with this program; if not, write to the Free Software

 

15

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

 

16

 

 

17

 

 

18

import os

 

19

import shutil

 

20

 

 

21

from bzrlib.osutils import isdir

 

22

from bzrlib.trace import note

 

23

from bzrlib.workingtree import WorkingTree

 

24

 

 

25

 

 

26

def is_detritus(subp):

 

27

    """Return True if the supplied path is detritus, False otherwise"""

 

28

    return subp.endswith('.THIS') or subp.endswith('.BASE') or\

 

29

        subp.endswith('.OTHER') or subp.endswith('~') or subp.endswith('.tmp')

 

30

 

 

31

 

 

32

def iter_deletables(tree, unknown=False, ignored=False, detritus=False):

 

33

    """Iterate through files that may be deleted"""

 

34

    for subp in tree.extras():

 

35

        if detritus and is_detritus(subp):

 

36

            yield tree.abspath(subp), subp

 

37

            continue

 

38

        if tree.is_ignored(subp):

 

39

            if ignored:

 

40

                yield tree.abspath(subp), subp

 

41

        else:

 

42

            if unknown:

 

43

                yield tree.abspath(subp), subp

 

44

 

 

45

 

 

46

def clean_tree(directory, unknown=False, ignored=False, detritus=False,

 

47

               dry_run=False, no_prompt=False):

 

48

    """Remove files in the specified classes from the tree"""

 

49

    tree = WorkingTree.open_containing(directory)[0]

 

50

    tree.lock_read()

 

51

    try:

 

52

        deletables = list(iter_deletables(tree, unknown=unknown,

 

53

            ignored=ignored, detritus=detritus))

 

54

        if len(deletables) == 0:

 

55

            note('Nothing to delete.')

 

56

            return 0

 

57

        if not no_prompt:

 

58

            for path, subp in deletables:

 

59

                print subp

 

60

            val = raw_input('Are you sure you wish to delete these [y/N]?')

 

61

            if val.lower() not in ('y', 'yes'):

 

62

                print 'Canceled'

 

63

                return 0

 

64

        delete_items(deletables, dry_run=dry_run)

 

65

    finally:

 

66

        tree.unlock()

 

67

 

 

68

 

 

69

def delete_items(deletables, dry_run=False):

 

70

    """Delete files in the deletables iterable"""

 

71

    has_deleted = False

 

72

    for path, subp in deletables:

 

73

        if not has_deleted:

 

74

            note("deleting paths:")

 

75

            has_deleted = True

 

76

        note('  ' + subp)

 

77

        if not dry_run:

 

78

            if isdir(path):

 

79

                shutil.rmtree(path)

 

80

            else:

 

81

                os.unlink(path)

 

82

    if not has_deleted:

 

83

        note("No files deleted.")

• Older »

Loggerhead is a web-based interface for Breezy
Version: 2.0.1