Mercurial > repos > cathywise > truststore_import
view TrustStoreGalaxyImport.py @ 4:32d9b3343955
Toooool upload.
author | Catherine Wise <catherine.wise@csiro.au> |
---|---|
date | Thu, 12 Dec 2013 13:07:45 +1100 |
parents | |
children | 159d2159e745 |
line wrap: on
line source
import sys import shutil from py_ts import TrustStoreClient, ts_utils if __name__ == '__main__': kms_url = sys.argv[1] ims_url = sys.argv[2] username = sys.argv[3] password = sys.argv[4] client_key = sys.argv[5] client_secret = sys.argv[6] storename = sys.argv[7] path = sys.argv[8] filename = sys.argv[9] outputFile = sys.argv[10] config = TrustStoreClient.Config(ims_url, kms_url, client_key, client_secret) ts = TrustStoreClient.TrustStoreClient(headless=True, config=config) try: ts.authenticate(username, password) except TrustStoreClient.TrustStoreClientAuthenticationException as e: print e sys.exit(5) ts.getPrivateKey('privkey.pem') listing = ts.listStores() found = False for store in listing: if store.friendly_name == storename: found = True root = ts.listDirectory(store) location = None if path != "/": location = ts_utils.ts_utils.dirAtPath(root, path) if not location: print "Path not found" 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) else: print "File not found" sys.exit(4) if not found: print "Store not found" sys.exit(2)