d_mcknight/Varken
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Wrap create bucket in try/catch

Browse source
This commit is contained in:
Gabe Revells 2021-02-13 16:40:02 -05:00 • committed by Daniel
parent 90ae8fac32
commit 622f9a65c1
1 changed files with 12 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

21
varken/influxdb2manager.py
View file

@ -22,18 +22,21 @@ class InfluxDB2Manager(object):
bucket_api = self.influx.buckets_api() bucket_api = self.influx.buckets_api()
bucket = bucket_api.find_bucket_by_name(self.server.bucket) try:
bucket = bucket_api.find_bucket_by_name(self.server.bucket)
if bucket is None: if bucket is None:
self.logger.info('Creating bucket %s', self.server.bucket) self.logger.info('Creating bucket %s', self.server.bucket)
org_api = influxdb_client.service.organizations_service.OrganizationsService(self.influx.api_client) org_api = influxdb_client.service.organizations_service.OrganizationsService(self.influx.api_client)
orgs = org_api.get_orgs() orgs = org_api.get_orgs()
for org in orgs.orgs: for org in orgs.orgs:
if org.name == self.server.org: if org.name == self.server.org:
my_org = org my_org = org
self.influx.buckets_api().create_bucket(bucket_name=self.server.bucket, org_id=my_org.id) self.influx.buckets_api().create_bucket(bucket_name=self.server.bucket, org_id=my_org.id)
except Exception as e:
self.logger.error('Failed creating new InfluxDB bucket! Error: %s', e)
def write_points(self, data): def write_points(self, data):
d = data d = data