@@ -185,7 +185,7 @@ def main():
185
185
'CLIENT_SECRET' : '' ,
186
186
'OPENID_URL' : 'https://keycloak.icecube.wisc.edu/auth/realms/IceCube' ,
187
187
'FILE_CATALOG_URL' : 'https://file-catalog.icecube.wisc.edu' ,
188
- 'ES_ADDRESS' : 'http ://elk-1 .icecube.wisc.edu:9200 ' ,
188
+ 'ES_ADDRESS' : 'https ://elastic .icecube.aq ' ,
189
189
'ES_INDEX' : 'long-term-archive' ,
190
190
'ES_TIMEOUT' : 60. ,
191
191
'START_DATE' : '' ,
@@ -203,10 +203,22 @@ def main():
203
203
parser .add_argument ('-n' , '--dry-run' , default = False , action = 'store_true' ,
204
204
help = 'do not ingest into ES, just print' )
205
205
parser .add_argument ('--log-level' , default = 'info' , choices = ['debug' , 'info' , 'warning' , 'error' ])
206
+ parser .add_argument ('--es_client_id' ,default = None ,
207
+ help = 'ES oauth2 client id' )
208
+ parser .add_argument ('--es_client_secret' ,default = None ,
209
+ help = 'ES oauth2 client secret' )
210
+ parser .add_argument ('--token_url' ,default = None ,
211
+ help = 'ES oauth2 realm token url' )
206
212
args = parser .parse_args ()
207
213
208
214
logging .basicConfig (level = getattr (logging , args .log_level .upper ()), format = '%(asctime)s %(levelname)s %(name)s : %(message)s' )
209
215
216
+ es_api = ClientCredentialsAuth (address = 'https://elasticsearch.icecube.aq' ,
217
+ token_url = args .token_url ,
218
+ client_secret = args .es_client_secret ,
219
+ client_id = args .es_client_id )
220
+ es_token = es_api .make_access_token ()
221
+
210
222
if args .from_file :
211
223
rest_client = None
212
224
else :
@@ -225,6 +237,7 @@ def main():
225
237
request_timeout = config ['ES_TIMEOUT' ],
226
238
retry_on_timeout = True ,
227
239
max_retries = 2 ,
240
+ bearer_auth = es_token ,
228
241
)
229
242
230
243
c = Collect (
0 commit comments