path: root/scripts/common/__init__.py

#!/usr/bin/env python3
""" common functions and stuff """

import logging
import os
import sys
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)