summaryrefslogtreecommitdiff
path: root/heap/ebus/sql/model.py
blob: 7e21e82431226bc73924aab32d76b7deb4fc2fe3 (plain)
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
from datetime import datetime
from sqlalchemy import Column, Integer, Float, DateTime, String, ForeignKey
from sqlalchemy.orm import relationship, backref
from sqlalchemy.ext.declarative import declarative_base

ModelBase = declarative_base()

class Sensor(ModelBase):
    __tablename__ = 'sensor'
    
    id = Column(Integer, primary_key=True)
    name = Column(String)
    description = Column(String, default="")

    def __init__(self, name, description=""):
        self.name = name
        self.description = description
        
    def __repr__(self):
        return "<Sensor('%s','%s','%s')>" % (self.id, self.name, self.description)

# http://www.sqlalchemy.org/docs/05/reference/ext/declarative.html#inheritance-configuration
class Value(ModelBase):
    __tablename__ = 'value'

    id = Column(Integer, primary_key=True)
    timestamp = Column(DateTime)
    sensor_id = Column(Integer, ForeignKey("sensor.id"))
    sensor = relationship(Sensor, backref=backref('values', order_by=timestamp))

    discriminator = Column('type', String(50))
    __mapper_args__ = {'polymorphic_on': discriminator}

    def __init__(self, sensor, timestamp=datetime.now()):
        self.timestamp = timestamp
        self.sensor  = sensor

    def __repr__(self):
        return "<Value('%s','%s','%s','%s'>" % (self.id, self.sensor, self.value, self.timestamp)

    def value(self):
        raise NotImplementedException()

class ValueFloat(Value):
    __mapper_args__ = {'polymorphic_identity': 'float'}
    value_float = Column(Float(precision=4))
    
    def __init__(self, sensor, value, timestamp=datetime.now()):
        Value.__init__(self, sensor, timestamp)
        self.value_float = value

    def value(self):
        return self.value_float

class ValueInt(Value):
    __mapper_args__ = {'polymorphic_identity': 'int'}
    value_int = Column(Integer)

    def __init__(self, sensor, value, timestamp=datetime.now()):
        Value.__init__(self, sensor, timestamp)
        self.value_int = value

    def value(self):
        return self.value_int

class ValueString(Value):
    __mapper_args__ = {'polymorphic_identity': 'string'}
    value_string = Column(String)

    def __init__(self, sensor, value, timestamp=datetime.now()):
        Value.__init__(self, sensor, timestamp)
        self.value_string = value

    def value(self):
        return self.value_string