Added basic database models

requirements.txt

@@ -14,3 +14,6 @@ requests
 
 # Config file
 pyyaml
+
+# Database
+peewee

src/storage/database.py

@@ -0,0 +1,3 @@
+from peewee import SqliteDatabase
+
+database = SqliteDatabase("mash_database.sqlite")

src/storage/helper.py

@@ -0,0 +1,7 @@
+from .database import database
+from .models import *
+
+
+def create_tables():
+    with database:
+        database.create_tables([Device, Feature, StateLog])

src/storage/models/__init__.py

@@ -0,0 +1,3 @@
+from base_model import BaseModel
+from device import Device, Feature
+from logs import StateLog

src/storage/models/base_model.py

@@ -1,2 +1,11 @@
-class BaseModel():
+from ..database import database
+from peewee import AutoField
 
+class BaseModel():
+    id = AutoField(
+        primary_key=True,
+        unique=True,
+    )
+
+    class Meta:
+        database = database

src/storage/models/device.py

@@ -0,0 +1,13 @@
+from .base_model import BaseModel
+from peewee import CharField, DateTimeField, ForeignKeyField, BooleanField
+
+
+class Device(BaseModel):
+    friendly_name = CharField(max_length=200)
+
+
+class Feature(BaseModel):
+    friendly_name = CharField(max_length=200)
+    device = ForeignKeyField(Device, backref="sensors")
+    type = CharField(max_length=150)
+    is_sensor = BooleanField()

src/storage/models/logs.py

@@ -0,0 +1,11 @@
+from .base_model import BaseModel
+from .device import Feature
+from peewee import DateTimeField, ForeignKeyField, DecimalField, BooleanField
+
+
+class StateLog(BaseModel):
+    timestamp = DateTimeField()
+    feature = ForeignKeyField(Feature, backref="values")
+    char_value = DecimalField(5, 2, auto_round=True)
+    numeric_value = DecimalField(12, 3, auto_round=True)
+    previous_state = ForeignKeyField("StateLog")