File: clean.py

package info (click to toggle)
taskcoach 1.4.1-4
  • links: PTS, VCS
  • area: main
  • in suites: jessie, jessie-kfreebsd
  • size: 32,496 kB
  • ctags: 17,810
  • sloc: python: 72,170; makefile: 254; ansic: 120; xml: 29; sh: 16
file content (55 lines) | stat: -rw-r--r-- 2,214 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
'''
Task Coach - Your friendly task manager
Copyright (C) 2004-2014 Task Coach developers <developers@taskcoach.org>

Task Coach is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

Task Coach is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
'''

import os, fnmatch
from distutils.command.clean import clean as BaseCleanCommand
from distutils import log


class clean(BaseCleanCommand, object):
    user_options = BaseCleanCommand.user_options + \
        [('really-clean', 'r', 'remove even more files')]
    boolean_options = BaseCleanCommand.boolean_options + ['really-clean']

    def initialize_options(self):
        super(clean, self).initialize_options()
        # pylint: disable=W0201
        self.really_clean = False
        self.cleaning_patterns = ['*.pyc']
        self.really_clean_patterns = ['*.bak']
                    
    def finalize_options(self):
        super(clean, self).finalize_options()
        if self.really_clean:
            self.cleaning_patterns.extend(self.really_clean_patterns)
                    
    def run(self):
        super(clean, self).run()
        if not self.verbose:
            log.info("recursively removing '" + "', '".join(self.cleaning_patterns) + "'")
        for root, _, files in os.walk('.'):
            for pattern in self.cleaning_patterns:
                for filename in fnmatch.filter(files, pattern):
                    filename = os.path.join(root, filename)
                    try:
                        if not self.dry_run:
                            os.unlink(filename)
                        if self.verbose:
                            log.info("removing '%s'"%filename.strip('.\\'))
                    except IOError:
                        pass