Allow alternate input file names to be given on command line.

svn path=/scripts/testbed-rootcert.py; revision=3529

1 files changed, 4 insertions, 4 deletions

diff --git a/scripts/testbed-rootcert.py b/scripts/testbed-rootcert.py
index 6c75da9c..e4cf304f 100644
--- a/scripts/testbed-rootcert.py
+++ b/scripts/testbed-rootcert.py
@@ -24,8 +24,8 @@ PERFORMANCE OF THIS SOFTWARE.
 
 import csv, rpki.myrpki, sys
 
-if len(sys.argv) != 2:
-  raise RuntimeError, "Usage: %s [holder]" % sys.argv[0]
+if len(sys.argv) not in (2, 4):
+  sys.exit("Usage: %s holder [asns.csv prefixes.csv]" % sys.argv[0])
 
 print '''\
 [req]
@@ -51,7 +51,7 @@ sbgp-ipAddrBlock                = critical,@rfc3997_addrs
 ''' % { "holder" : sys.argv[1].lower(),
         "HOLDER" : sys.argv[1].upper() }
 
-for i, asn in enumerate(asn for handle, asn in rpki.myrpki.csv_reader("asns.csv", columns = 2)):
+for i, asn in enumerate(asn for handle, asn in rpki.myrpki.csv_reader(sys.argv[2] if len(sys.argv) > 2 else "asns.csv", columns = 2)):
   print "AS.%d = %s" % (i, asn)
 
 print '''\
@@ -60,6 +60,6 @@ print '''\
 
 '''
 
-for i, prefix in enumerate(prefix for handle, prefix in rpki.myrpki.csv_reader("prefixes.csv", columns = 2)):
+for i, prefix in enumerate(prefix for handle, prefix in rpki.myrpki.csv_reader(sys.argv[3] if len(sys.argv) > 2 else "prefixes.csv", columns = 2)):
   v = 6 if ":" in prefix else 4
   print "IPv%d.%d = %s" % (v, i, prefix)