1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
#!/usr/bin/env python3
''' common functions and stuff '''
import os
import sys
import logging
from datetime import datetime
QUEUE = bool(os.environ.get('el_QUEUE', False))
name = os.path.splitext(os.path.split(sys.argv[0])[-1])[0]
log = logging.getLogger(__name__)
log.setLevel(logging.INFO)
logging.basicConfig(format="%(name)s: %(levelname)s %(message)s")
# Use queue or postgres
if QUEUE is True:
log.debug('Importing dbi from queue')
from .queue import dbi
else:
log.debug('Importing dbi from postgres')
from .postgres import dbi
# Insert state
def statein(sensor, value, device_class, unit, **kwargs):
''' Insert new state in db '''
verbose = bool(kwargs['verbose']) if 'verbose' in kwargs else False
sql = """INSERT INTO states
(sensor_id,
value,
device_class,
unit,
time)
SELECT sensors.id, %s, %s, %s, %s
FROM sensors
WHERE sensors.name = %s;"""
values = (value,
device_class,
unit,
datetime.utcnow(),
sensor)
log.debug('dbi(' + sql + ',' + str(values) + ')')
dbi(sql, values, verbose=verbose)
|
