status-im/infra-utils
2
0
Fork 0
mirror of https://github.com/status-im/infra-utils.git synced 2025-02-22 17:08:25 +00:00
Code Issues Projects Releases Wiki Activity

peer_id.py: fix query check

Browse Source 
Signed-off-by: Jakub Sokołowski <jakub@status.im>
This commit is contained in:
Jakub Sokołowski 2020-05-27 09:22:40 +02:00
parent 4f3ccd269e
commit c1ff90f5e9
No known key found for this signature in database
GPG Key ID: 4EF064D0E6D63020
1 changed files with 14 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
elasticsearch/peer_id.py
View File

@ -75,21 +75,21 @@ def main():
body['query']['bool']['must_not'] = { 'exists': { 'field': 'peer_id' } }
for index in indices:
resp = es.count(index=index, body=body)
count = resp.get('count')
print('{:22} count: {:6}'.format(index, count))
resp = es.count(index=index, body=body)
count = resp.get('count')
print('{:22} count: {:6}'.format(index, count))
if opts.query > 0:
resp = es.search(index=index, body=body)
#print_logs(resp['hits']['hits'])
elif opts.update and count > 0:
# add the script for extracting peer_id
body['script'] = { 'lang': 'painless', 'inline': PAINLESS_SCRIPT }
try:
rval = es.update_by_query(index=index, body=body)
except Exception as ex:
print(json.dumps(ex.info, indent=2))
print('{:22} Updated: {:10} Failed: {}'.format(index, rval['updated'], rval.get('failed', 0)))
if opts.query and count > 0:
resp = es.search(index=index, body=body)
#print_logs(resp['hits']['hits'])
elif opts.update and count > 0:
# add the script for extracting peer_id
body['script'] = { 'lang': 'painless', 'inline': PAINLESS_SCRIPT }
try:
rval = es.update_by_query(index=index, body=body)
except Exception as ex:
print(json.dumps(ex.info, indent=2))
print('{:22} Updated: {:10} Failed: {}'.format(index, rval['updated'], rval.get('failed', 0)))
if __name__ == '__main__':
main()