From d152ed13ec0b99e1969da3ee37fec6e41b2172e2 Mon Sep 17 00:00:00 2001
From: Dafydd Harries <daf@rhydd.org>
Date: Mon, 18 Mar 2013 01:53:29 -0400
Subject: make scripts import-safe

---
 load_descriptions.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

(limited to 'load_descriptions.py')

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()
 
-- 
cgit v1.2.3