added command-line record editer, a few SFI demo scripts

[sfa.git] / cmdline / genicli.py

diff --git a/cmdline/genicli.py b/cmdline/genicli.py
index 4101b5f..40c9ff9 100644 (file)
--- a/cmdline/genicli.py
+++ b/cmdline/genicli.py
@@ -9,7 +9,7 @@ from geniticket import *
 
 long_opts = ["keyfile=", "help", "outfile=", "credfile=", "ticketfile=",
              "username=", "email=", "ip=", "dns=", "dump_parents", "server=",
-             "filter="]
+             "filter=", "short"]
 
 # default command line options
 username = "client"
@@ -23,6 +23,7 @@ cert_file = None
 out_file = None
 ticket_file = None
 
+short = False
 ip = None
 dns = None
 email = None
@@ -31,6 +32,8 @@ gid_pkey_fn = None
 gid_fn = None
 filter = None
 
+dump_fn = None
+
 dump_parents = False
 
 leaf_name = None
@@ -54,10 +57,13 @@ def showhelp():
    print "    --dump_parents   ... dump parents"
    print "    --server         ... geni server (registry/component) to connect to"
    print "    --filter <type>  ... filter the results of a list operation (user | slice | node ...)"
+   print "    --short          ... list records in short format (name only)"
    print "commands:"
    print "    resolve <hrn>"
-   print "    dumpCredential"
+   print "    dumpCredential <filename>"
+   print "    dumpGid <filename>"
    print "    getCredential <type> <hrn>"
+   print "    list <hrn>"
    print "    start <hrn>"
    print "    createKey <filename>"
    print "    createGid <hrn> <uuid|None> <pubkey_fn>"
@@ -78,6 +84,8 @@ def process_options():
    global dump_parents
    global server_url
    global filter
+   global short
+   global dump_fn
 
    (options, args) = getopt.getopt(sys.argv[1:], '', long_opts)
    for opt in options:
@@ -111,6 +119,8 @@ def process_options():
            server_url = val
        elif name == "--filter":
            filter = val
+       elif name == "--short":
+           short = True
 
    if not args:
        print "no operation specified"
@@ -131,6 +141,13 @@ def process_options():
        type = args[1]
        hrn = args[2]
 
+   elif opname == "list":
+       if len(args) < 2:
+           print "syntax: list <hrn>"
+           sys.exit(-1)
+       hrn = args[1]
+
+
    elif opname == "createGid":
        if len(args) < 4:
            print "syntax: createGid <hrn> <uuid|None> <pubkey_fn>"
@@ -166,6 +183,12 @@ def process_options():
            sys.exit(-1)
        hrn = args[1]
 
+   elif opname == "dumpGid":
+       if len(args) < 2:
+           print "syntax: dumpGid <filename>"
+           sys.exit(-1)
+       dump_fn = args[1]
+
    leaf_name = get_leaf(username)
 
    if cert_file == None:
@@ -195,6 +218,10 @@ def get_authority(x):
 def dumpCredential():
    pass
 
+def dumpGid():
+   gid = GID(filename = dump_fn)
+   gid.dump()
+
 # creates a self-signed certificate and private key
 def createKey():
    k = Keypair(create=True)
@@ -240,7 +267,7 @@ def main():
 
    # if the operation is not a local operation, then create a geniclient to
    # talk to the server
-   if (opname != "dumpCredential") and (opname != "help") and (opname != "createKey"):
+   if (opname != "dumpCredential") and (opname != "help") and (opname != "createKey") and (opname != "dumpGid"):
        if not os.path.exists(key_file):
            print "key file", key_file, "does not exist"
            sys.exit(-1)
@@ -256,7 +283,7 @@ def main():
 
    # if a cred_file was specified, then load the credential
    if (cred_file=="None") or (opname == "help") or (opname == "createKey") or \
-      (opname == "redeemTicket"):
+      (opname == "redeemTicket") or (opname == "dumpCredential") or (opname == "dumpGid"):
       cred = None
    else:
       cred = Credential(filename = cred_file)
@@ -264,6 +291,9 @@ def main():
    if opname == "dumpCredential":
       dumpCredential()
 
+   elif opname == "dumpGid":
+      dumpGid()
+
    elif opname == "help":
       showhelp()
 
@@ -290,13 +320,16 @@ def main():
           print "NO RESULT"
 
    elif (opname == "list"):
-      result = client.list(cred)
+      result = client.list(cred, hrn)
       if result:
           if filter:
               result = [r for r in result if r.type==filter]
+          print "RESULT:"
           for record in result:
-              print "RESULT:"
-              record.dump(dump_parents=dump_parents)
+              if short:
+                  print "  ", record.get_name()
+              else:
+                  record.dump(dump_parents=dump_parents)
       else:
           print "NO RESULT"
 
@@ -336,20 +369,6 @@ def main():
 
    elif (opname == "remove"):
        client.remove(cred, type, hrn)
-#       record_list = client.resolve(cred, hrn)
-#       if not record_list:
-#           print "no records match hrn"
-#
-#       matching_records = []
-#       for record in record_list:
-#           if record.get_type() == type:
-#               matching_records.append(record)
-#
-#       if not matching_records:
-#           print "records match hrn, but no records match type"
-#
-#       for record in matching_records:
-#           client.remove(cred, record)
 
    elif (opname == "update"):
        record_list = client.resolve(cred, hrn)