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 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179
|
#! /usr/bin/python3
# vim: et ts=4 sw=4
# Copyright © 2010-2012 Piotr Ożarowski <piotr@debian.org>
# Copyright © 2025 Akexander Sulfrian <alexander@sulfrian.net>
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
import logging
import optparse
import sys
from glob import glob
from os import environ, remove, rmdir
from os.path import dirname, basename, join, splitext
sys.path.insert(1, '/usr/share/python3/')
from debpython import files as dpf
from debpython.interpreter import Interpreter
from debpython.version import SUPPORTED, getver, vrepr
# initialize script
logging.basicConfig(format='%(levelname).1s: %(module)s:%(lineno)d: '
'%(message)s')
log = logging.getLogger(__name__)
# Workaround for bug in debpython
def from_directory(dname, extensions):
if isinstance(dname, (list, tuple)):
for item in dname:
for fn in dpf.from_directory(item, extensions):
yield fn
for fn in dpf.from_directory(item, extensions):
yield fn
def get_magic_tag_to_remove(version):
"""Returns magic tag or True if all of them should be removed."""
i = Interpreter('python')
map_ = {}
for v in SUPPORTED:
try:
map_[v] = i.magic_tag(v)
except Exception:
log.debug('magic tag for %s not recognized', vrepr(v), exc_info=True)
if version not in map_:
try:
map_[version] = i.magic_tag(version)
except Exception as e:
log.error('cannot find magic tag for Python %s: %s', vrepr(version), e)
exit(4)
tag = map_[version]
# skip shared tags
for v, t in map_.items():
if v == version:
continue
if t == tag:
log.info('magic tag(s) used by python%s. Nothing to remove.',
vrepr(v))
exit(0)
log.debug('magic tags to remove: %s', tag)
return tag
def destroyer(magic_tag=None):
"""Remove .enamlc files associated to received .enaml file from __enamlcache__
directory. If the resulting directory is empty, then the directory is also
removed.
:param magic_tag:
* If None, removes all associated .enamlc files from __enamlcache__
directory.
* Otherwise removes given magic tag from __enamlcache__ directory.
:type magic_tag: None or str
"""
pattern = ''
if magic_tag is not None:
pattern = '%s-' % magic_tag
def find_files_to_remove(enamlfile):
directory = join(dirname(enamlfile), '__enamlcache__')
fname = splitext(basename(enamlfile))[0]
glob_pattern = "%s.enaml-%s*.enamlc" % (fname, pattern)
for fn in glob(glob_pattern, root_dir=directory):
yield join(directory, fn)
def myremove(fname):
remove(fname)
directory = dirname(fname)
# remove __enamlcache__ directory if it's empty
if directory.endswith('__enamlcache__'):
try:
rmdir(directory)
except Exception:
pass
counter = 0
try:
while True:
enamlfile = (yield)
for filename in find_files_to_remove(enamlfile):
try:
myremove(filename)
counter += 1
except (IOError, OSError) as e:
log.error('cannot remove %s', filename)
log.debug(e)
except GeneratorExit:
log.info("removed files: %s", counter)
def main():
usage = '%prog [-V VERSION] [-p PACKAGE] [DIR_OR_FILE]'
parser = optparse.OptionParser(usage)
parser.add_option('-v', '--verbose', action='store_true', dest='verbose',
help='turn verbose mode on')
parser.add_option('-q', '--quiet', action='store_false', dest='verbose',
default=False, help='be quiet')
parser.add_option('-p', '--package',
help='specify Debian package name to clean')
parser.add_option('-V', dest='version',
help='specify Python version to clean')
(options, args) = parser.parse_args()
if options.verbose or environ.get('ENAMLCLEAN_DEBUG') == '1':
log.setLevel(logging.DEBUG)
log.debug('argv: %s', sys.argv)
log.debug('options: %s', options)
log.debug('args: %s', args)
else:
log.setLevel(logging.WARNING)
if options.version:
magic_tag = get_magic_tag_to_remove(getver(options.version))
d = destroyer(magic_tag)
else:
d = destroyer() # remove everything
next(d) # initialize coroutine
if not options.package and not args:
parser.print_usage()
exit(1)
if options.package:
log.info('cleaning package %s', options.package)
pfiles = set(dpf.from_package(options.package,
extensions=('.enaml',)))
if args:
log.info('cleaning directories: %s', args)
files = set(from_directory(args, extensions=('.enaml',)))
if options.package:
files = files & pfiles
else:
files = pfiles
for filename in files:
d.send(filename)
if __name__ == '__main__':
main()
|