GC_database/build_database_BM280.py at main · GrowCab/GC_database · GitHub

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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
# -*- coding: UTF-8 -*-

import os
from config.config import db
from models.models import Sensor, Chamber, Unit, SensorUnit, Configuration, ExpectedMeasure, ChamberSensor, \
    SensorMeasure, \
    MeasureGroup, Actuator, ActuatorEffect, ActuatorMeasure, ChamberActuator

if __name__ == "__main__":

    # Delete database file if it exists currently
    if os.path.exists("growcab.db"):
        os.remove("growcab.db")

    # Create the database
    db.create_all()

    # populate the database

    # Add a new example chamber
    chamber1 = Chamber(description="Example")
    db.session.add(chamber1)
    db.session.commit()

    # Add units

    cel = Unit(description='C', label="Temperature 🌡", hardware_label="temperature")
    lux = Unit(description='💡', label="Light 💡", hardware_label="visible_light")
    hum = Unit(description='%', label="Humidity ☁️", hardware_label="humidity", controllable = False)

    db.session.bulk_save_objects([lux, hum, cel], return_defaults=True)
    db.session.commit()

    # Add a new sensors
    temperature_and_humidity_sensor = Sensor(description="Temperature", hardware_classname="BME280")
    db.session.add_all([temperature_and_humidity_sensor])
    db.session.commit()

    # Add sensor units
    celsius_unit = SensorUnit(min=-5, max=40, unit=cel, sensor=temperature_and_humidity_sensor)
    humidity_unit = SensorUnit(min=0, max=100, unit=hum, sensor=temperature_and_humidity_sensor)
    db.session.add_all([celsius_unit, humidity_unit])
    db.session.commit()

    # Add sensors to the chamber
    chamber_sensor1 = ChamberSensor(chamber=chamber1, sensor=temperature_and_humidity_sensor)
    db.session.add_all([
        chamber_sensor1
    ])
    db.session.commit()

    # Add a configuration to the chamber
    chamber1_configuration = Configuration(chamber_id=chamber1.id, description="Until forever")
    db.session.add(chamber1_configuration)
    db.session.commit()

    expected1 = ExpectedMeasure(expected_value=28, end_hour=6, end_minute=0,
                                configuration=chamber1_configuration, unit=cel)
    expected2 = ExpectedMeasure(expected_value=16, end_hour=23, end_minute=59,
                                configuration=chamber1_configuration, unit=cel)
    db.session.add_all([expected1, expected2])
    db.session.commit()

    db.session.add_all([
        ExpectedMeasure(expected_value=0, end_hour=6, end_minute=0, configuration=chamber1_configuration,
                        unit_id=lux.id),
        ExpectedMeasure(expected_value=1, end_hour=23, end_minute=59, configuration=chamber1_configuration,
                        unit_id=lux.id),
        ExpectedMeasure(expected_value=50, end_hour=23, end_minute=59, configuration=chamber1_configuration,
                        unit_id=hum.id)
    ])
    db.session.commit()

    all_sensors = Sensor.query.all()
    all_chambers = Chamber.query.all()
    all_sensor_units = SensorUnit.query.all()
    all_units = Unit.query.all()
    all_configurations = Configuration.query.all()

    light_actuator = Actuator(description="Light actuator")
    light_actuator.actuator_effect.append(ActuatorEffect(change=1, actuator_id=light_actuator.id, unit_id=lux.id))
    cooling_actuator = Actuator(description="cooling actuator")
    cooling_actuator.actuator_effect.append(ActuatorEffect(change=-10, actuator_id=cooling_actuator.id, unit_id=cel.id))
    heating_actuator = Actuator(description="heating actuator")
    heating_actuator.actuator_effect.append(ActuatorEffect(change=5, actuator_id=heating_actuator.id, unit_id=hum.id))


    db.session.add_all([light_actuator, cooling_actuator, heating_actuator])
    db.session.commit()

    chamber1_light_actuator = ChamberActuator(chamber=chamber1, actuator=light_actuator)
    db.session.add(chamber1_light_actuator)
    db.session.commit()

    measure_group = MeasureGroup()
    measure_group.sensor_measure.extend([
        SensorMeasure(sensor_unit=humidity_unit, chamber_sensor=chamber_sensor1, current_value=30),
        SensorMeasure(sensor_unit=celsius_unit, chamber_sensor=chamber_sensor1, current_value=20)
    ])
    measure_group.actuator_measure.extend([
        ActuatorMeasure(chamber_actuator=chamber1_light_actuator, measure_group=measure_group, current_value=0)
    ])
    db.session.add(measure_group)
    db.session.commit()

    [print(s) for s in all_sensors]