14
14
from astroquery .utils .tap .core import TapPlus
15
15
from astroquery .cadc .cadctap .tapconn import TapConnCadc
16
16
from astroquery .cadc .cadctap .jobSaxParser import JobSaxParserCadc
17
+ from astroquery .utils .tap import taputils
17
18
18
19
19
20
__all__ = ['TapPlusCadc' ]
@@ -74,31 +75,35 @@ def __init__(self, url=None, host=None, server_context=None,
74
75
tap_context = self ._Tap__parseUrl (url )
75
76
if protocol == "http" :
76
77
connHandler = TapConnCadc (False ,
77
- host ,
78
- server_context ,
79
- tap_context ,
80
- port ,
81
- sslport )
78
+ host = host ,
79
+ server_context = server_context ,
80
+ tap_context = tap_context ,
81
+ port = port ,
82
+ sslport = sslport )
82
83
else :
83
84
# https port -> sslPort
84
85
connHandler = TapConnCadc (True ,
85
- host ,
86
- server_context ,
87
- tap_context ,
88
- port ,
89
- port )
86
+ host = host ,
87
+ server_context = server_context ,
88
+ tap_context = tap_context ,
89
+ port = port ,
90
+ sslport = port )
90
91
else :
91
92
connHandler = TapConnCadc (default_protocol_is_https ,
92
- host ,
93
- server_context ,
94
- tap_context ,
95
- port ,
96
- sslport )
93
+ host = host ,
94
+ server_context = server_context ,
95
+ tap_context = tap_context ,
96
+ port = port ,
97
+ sslport = sslport )
97
98
self .__certificate = None
98
- super (TapPlusCadc , self ).__init__ (url , host , server_context ,
99
- tap_context , port , sslport ,
100
- default_protocol_is_https ,
101
- connHandler , verbose )
99
+ super (TapPlusCadc , self ).__init__ (url = url , host = host ,
100
+ server_context = server_context ,
101
+ tap_context = tap_context , port = port ,
102
+ sslport = sslport ,
103
+ default_protocol_is_https = default_protocol_is_https ,
104
+
105
+ connhandler = connHandler ,
106
+ verbose = verbose )
102
107
103
108
def load_table (self , table , verbose = False ):
104
109
"""Loads the specified table
@@ -130,7 +135,8 @@ def load_table(self, table, verbose=False):
130
135
return
131
136
132
137
def _Tap__launchJobMultipart (self , query , uploadResource , uploadTableName ,
133
- outputFormat , context , verbose , name = None ):
138
+ outputFormat , context , verbose , name = None ,
139
+ autorun = True ):
134
140
"""
135
141
136
142
Notes
@@ -152,9 +158,8 @@ def _Tap__launchJobMultipart(self, query, uploadResource, uploadTableName,
152
158
files = [[uploadTableName , uploadResource , chunk ]]
153
159
contentType , body = self ._Tap__connHandler .encode_multipart (args ,
154
160
files )
155
- response = self ._Tap__connHandler .execute_post (context ,
156
- body ,
157
- contentType )
161
+ response = self ._Tap__connHandler .execute_tappost (context , body ,
162
+ contentType )
158
163
if verbose :
159
164
print (response .status , response .reason )
160
165
print (response .getheaders ())
@@ -163,13 +168,13 @@ def _Tap__launchJobMultipart(self, query, uploadResource, uploadTableName,
163
168
location = self ._Tap__connHandler .find_header (
164
169
response .getheaders (),
165
170
"location" )
166
- jobid = self . _Tap__getJobId (location )
171
+ jobid = taputils . get_jobid_from_location (location )
167
172
runresponse = self .__runAsyncQuery (jobid , verbose )
168
173
return runresponse
169
174
return response
170
175
171
176
def _Tap__launchJob (self , query , outputFormat ,
172
- context , verbose , name = None ):
177
+ context , verbose , name = None , autorun = True ):
173
178
"""
174
179
175
180
Notes
@@ -184,15 +189,15 @@ def _Tap__launchJob(self, query, outputFormat,
184
189
"tapclient" : str (TAP_CLIENT_ID ),
185
190
"QUERY" : str (query )}
186
191
data = self ._Tap__connHandler .url_encode (args )
187
- response = self ._Tap__connHandler .execute_post (context , data )
192
+ response = self ._Tap__connHandler .execute_tappost (context , data )
188
193
if verbose :
189
194
print (response .status , response .reason )
190
195
print (response .getheaders ())
191
196
if 'async' in context :
192
197
location = self ._Tap__connHandler .find_header (
193
198
response .getheaders (),
194
199
"location" )
195
- jobid = self . _Tap__getJobId (location )
200
+ jobid = taputils . get_jobid_from_location (location )
196
201
runresponse = self .__runAsyncQuery (jobid , verbose )
197
202
return runresponse
198
203
return response
@@ -206,8 +211,8 @@ def __runAsyncQuery(self, jobid, verbose):
206
211
args = {
207
212
"PHASE" : "RUN" }
208
213
data = self ._Tap__connHandler .url_encode (args )
209
- response = self ._Tap__connHandler .execute_post ('async/' + jobid + '/phase' ,
210
- data )
214
+ response = self ._Tap__connHandler .execute_tappost ('async/' + jobid + '/phase' ,
215
+ data )
211
216
if verbose :
212
217
print (response .status , response .reason )
213
218
print (response .getheaders ())
@@ -237,7 +242,7 @@ def load_async_job(self, jobid=None, verbose=False):
237
242
print ("No job identifier found" )
238
243
return
239
244
subContext = "async/" + str (jobid )
240
- response = self ._Tap__connHandler .execute_get (subContext )
245
+ response = self ._Tap__connHandler .execute_tapget (subContext )
241
246
if verbose :
242
247
print (response .status , response .reason )
243
248
print (response .getheaders ())
@@ -310,6 +315,7 @@ def login(self, user=None, password=None, certificate_file=None,
310
315
raise AttributeError (
311
316
'Choose one form of authentication only' )
312
317
if certificate_file is not None :
318
+ print (dir (self ._TapPlus__getconnhandler ().cookies_set ))
313
319
if self ._TapPlus__getconnhandler ().cookies_set ():
314
320
raise AttributeError ('Already logged in with user/password' )
315
321
if not os .path .isfile (certificate_file ):
0 commit comments