craftbeerpi4-pione/core/database/model.py

from core.database.orm_framework import DBModel


class ActorModel(DBModel):
    __fields__ = ["name", "type", "config"]
    __table_name__ = "actor"
    __json_fields__ = ["config"]


class SensorModel(DBModel):
    __fields__ = ["name", "type", "config"]
    __table_name__ = "sensor"
    __json_fields__ = ["config"]

class ConfigModel(DBModel):
    __fields__ = ["type", "value", "description", "options"]
    __table_name__ = "config"
    __json_fields__ = ["options"]
    __priamry_key__ = "name"

class KettleModel(DBModel):
    __fields__ = ["name","sensor", "heater", "automatic", "logic", "config", "agitator", "target_temp"]
    __table_name__ = "kettle"
    __json_fields__ = ["config"]


class StepModel(DBModel):
    __fields__ = ["order", "name", "type", "stepstate", "state", "start", "end",  "config",  "kettleid"]
    __table_name__ = "step"
    __json_fields__ = ["config", "stepstate"]


    '''
    @classmethod
    def sort(cls, new_order):
        cur = get_db().cursor()
        for key, value in new_order.items():
            cur.execute("UPDATE %s SET '%s' = ? WHERE id = ?" % (cls.__table_name__, "order"), (value, key))
        get_db().commit()

    @classmethod
    def get_max_order(cls):
        cur = get_db().cursor()
        cur.execute("SELECT max(step.'order') as 'order' FROM %s" % cls.__table_name__)
        r = cur.fetchone()
        return r.get("order")

    @classmethod
    def delete_all(cls):
        cur = get_db().cursor()
        cur.execute("DELETE FROM %s" % cls.__table_name__)
        get_db().commit()

    @classmethod
    def get_by_state(cls, state, order=True):
        cur = get_db().cursor()
        cur.execute("SELECT * FROM %s WHERE state = ? ORDER BY %s.'order'" % (cls.__table_name__, cls.__table_name__,), state)
        r = cur.fetchone()
        if r is not None:
            return cls(r)
        else:
            return None

    @classmethod
    def update_step_state(cls, id, state):
        cur = get_db().cursor()
        cur.execute("UPDATE %s SET stepstate = ? WHERE id =?" % cls.__table_name__, (json.dumps(state), id))
        get_db().commit()

    @classmethod
    def reset_all_steps(cls):
        cur = get_db().cursor()
        cur.execute("UPDATE %s SET state = 'I', stepstate = NULL , start = NULL, end = NULL " % cls.__table_name__)
        get_db().commit()
    '''