From bd0cd32e856974218e3663f5e7dd71a788f2ea71 Mon Sep 17 00:00:00 2001 From: Tony Mack Date: Tue, 2 Apr 2013 09:54:07 -0400 Subject: [PATCH] use command line options --- plstackapi/importer/plclassic/importer.py | 25 ++++++++++++++++++++++- 1 file changed, 24 insertions(+), 1 deletion(-) 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() -- 2.47.0