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

Clean up linting errors

Browse source
This commit is contained in:
Gabe Revells 2021-02-13 16:38:42 -05:00
parent ede4762818
commit a78da1281b
2 changed files with 12 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
varken/influxdb2manager.py
View file

@ -1,8 +1,7 @@
from sys import exit from sys import exit
from logging import getLogger from logging import getLogger
from requests.exceptions import ConnectionError
import influxdb_client import influxdb_client
from influxdb_client import InfluxDBClient, Point from influxdb_client import InfluxDBClient
from influxdb_client.client.write_api import SYNCHRONOUS from influxdb_client.client.write_api import SYNCHRONOUS
@ -15,7 +14,8 @@ class InfluxDB2Manager(object):
exit() exit()
self.influx = InfluxDBClient(url=self.server.url, token=self.server.token, org=self.server.org, self.influx = InfluxDBClient(url=self.server.url, token=self.server.token, org=self.server.org,
timeout=self.server.timeout, verify_ssl=self.server.verify_ssl, ssl_ca_cert=self.server.ssl) timeout=self.server.timeout, verify_ssl=self.server.verify_ssl,
ssl_ca_cert=self.server.ssl)
self.influx_write_api = self.influx.write_api(write_options=SYNCHRONOUS) self.influx_write_api = self.influx.write_api(write_options=SYNCHRONOUS)
# Create the bucket if needed # Create the bucket if needed
@ -41,6 +41,6 @@ class InfluxDB2Manager(object):
try: try:
self.influx_write_api.write(bucket=self.server.bucket, record=d) self.influx_write_api.write(bucket=self.server.bucket, record=d)
except (InfluxDBServerError, ConnectionError) as e: except Exception as e:
self.logger.error('Error writing data to influxdb2. Dropping this set of data. ' self.logger.error('Error writing data to influxdb2. Dropping this set of data. '
'Check your database! Error: %s', e) 'Check your database! Error: %s', e)

6
varken/iniparser.py
View file

@ -146,12 +146,14 @@ class INIParser(object):
self.config_blacklist() self.config_blacklist()
# Parse InfluxDB options # Parse InfluxDB options
self.influx2_enabled = env.get('VRKN_GLOBAL_INFLUXDB2_ENABLED', self.config.getboolean('global', 'influx2_enabled')) self.influx2_enabled = env.get('VRKN_GLOBAL_INFLUXDB2_ENABLED',
self.config.getboolean('global', 'influx2_enabled'))
if self.influx2_enabled: if self.influx2_enabled:
# Use INFLUX version 2 # Use INFLUX version 2
try: try:
url = self.url_check(env.get('VRKN_INFLUXDB2_URL', self.config.get('influx2', 'url')), section='influx2') url = self.url_check(env.get('VRKN_INFLUXDB2_URL', self.config.get('influx2', 'url')),
section='influx2')
ssl = boolcheck(env.get('VRKN_INFLUXDB2_SSL', self.config.get('influx2', 'ssl'))) ssl = boolcheck(env.get('VRKN_INFLUXDB2_SSL', self.config.get('influx2', 'ssl')))
verify_ssl = boolcheck(env.get('VRKN_INFLUXDB2_VERIFY_SSL', self.config.get('influx2', 'verify_ssl'))) verify_ssl = boolcheck(env.get('VRKN_INFLUXDB2_VERIFY_SSL', self.config.get('influx2', 'verify_ssl')))