From: Tony Mack <tmack@paris.CS.Princeton.EDU>
Date: Tue, 2 Apr 2013 13:54:07 +0000 (-0400)
Subject: use command line options
X-Git-Tag: 1.0~290
X-Git-Url: http://git.onelab.eu/?a=commitdiff_plain;h=bd0cd32e856974218e3663f5e7dd71a788f2ea71;p=plstackapi.git

use command line options
---

diff --git a/plstackapi/importer/plclassic/importer.py b/plstackapi/importer/plclassic/importer.py
index 599e548..8315431 100644
--- a/plstackapi/importer/plclassic/importer.py
+++ b/plstackapi/importer/plclassic/importer.py
@@ -1,5 +1,8 @@
 import os
 os.environ.setdefault("DJANGO_SETTINGS_MODULE", "plstackapi.planetstack.settings")
+import sys
+from optparse import OptionParser
+from getpass import getpass
 import xmlrpclib
 from plstackapi.importer.plclassic.site_importer import SiteImporter
 from plstackapi.importer.plclassic.user_importer import UserImporter
@@ -45,4 +48,24 @@ class Importer:
 
 
 if __name__ == '__main__':
-    Importer().run()
+    parser = OptionParser()
+        
+    parser.add_option("-u", "--username", dest="username",
+                        help="PLC username with which to authenticate")
+    parser.add_option("", "--url", dest="url",
+                        help="PLC url to contact")
+
+    (config, args) = parser.parse_args()
+    if len(sys.argv) == 1:
+        parser.print_help()
+        sys.exit(1)
+
+    password = None
+    try:
+        password = getpass()
+    except (EOFError, KeyboardInterrupt):
+        print
+        sys.exit(0)
+
+    
+    Importer(config.username, password, config.url).run()