diff TrustStoreGalaxyImport.py @ 5:159d2159e745

Add list files.

--- a/TrustStoreGalaxyImport.py	Thu Dec 12 13:07:45 2013 +1100
+++ b/TrustStoreGalaxyImport.py	Thu Dec 12 16:17:43 2013 +1100
@@ -2,6 +2,14 @@
 import shutil
 from py_ts import TrustStoreClient, ts_utils
 
+def printNice(elem, f, depth):
+    try:
+        f.write('\t'*depth + elem.name + " (" + str(len(elem.fragments)) + " parts)\n")
+    except AttributeError:
+        f.write('\t'*depth + elem.name + "\n")
+        for child in elem.children:
+            printNice(child, f, depth+1)
+
 if __name__ == '__main__':
 
     kms_url = sys.argv[1]
@@ -12,8 +20,13 @@
     client_secret = sys.argv[6]
     storename = sys.argv[7]
     path = sys.argv[8]
-    filename = sys.argv[9]
-    outputFile = sys.argv[10]
+    filename = ""
+    outputFile = ""
+    if len(sys.argv) > 10:
+        filename = sys.argv[9]
+        outputFile = sys.argv[10]
+    else:
+        outputFile = sys.argv[9]
 
     config = TrustStoreClient.Config(ims_url, kms_url, client_key, client_secret)
     ts = TrustStoreClient.TrustStoreClient(headless=True, config=config)
@@ -37,13 +50,22 @@
                     sys.exit(3)
             else:
                 location = root
-            downloadMe = ts_utils.ts_utils.recurseToChildNamed(location, filename)
-            if downloadMe:
-                download = ts.getFile(store, downloadMe)
-                shutil.copy(download, outputFile)
+            if filename and filename != "":
+                downloadMe = ts_utils.ts_utils.recurseToChildNamed(location, filename)
+                if downloadMe:
+                    download = ts.getFile(store, downloadMe)
+                    shutil.copy(download, outputFile)
+                else:
+                    print "File not found"
+                    sys.exit(4)
             else:
-                print "File not found"
-                sys.exit(4)
+                with open(outputFile, 'w+') as f:
+                    try:
+                        for child in root.children:
+                            printNice(child, f, 0)
+                    except AttributeError as e:
+                        print e
+                        print root
     if not found:
         print "Store not found"
         sys.exit(2)