Use command line

svn path=/scripts/x509-dot.py; revision=1841

1 files changed, 4 insertions, 6 deletions

diff --git a/scripts/x509-dot.py b/scripts/x509-dot.py
index a7af9dca..ff5a2f15 100644
--- a/scripts/x509-dot.py
+++ b/scripts/x509-dot.py
@@ -56,13 +56,11 @@ class x509(object):
 
     print
 
-# Eventually this should use sys.argv[1:]
-
-#certs = [x509(filename) for filename in glob.glob("../rpkid/biz-certs/*.cer")]
-
 certs = []
-for dirpath, dirnames, filenames in os.walk("../rpkid/testbed.dir/publication"):
-  certs += [x509(dirpath + "/" + filename) for filename in filenames if filename.endswith(".cer")]
+
+for topdir in sys.argv[1:] or ["."]:
+  for dirpath, dirnames, filenames in os.walk(topdir):
+    certs += [x509(dirpath + "/" + filename) for filename in filenames if filename.endswith(".cer")]
 
 for i in xrange(len(certs)):
   certs[i].set_node("cert_%d" % i)