diff options
author | Dafydd Harries <daf@rhydd.org> | 2013-03-18 01:53:29 -0400 |
---|---|---|
committer | Dafydd Harries <daf@rhydd.org> | 2013-03-18 01:53:29 -0400 |
commit | d152ed13ec0b99e1969da3ee37fec6e41b2172e2 (patch) | |
tree | 4d7e03ad7807bd7c2c52a90ec13a503feb3aa487 /load_descriptions.py | |
parent | 97e0bc2e76ee25633944de3d62eaa00380fcb6c4 (diff) |
make scripts import-safe
Diffstat (limited to 'load_descriptions.py')
-rw-r--r-- | load_descriptions.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/load_descriptions.py b/load_descriptions.py index 36d6257..14d0ee4 100644 --- a/load_descriptions.py +++ b/load_descriptions.py @@ -4,9 +4,10 @@ import sys import debian.deb822 import pandas as pd -descs = debian.deb822.Packages.iter_paragraphs(sys.stdin) -df = pd.DataFrame([dict(p) for p in descs]) -store = pd.HDFStore('pkg.h5') -store['descriptions'] = df -store.close() +if __name__ == '__main__': + descs = debian.deb822.Packages.iter_paragraphs(sys.stdin) + df = pd.DataFrame([dict(p) for p in descs]) + store = pd.HDFStore('pkg.h5') + store['descriptions'] = df + store.close() |