Update image-minimizer for py3
Also switch to using argparse instead of optparse.
This commit is contained in:
parent
ca0bb2681e
commit
cb7dadb35f
@ -1,8 +1,8 @@
|
|||||||
#!/usr/bin/python2
|
#!/usr/bin/python3
|
||||||
#
|
#
|
||||||
# image-minimizer: removes files and packages on the filesystem
|
# image-minimizer: removes files and packages on the filesystem
|
||||||
#
|
#
|
||||||
# Copyright 2007-2014 Red Hat, Inc.
|
# Copyright 2007-2015 Red Hat, Inc.
|
||||||
#
|
#
|
||||||
# This program is free software; you can redistribute it and/or modify
|
# This program is free software; you can redistribute it and/or modify
|
||||||
# it under the terms of the GNU General Public License as published by
|
# it under the terms of the GNU General Public License as published by
|
||||||
@ -17,7 +17,7 @@
|
|||||||
# along with this program; if not, write to the Free Software
|
# along with this program; if not, write to the Free Software
|
||||||
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
||||||
import glob
|
import glob
|
||||||
import optparse
|
import argparse
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
import rpm
|
import rpm
|
||||||
@ -53,7 +53,7 @@ class ImageMinimizer:
|
|||||||
def add_pattern(self, files, pattern):
|
def add_pattern(self, files, pattern):
|
||||||
globs = glob.glob(pattern)
|
globs = glob.glob(pattern)
|
||||||
if self.verbose and len(globs) == 0:
|
if self.verbose and len(globs) == 0:
|
||||||
print "%s file not found" % pattern
|
print("%s file not found" % pattern)
|
||||||
for g in globs:
|
for g in globs:
|
||||||
if os.path.isdir(g):
|
if os.path.isdir(g):
|
||||||
self.add_directory(files, g)
|
self.add_directory(files, g)
|
||||||
@ -63,7 +63,7 @@ class ImageMinimizer:
|
|||||||
def add_pattern_rpm(self, rpms, pattern):
|
def add_pattern_rpm(self, rpms, pattern):
|
||||||
if self.ts is None:
|
if self.ts is None:
|
||||||
if self.prefix is None:
|
if self.prefix is None:
|
||||||
raise Exception ('Must specify installation root for droprpm/keeprpm')
|
raise Exception('Must specify installation root for droprpm/keeprpm')
|
||||||
self.ts = rpm.TransactionSet(self.prefix)
|
self.ts = rpm.TransactionSet(self.prefix)
|
||||||
mi = self.ts.dbMatch()
|
mi = self.ts.dbMatch()
|
||||||
mi.pattern('name', rpm.RPMMIRE_GLOB, pattern)
|
mi.pattern('name', rpm.RPMMIRE_GLOB, pattern)
|
||||||
@ -72,7 +72,7 @@ class ImageMinimizer:
|
|||||||
not_found = False
|
not_found = False
|
||||||
rpms.add(hdr['name'])
|
rpms.add(hdr['name'])
|
||||||
if self.verbose and not_found:
|
if self.verbose and not_found:
|
||||||
print "%s package not found" % pattern
|
print("%s package not found" % pattern)
|
||||||
|
|
||||||
# Parses each line in the ifle
|
# Parses each line in the ifle
|
||||||
def parse_line(self, line):
|
def parse_line(self, line):
|
||||||
@ -107,7 +107,7 @@ class ImageMinimizer:
|
|||||||
elif command == 'droprpm':
|
elif command == 'droprpm':
|
||||||
self.add_pattern_rpm(self.drops_rpm, pattern)
|
self.add_pattern_rpm(self.drops_rpm, pattern)
|
||||||
else:
|
else:
|
||||||
raise Exception ('Unknown Command: ' + command)
|
raise Exception('Unknown Command: ' + command)
|
||||||
|
|
||||||
def remove(self):
|
def remove(self):
|
||||||
for tag in sorted(self.drops, reverse=True):
|
for tag in sorted(self.drops, reverse=True):
|
||||||
@ -116,34 +116,34 @@ class ImageMinimizer:
|
|||||||
self.visited.add(tag)
|
self.visited.add(tag)
|
||||||
else:
|
else:
|
||||||
if self.dryrun:
|
if self.dryrun:
|
||||||
print 'rm ' + tag
|
print('rm ' + tag)
|
||||||
else:
|
else:
|
||||||
if self.verbose:
|
if self.verbose:
|
||||||
print 'rm ' + tag
|
print('rm ' + tag)
|
||||||
os.remove(tag)
|
os.remove(tag)
|
||||||
|
|
||||||
#remove all empty directory. Every 8k counts!
|
#remove all empty directory. Every 8k counts!
|
||||||
for d in sorted(self.visited, reverse=True):
|
for d in sorted(self.visited, reverse=True):
|
||||||
if len(os.listdir(d)) == 0:
|
if len(os.listdir(d)) == 0:
|
||||||
if self.dryrun:
|
if self.dryrun:
|
||||||
print 'rm -rf ' + d
|
print('rm -rf ' + d)
|
||||||
else:
|
else:
|
||||||
if self.verbose:
|
if self.verbose:
|
||||||
print 'rm -rf ' + d
|
print('rm -rf ' + d)
|
||||||
os.rmdir(d)
|
os.rmdir(d)
|
||||||
|
|
||||||
def remove_rpm(self):
|
def remove_rpm(self):
|
||||||
|
|
||||||
def runCallback(reason, amount, total, key, client_data):
|
def runCallback(reason, amount, total, key, client_data):
|
||||||
if self.verbose and reason == rpm.RPMCALLBACK_UNINST_STOP:
|
if self.verbose and reason == rpm.RPMCALLBACK_UNINST_STOP:
|
||||||
print key, "erased"
|
print(key, "erased")
|
||||||
|
|
||||||
if len(self.drops_rpm) == 0:
|
if len(self.drops_rpm) == 0:
|
||||||
return
|
return
|
||||||
|
|
||||||
for pkg in self.drops_rpm:
|
for pkg in self.drops_rpm:
|
||||||
if self.dryrun:
|
if self.dryrun:
|
||||||
print "erasing ", pkg
|
print("erasing ", pkg)
|
||||||
else:
|
else:
|
||||||
self.ts.addErase(pkg)
|
self.ts.addErase(pkg)
|
||||||
if not self.dryrun:
|
if not self.dryrun:
|
||||||
@ -158,40 +158,35 @@ class ImageMinimizer:
|
|||||||
|
|
||||||
|
|
||||||
def parse_options():
|
def parse_options():
|
||||||
usage = "usage: %prog [options] filename"
|
parser = argparse.ArgumentParser(description="Image Minimizer")
|
||||||
parser = optparse.OptionParser(usage=usage)
|
|
||||||
|
|
||||||
parser.set_defaults(root=os.environ.get('INSTALL_ROOT', '/mnt/sysimage/'), dry_run=False)
|
parser.set_defaults(root=os.environ.get('INSTALL_ROOT', '/mnt/sysimage/'), dry_run=False)
|
||||||
|
|
||||||
parser.add_option("-i", "--installroot", type="string", dest="root",
|
parser.add_argument("-i", "--installroot", metavar="STRING", dest="root",
|
||||||
help="Root path to prepend to all file patterns and installation root for RPM "
|
help="Root path to prepend to all file patterns and installation root for RPM "
|
||||||
"operations. Defaults to INSTALL_ROOT or /mnt/sysimage/")
|
"operations. Defaults to INSTALL_ROOT or /mnt/sysimage/")
|
||||||
|
|
||||||
parser.add_option("--dryrun", action="store_true", dest="dryrun",
|
parser.add_argument("--dryrun", metavar="BOOL", action="store_true", dest="dryrun",
|
||||||
help="If set, no filesystem changes are made.")
|
help="If set, no filesystem changes are made.")
|
||||||
|
|
||||||
parser.add_option("-v", "--verbose", action="store_true", dest="verbose",
|
parser.add_argument("-v", "--verbose", metavar="BOOL", action="store_true", dest="verbose",
|
||||||
help="Display every action as it is performed.")
|
help="Display every action as it is performed.")
|
||||||
|
|
||||||
(options, args) = parser.parse_args()
|
parser.add_argument("filename", metavar="STRING", help="Filename to process")
|
||||||
if len(args) == 0:
|
|
||||||
parser.print_help()
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
return (options, args)
|
return parser.parse_args()
|
||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
try:
|
try:
|
||||||
(options, args) = parse_options()
|
args = parse_options()
|
||||||
filename = args[0]
|
minimizer = ImageMinimizer(args.filename, args.root, args.dryrun,
|
||||||
minimizer = ImageMinimizer(filename, options.root, options.dryrun,
|
args.verbose)
|
||||||
options.verbose)
|
|
||||||
minimizer.filter()
|
minimizer.filter()
|
||||||
except SystemExit, e:
|
except SystemExit as e:
|
||||||
sys.exit(e.code)
|
sys.exit(e.code)
|
||||||
except KeyboardInterrupt, e:
|
except KeyboardInterrupt as e:
|
||||||
print >> sys.stderr, "Aborted at user request"
|
print("Aborted at user request", file=sys.stderr)
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
|
Loading…
Reference in New Issue
Block a user