1
0
Fork 0
mirror of https://github.com/Luzifer/mqtt2influx.git synced 2024-10-18 05:44:19 +00:00
mqtt2influx/vault.py
2020-07-11 18:20:49 +02:00

19 lines
577 B
Python

import hvac
import os
if not 'VAULT_ADDR' in os.environ or not 'VAULT_ROLE_ID' in os.environ:
raise Exception('VAULT_ADDR or VAULT_ROLE_ID are missing')
vault = hvac.Client(os.environ['VAULT_ADDR'])
auth = vault.auth_approle(os.environ['VAULT_ROLE_ID'])
if 'auth' in auth and 'client_token' in auth['auth']:
vault.token = auth['auth']['client_token']
else:
raise Exception('Authorization to Vault failed!')
def read_data(key):
resp = vault.read(key)
if 'data' not in resp:
raise Exception('Unable to read configuration')
return resp['data']