annotate TrustStoreGalaxyBrowse.py @ 14:589f9250e30f

Fix son reading.
author Catherine Wise <catherine.wise@csiro.au>
date Wed, 13 May 2015 10:54:07 +1000
parents 11344d545e88
children 7b3039ccf50f
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
1 """TrustStore downloaded for Galaxy."""
0
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
2 from __future__ import division, absolute_import, print_function, unicode_literals
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
3 import sys
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
4 import shutil
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
5 import gzip
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
6 import tempfile
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
7 import os
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
8 import json
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
9 import operator
0
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
10 from py_ts import TrustStoreClient, utils
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
11 from galaxy.datatypes.checkers import util
0
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
12
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
13 CLIENT_KEY = "desktop"
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
14 CLIENT_SECRET = "cpU92F1PT7VOCANjSknuCDp4DrubmujoBaF6b0miz8OpKNokEbGMHCaSFK5/lISbBmaaGVCgeADI2A39F3Hkeg=="
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
15 CHUNK_SIZE = 2**20 # 1Mb
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
16 SAFE_CHARS = '.-()[]0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ '
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
17
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
18 def print_nice(elem, f, depth):
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
19 """Print the file name."""
0
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
20 try:
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
21 f.write('\t'*depth + elem.name + " (" + str(len(elem.fragments)) + " parts)\n")
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
22 except AttributeError:
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
23 f.write('\t'*depth + elem.name + "\n")
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
24 for child in elem.children:
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
25 print_nice(child, f, depth+1)
0
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
26
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
27 def check_gzip(file_path):
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
28 """Check if file is gziped."""
0
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
29 try:
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
30 temp = open(file_path, "U")
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
31 magic_check = temp.read(2)
0
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
32 temp.close()
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
33 if magic_check != util.gzip_magic:
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
34 return (False, False)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
35 except Exception:
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
36 return (False, False)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
37 return (True, True)
0
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
38
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
39 def ungzip(download, outputFile):
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
40 """Uncompress file."""
0
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
41 is_gzipped, is_valid = check_gzip(download)
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
42
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
43 if is_gzipped and not is_valid:
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
44 print("File is compressed (gzip) but not valid.")
0
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
45 sys.exit(4)
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
46 elif is_gzipped and is_valid:
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
47 # We need to uncompress the temp_name file, but BAM files must
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
48 # remain compressed in the BGZF format
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
49 file_handle, uncompressed = tempfile.mkstemp(prefix='data_id_upload_gunzip_', dir=os.path.dirname(outputFile), text=False )
0
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
50 gzipped_file = gzip.GzipFile(download, 'rb')
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
51 while 1:
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
52 try:
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
53 chunk = gzipped_file.read(CHUNK_SIZE)
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
54 except IOError:
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
55 os.close(file_handle)
0
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
56 os.remove(uncompressed)
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
57 print('Problem decompressing gzipped data %s %s' % (download, outputFile))
0
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
58 sys.exit(4)
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
59 if not chunk:
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
60 break
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
61 os.write(file_handle, chunk)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
62 os.close(file_handle)
0
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
63 gzipped_file.close()
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
64
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
65 shutil.copy(uncompressed, outputFile)
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
66 try:
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
67 os.remove(uncompressed)
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
68 os.remove(download)
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
69 except OSError:
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
70 pass
0
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
71 else:
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
72 shutil.copy(download, outputFile)
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
73
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
74 def construct_multi_filename(id, name, file_type):
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
75 """ Implementation of *Number of Output datasets cannot be determined until
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
76 tool run* from documentation_.
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
77 .. _documentation: http://wiki.galaxyproject.org/Admin/Tools/Multiple%20Output%20Files
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
78 From https://github.com/mdshw5/galaxy-json-data-source/blob/master/json_data_source.py
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
79 """
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
80 filename = "%s_%s_%s_%s_%s" % ('primary', id, name, 'visible', file_type)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
81 return filename
0
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
82
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
83 def metadata_to_json(dataset_id, filename, name, extesion, ds_type='dataset', primary=False):
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
84 """ Return line separated JSON
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
85 From https://github.com/mdshw5/galaxy-json-data-source/blob/master/json_data_source.py
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
86 """
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
87 meta_dict = dict(type=ds_type,
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
88 ext=extesion,
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
89 filename=filename,
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
90 name=name,
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
91 metadata={})
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
92 if primary:
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
93 meta_dict['base_dataset_id'] = dataset_id
0
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
94 else:
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
95 meta_dict['dataset_id'] = dataset_id
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
96 return "%s\n" % json.dumps(meta_dict)
0
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
97
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
98 def main():
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
99 properties_file = sys.argv[1]
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
100 json_params = None
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
101 metadata_path = None
14
589f9250e30f Fix son reading.
Catherine Wise <catherine.wise@csiro.au>
parents: 13
diff changeset
102 all_params = None
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
103 with open(properties_file, 'r') as file_:
13
Catherine Wise <catherine.wise@csiro.au>
parents: 10
diff changeset
104 settings = file_.read()
Catherine Wise <catherine.wise@csiro.au>
parents: 10
diff changeset
105 print(settings)
Catherine Wise <catherine.wise@csiro.au>
parents: 10
diff changeset
106 all_params = json.loads(settings)
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
107 json_params = all_params.get("param_dict")
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
108 metadata_path = all_params["job_config"]["TOOL_PROVIDED_JOB_METADATA_FILE"]
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
109
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
110 output_filename = json_params.get('output', None)
14
589f9250e30f Fix son reading.
Catherine Wise <catherine.wise@csiro.au>
parents: 13
diff changeset
111 output_data = all_params.get('output_data')
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
112 extra_files_path, file_name, ext, out_data_name, hda_id, dataset_id = \
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
113 operator.itemgetter('extra_files_path', 'file_name', 'ext', 'out_data_name', 'hda_id', 'dataset_id')(output_data[0])
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
114
10
16b0ceecf1f7 * Open in new window/tab * Apparently URL has to be capitalised (why... who knows, there's no docs).
Catherine Wise <catherine.wise@csiro.au>
parents: 9
diff changeset
115 url_params = json_params['URL'].split(";")
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
116 if len(url_params) < 3:
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
117 print("The url we got back is malformed: "+ url_params)
0
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
118 sys.exit(5)
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
119 short_url = url_params[0]
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
120 username = url_params[1]
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
121 password = url_params[2]
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
122 if "/short" not in short_url:
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
123 print("The url we got back is malformed: " + url_params)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
124 sys.exit(5)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
125 kms_url = short_url.split("/short")[0]
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
126
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
127 config = TrustStoreClient.Config(
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
128 None, kms_url, CLIENT_KEY, CLIENT_SECRET, tmpDir='/mnt/galaxy/tmp')
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
129 truststore = TrustStoreClient.TrustStoreClient(headless=False, config=config)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
130 try:
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
131 truststore.authenticate(username, password)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
132 except TrustStoreClient.TrustStoreClientAuthenticationException as err:
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
133 print(err)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
134 sys.exit(5)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
135 truststore.getPrivateKey('privkey.pem')
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
136
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
137 path_texts = truststore.lengthenPath(short_url)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
138 if len(path_texts) < 2:
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
139 print("The path we got was malformed.")
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
140 sys.exit(3)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
141 paths = path_texts[1:]
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
142 store_id = path_texts[0]
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
143
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
144 store = truststore.getStoreByID(store_id)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
145 if store is None:
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
146 print("Coudn't find store with that ID, or don't have access.")
0
2a996fea49a3 Initial.
Catherine Wise <catherine.wise@csiro.au>
parents:
diff changeset
147 sys.exit(2)
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
148 root = truststore.listDirectory(store)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
149
14
589f9250e30f Fix son reading.
Catherine Wise <catherine.wise@csiro.au>
parents: 13
diff changeset
150 first = True
589f9250e30f Fix son reading.
Catherine Wise <catherine.wise@csiro.au>
parents: 13
diff changeset
151
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
152 with open(metadata_path, 'wb') as metadata_file:
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
153 for path in paths:
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
154 locations = utils.Navigation.files_at_path(root, path)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
155 for location in locations:
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
156 if not locations:
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
157 print("Path not found: " + path)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
158 continue
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
159 filename = "".join(c in SAFE_CHARS and c or '-' for c in location.name)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
160 extension = os.path.splitext(filename)[1]
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
161 name = construct_multi_filename(hda_id, filename, extension)
14
589f9250e30f Fix son reading.
Catherine Wise <catherine.wise@csiro.au>
parents: 13
diff changeset
162 target_output_filename = None
589f9250e30f Fix son reading.
Catherine Wise <catherine.wise@csiro.au>
parents: 13
diff changeset
163 if first:
589f9250e30f Fix son reading.
Catherine Wise <catherine.wise@csiro.au>
parents: 13
diff changeset
164 target_output_filename = file_name
589f9250e30f Fix son reading.
Catherine Wise <catherine.wise@csiro.au>
parents: 13
diff changeset
165 first = False
589f9250e30f Fix son reading.
Catherine Wise <catherine.wise@csiro.au>
parents: 13
diff changeset
166 else:
589f9250e30f Fix son reading.
Catherine Wise <catherine.wise@csiro.au>
parents: 13
diff changeset
167 target_output_filename = os.path.normpath("/".join([extra_files_path, name]))
9
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
168 metadata_file.write(
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
169 metadata_to_json(dataset_id, filename, name, extension, target_output_filename))
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
170 download = truststore.getFile(store, location)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
171 if download is None:
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
172 print("File %s not found." % location.name)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
173 sys.exit(4)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
174 ungzip(download, target_output_filename)
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
175
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
176
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
177 if __name__ == '__main__':
0f2a5664c9eb Actually update downloader (probably not working).
Catherine Wise <catherine.wise@csiro.au>
parents: 0
diff changeset
178 main()