X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=plcsh;h=ad10824dcb1e1ae7ac1590b4661f5c52846b8a24;hb=02d525f6d1b8163f51fd3796e13a94f3f71defba;hp=7c4b69af0aa6390da940033aa8e84d4d91b05f43;hpb=4126c5644a3d4d8eb6fa43d289194934f5c9b534;p=plcapi.git

diff --git a/plcsh b/plcsh
index 7c4b69a..ad10824 100755
--- a/plcsh
+++ b/plcsh
@@ -5,12 +5,14 @@
 # Mark Huang <mlhuang@cs.princeton.edu>
 # Copyright (C) 2005 The Trustees of Princeton University
 #
-# $Id: plcsh,v 1.2 2007/01/11 05:28:49 mlhuang Exp $
+# $Id$
 #
 
 import os
 import sys
+from socket import gethostname
 from optparse import OptionParser
+from getpass import getpass
 from traceback import print_exc
 
 sys.path.append(os.path.dirname(os.path.realpath(sys.argv[0])))
@@ -20,7 +22,9 @@ parser = OptionParser(add_help_option = False)
 parser.add_option("-f", "--config", help = "PLC configuration file")
 parser.add_option("-h", "--url", help = "API URL")
 parser.add_option("-c", "--cacert", help = "API SSL certificate")
+parser.add_option("-k", "--insecure", help = "Do not check SSL certificate")
 parser.add_option("-m", "--method", help = "API authentication method")
+parser.add_option("-s", "--session", help = "API session key")
 parser.add_option("-u", "--user", help = "API user name")
 parser.add_option("-p", "--password", help = "API password")
 parser.add_option("-r", "--role", help = "API role")
@@ -31,7 +35,7 @@ parser.add_option("--help", action = "help", help = "show this help message and
 # If user is specified but password is not
 if options.user is not None and options.password is None:
     try:
-        options.password = getpass.getpass()
+        options.password = getpass()
     except (EOFError, KeyboardInterrupt):
         print
         sys.exit(0)
@@ -43,7 +47,8 @@ try:
                   config = options.config,
                   url = options.url, xmlrpc = options.xmlrpc, cacert = options.cacert,
                   method = options.method, role = options.role,
-                  user = options.user, password = options.password)
+                  user = options.user, password = options.password,
+                  session = options.session)
     # Register a few more globals for backward compatibility
     auth = shell.auth
     api = shell.api
@@ -54,10 +59,15 @@ except Exception, err:
     parser.print_help()
     sys.exit(1)
 
-# If called by a script
+# If called by a script 
 if len(sys.argv) > 1 and os.path.exists(sys.argv[1]):
     # Pop us off the argument stack
     sys.argv.pop(0)
+    script = sys.argv[0]
+    
+    # Add of script to sys.path 
+    path = os.path.dirname(os.path.abspath(script))
+    sys.path.append(path)
     execfile(sys.argv[0])
 
 # Otherwise, run an interactive shell environment
@@ -68,6 +78,11 @@ else:
     elif shell.auth['AuthMethod'] == "anonymous":
         prompt = "[anonymous]"
         print "Connected anonymously"
+    elif shell.auth['AuthMethod'] == "session":
+        # XXX No way to tell node and user sessions apart from the
+        # client point of view.
+        prompt = "[%s]" % gethostname()
+        print "%s connected using session authentication" % gethostname()
     else:
         prompt = "[%s]" % shell.auth['Username']
         print "%s connected using %s authentication" % \