From d152ed13ec0b99e1969da3ee37fec6e41b2172e2 Mon Sep 17 00:00:00 2001 From: Dafydd Harries Date: Mon, 18 Mar 2013 01:53:29 -0400 Subject: make scripts import-safe --- license_summary.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) (limited to 'license_summary.py') diff --git a/license_summary.py b/license_summary.py index 7e7899c..e5edaba 100644 --- a/license_summary.py +++ b/license_summary.py @@ -2,9 +2,6 @@ import pandas as pd -store = pd.HDFStore('cp.h5') -cpf = store['cp_files'] - def histogram(values): hist = {} @@ -14,8 +11,12 @@ def histogram(values): return hist -licenses = histogram(cpf['_license']) +if __name__ == '__main__': + store = pd.HDFStore('cp.h5') + cpf = store['cp_files'] + + licenses = list(histogram(cpf['_license'])) -for (k, v) in sorted(licenses.iteritems(), key=lambda x: x[1], reverse=True): - print '%-40s %6d' % (k.encode('utf8'), v) + for (k, v) in sorted(licenses, key=lambda x: x[1], reverse=True): + print '%-40s %6d' % (k.encode('utf8'), v) -- cgit v1.2.3