removed another bunch of references to geni

[sfa.git] / sfa / client / sfadump.py

diff --git a/sfa/client/sfadump.py b/sfa/client/sfadump.py
new file mode 100644 (file)
index 0000000..6496ca6
--- /dev/null
+++ b/sfa/client/sfadump.py
@@ -0,0 +1,94 @@
+#! /usr/bin/env python
+# $Id$
+# $URL$
+from __future__ import with_statement
+
+import sys
+import os, os.path
+import tempfile
+import xmlrpclib
+from types import StringTypes, ListType
+from optparse import OptionParser
+
+from sfa.trust.certificate import Certificate
+from sfa.trust.credential import Credential
+from sfa.util.record import SfaRecord
+from sfa.util.rspec import RSpec
+
+def determine_sfa_filekind(fn):
+    cert = Certificate(filename = fn)
+
+    data = cert.get_data()
+    if data:
+        dict = xmlrpclib.loads(data)[0][0]
+    else:
+        dict = {}
+
+    if "gidCaller" in dict:
+        return "credential"
+
+    if "uuid" in dict:
+        return "gid"
+
+    return "unknown"
+
+def save_gid(gid):
+   hrn = gid.get_hrn()
+   lastpart = hrn.split(".")[-1]
+   filename = lastpart + ".gid"
+
+   if os.path.exists(filename):
+       print filename, ": already exists... skipping"
+       return
+
+   print filename, ": extracting gid of", hrn
+
+   gid.save_to_file(filename, save_parents = True)
+
+def extract_gids(cred, extract_parents):
+   gidCaller = cred.get_gid_caller()
+   if gidCaller:
+       save_gid(gidCaller)
+
+   gidObject = cred.get_gid_object()
+   if gidObject and ((gidCaller == None) or (gidCaller.get_hrn() != gidObject.get_hrn())):
+       save_gid(gidObject)
+
+   if extract_parents:
+       parent = cred.get_parent()
+       if parent:
+           extract_gids(parent, extract_parents)
+
+def create_parser():
+   # Generate command line parser
+   parser = OptionParser(usage="%prog [options] filename")
+
+   parser.add_option("-e", "--extractgids", action="store_true", dest="extract_gids", default=False, help="Extract GIDs from credentials")
+   parser.add_option("-p", "--dumpparents", action="store_true", dest="dump_parents", default=False, help="Show parents")
+
+   return parser
+
+def main():
+   parser = create_parser()
+   (options, args) = parser.parse_args()
+
+   if len(args) <= 0:
+        print "No filename given. Use -h for help."
+        return -1
+
+   filename = args[0]
+   kind = determine_sfa_filekind(filename)
+
+   if kind=="credential":
+       cred = Credential(filename = filename)
+       cred.dump(dump_parents = options.dump_parents)
+       if options.extract_gids:
+           extract_gids(cred, extract_parents = options.dump_parents)
+   elif kind=="gid":
+       gid = Gid(filename = filename)
+       gid.dump(dump_parents = options.dump_parents)
+   else:
+       print "unknown filekind", kind
+
+if __name__=="__main__":
+   main()