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 --- load_packages.py | 25 +++++++++++++------------ 1 file changed, 13 insertions(+), 12 deletions(-) (limited to 'load_packages.py') diff --git a/load_packages.py b/load_packages.py index c02eabd..423b227 100644 --- a/load_packages.py +++ b/load_packages.py @@ -4,19 +4,20 @@ import sys import debian.deb822 import pandas as pd -packages = debian.deb822.Packages.iter_paragraphs(sys.stdin) -df = pd.DataFrame([dict(p) for p in packages]) -store = pd.HDFStore('pkg.h5') +if __name__ == '__main__': + packages = debian.deb822.Packages.iter_paragraphs(sys.stdin) + df = pd.DataFrame([dict(p) for p in packages]) + store = pd.HDFStore('pkg.h5') -# No 'Source' field means that it has the same value as the 'Package' field. -# Set this explicitly. -nosrc = df['Source'].isnull() -df['Source'][nosrc] = df[nosrc]['Package'] -assert sum(pd.isnull(df['Source'])) == 0 + # No 'Source' field means that it has the same value as the 'Package' + # field. Set this explicitly. + nosrc = df['Source'].isnull() + df['Source'][nosrc] = df[nosrc]['Package'] + assert sum(pd.isnull(df['Source'])) == 0 -print df + print df -store = pd.HDFStore('pkg.h5') -store['packages'] = df -store.close() + store = pd.HDFStore('pkg.h5') + store['packages'] = df + store.close() -- cgit v1.2.3