aboutsummaryrefslogtreecommitdiff
path: root/support
diff options
context:
space:
mode:
authorGravatar Yann E. MORIN <yann.morin.1998@free.fr>2019-08-17 19:18:23 +0200
committerGravatar Arnout Vandecappelle (Essensium/Mind) <arnout@mind.be>2019-08-26 22:08:01 +0200
commita2d20ca6132556af87837e25f7c3d58a1c2f091e (patch)
treee4539ba4cb0dac771c7027242ee6fdc58374bd75 /support
parentcecaf7001f0c9fa2089800dbd19541d70130fd18 (diff)
downloadbuildroot-a2d20ca6132556af87837e25f7c3d58a1c2f091e.tar.gz
buildroot-a2d20ca6132556af87837e25f7c3d58a1c2f091e.tar.bz2
support/graph-size: introduce main()
It is nicer overall to have a main() function, like all our other scripts tend to have too. Signed-off-by: Yann E. MORIN <yann.morin.1998@free.fr> Cc: Thomas De Schampheleire <thomas.de_schampheleire@nokia.com> Signed-off-by: Arnout Vandecappelle (Essensium/Mind) <arnout@mind.be>
Diffstat (limited to 'support')
-rwxr-xr-xsupport/scripts/size-stats53
1 files changed, 29 insertions, 24 deletions
diff --git a/support/scripts/size-stats b/support/scripts/size-stats
index 8dfa391e24..bd10fbfb68 100755
--- a/support/scripts/size-stats
+++ b/support/scripts/size-stats
@@ -216,27 +216,32 @@ def gen_packages_csv(pkgsizes, outputf):
wr.writerow([pkg, size, "%.1f" % (float(size) / total * 100)])
-parser = argparse.ArgumentParser(description='Draw size statistics graphs')
-
-parser.add_argument("--builddir", '-i', metavar="BUILDDIR", required=True,
- help="Buildroot output directory")
-parser.add_argument("--graph", '-g', metavar="GRAPH",
- help="Graph output file (.pdf or .png extension)")
-parser.add_argument("--file-size-csv", '-f', metavar="FILE_SIZE_CSV",
- help="CSV output file with file size statistics")
-parser.add_argument("--package-size-csv", '-p', metavar="PKG_SIZE_CSV",
- help="CSV output file with package size statistics")
-args = parser.parse_args()
-
-# Find out which package installed what files
-pkgdict = build_package_dict(args.builddir)
-
-# Collect the size installed by each package
-pkgsize = build_package_size(pkgdict, args.builddir)
-
-if args.graph:
- draw_graph(pkgsize, args.graph)
-if args.file_size_csv:
- gen_files_csv(pkgdict, pkgsize, args.file_size_csv)
-if args.package_size_csv:
- gen_packages_csv(pkgsize, args.package_size_csv)
+def main():
+ parser = argparse.ArgumentParser(description='Draw size statistics graphs')
+
+ parser.add_argument("--builddir", '-i', metavar="BUILDDIR", required=True,
+ help="Buildroot output directory")
+ parser.add_argument("--graph", '-g', metavar="GRAPH",
+ help="Graph output file (.pdf or .png extension)")
+ parser.add_argument("--file-size-csv", '-f', metavar="FILE_SIZE_CSV",
+ help="CSV output file with file size statistics")
+ parser.add_argument("--package-size-csv", '-p', metavar="PKG_SIZE_CSV",
+ help="CSV output file with package size statistics")
+ args = parser.parse_args()
+
+ # Find out which package installed what files
+ pkgdict = build_package_dict(args.builddir)
+
+ # Collect the size installed by each package
+ pkgsize = build_package_size(pkgdict, args.builddir)
+
+ if args.graph:
+ draw_graph(pkgsize, args.graph)
+ if args.file_size_csv:
+ gen_files_csv(pkgdict, pkgsize, args.file_size_csv)
+ if args.package_size_csv:
+ gen_packages_csv(pkgsize, args.package_size_csv)
+
+
+if __name__ == "__main__":
+ main()