piclock_web_modules.py

#!/usr/bin/python3
import pygal
import sqlite3
import subprocess
from pygal.style import DarkSolarizedStyle
from sysdmanager import SystemdManager

def get_db_data(db_path, table_name, from_date, until_date):
    with sqlite3.connect(db_path) as connection:
        cursor = connection.cursor()
        cursor.execute("select * from {} where date between '{}' and '{}'".format(table_name, from_date, until_date))
        result = cursor.fetchall()
    return result

def graph_from_data(data, dots, minor_labels, air):
    time = []
    inside_temp = []
    humidity = []
    pressure = []
    outside_temp = []
    co2 = []
    tvoc = []

    for item in data:
        time.append(item[0].split('.')[0])
        inside_temp.append(float(item[1]))
        humidity.append(float(item[2]))
        pressure.append(float(item[3])-700)
        if item[4] == 'NA':
            outside_temp.append(None)
        else:
            outside_temp.append(float(item[4]))
        co2.append(float(item[5]))
        tvoc.append(float(item[6]))

    line_chart = pygal.Line(x_label_rotation=-45,
                            interpolate='hermite',
                            allow_interruptions = True,
                            x_labels_major_every=10,
                            show_minor_x_labels = minor_labels,
                            show_only_major_dots = dots,
                            style=DarkSolarizedStyle)

    line_chart.title = "Weather conditions"

    line_chart.x_labels = time
    line_chart.add('inside_temp', inside_temp)
    line_chart.add('humidity', humidity)
    line_chart.add('pressure+700', pressure)
    line_chart.add('outside_temp', outside_temp)
    if air:
        line_chart.add('CO2', co2)
        line_chart.add('TVOC', tvoc)
    return line_chart.render_response()

def db_to_graph(db_path,
                table_name,
                from_date,
                until_date,
                dots = True,
                minor_labels = True,
                air = False):
    data = get_db_data(db_path,
                       table_name,
                       from_date,
                       until_date)
    return graph_from_data(data, dots, minor_labels, air)

def service_control(service, action):
    manager = SystemdManager()
    if action == 'up':
        manager.start_unit('{}.service'.format(service))
        return "{}.service has been started".format(service)
    elif action == 'down':
        manager.stop_unit('{}.service'.format(service))
        return '{}.service has been stopped'.format(service)
    else:
        return 'Only up/down are supported'

def service_stats():
    command = 'systemctl status piclock_*.service | grep "● .*\.service -\|   Active:" | grep -v "─"'
    result = subprocess.check_output(command, shell=True)
    return result.decode('utf-8').strip().split('\n')

def main():
    db_to_graph('/root/temp-data/temp-data.db',
                'weather',
                '2019-02-02',
                '2019-02-03')

if __name__ == '__main__':
    main()