1 files changed, 74 insertions, 0 deletions
diff --git a/scripts/sys-gentoo-weight.py b/scripts/sys-gentoo-weight.py
new file mode 100755
index 0000000..98e023d
--- /dev/null
+++ b/scripts/sys-gentoo-weight.py
@@ -0,0 +1,74 @@
+#!/usr/bin/python
+
+import os, sys, output
+
+spinpos = 0
+spinner = "/-\\|/-\\|/-\\|/-\\|\\-/|\\-/|\\-/|\\-/|"
+
+def update_spinner():
+	global spinner, spinpos
+	spinpos = (spinpos+1) % len(spinner)
+	sys.stdout.write("\b\b "+spinner[spinpos])
+	sys.stdout.flush()
+
+def formatSize(x):
+	x = int(x)
+	if x < 1024:
+		return str(x) + ' B'
+	if x < 1024 * 1024:
+		return '%.2f KB' % (float(x) / 1024)
+	return '%.2f MB' % (float(x) / (1024.0 * 1024.0))
+
+packages = {}
+total_size = 0
+
+print output.blue('Counting...  ')
+
+counter = 0
+for category in sorted(os.listdir('/var/db/pkg/')):
+
+	print output.green(' *'),category,' ',
+	update_spinner()
+
+	for package in sorted(os.listdir('/var/db/pkg/' + category)):
+
+		fh = file('/var/db/pkg/' + category + '/' + package + '/CONTENTS')
+		content = fh.readlines()
+		fh.close()
+
+		size = 0
+		for entity in content:
+			counter += 1
+			if counter % 10 == 0:
+				update_spinner()
+			data = entity.strip().split(' ')
+			if data[0] != 'obj':
+				continue
+			try:
+				size += os.path.getsize(data[1])
+			except:
+				if '--report-missing' in sys.argv:
+					print '\n', output.red('   *'), 'File missing:', data[1],
+				pass
+
+		total_size += size
+		packages[category + '/' + package] = size
+
+	print '\b   '
+
+print
+
+if '--sort-size' in sys.argv:
+	pkgs = packages.items()
+	pkgs2 = []
+	for pkg in pkgs:
+		pkgs2 += [[ pkg[1], pkg[0] ]]
+	pkgs = pkgs2
+	del(pkgs2)
+	for package in sorted(pkgs):
+		print output.green(' * ') + output.bold(package[1]) + ' - ' + formatSize(package[0])
+else:
+	for package in sorted(packages.keys()):
+		print output.green(' * ') + output.bold(package) + ' - ' + formatSize(packages[package])
+
+print output.green('Total size:'), formatSize(total_size)