foo

2 files changed, 220 insertions, 0 deletions

diff --git a/ebus/model/__init__.py b/ebus/model/__init__.py
new file mode 100644
index 0000000..ecdf3bc
--- /dev/null
+++ b/ebus/model/__init__.py
@@ -0,0 +1,92 @@
+class DataField(object):
+    def __init__(self, name, offset):
+        self.name = name
+        self.offset = offset
+    def value(self,data):
+        raise NotImplemented()
+
+ class Data1b(DataField):
+     """
+    Beispiel für die Berechnung:
+    if ((x & 80h) == 80h) // y negativ
+    y     = - [dez(!x) + 1]
+    else 
+    y     = dez(x)
+    """
+     def value(self,data):
+         x = ord(data[self.offset])
+         if x & 0x80 == 0x80:
+             return (-1) * (0xff^x + 1)
+         else:
+             return x
+
+class Data1c(DataField):
+    def value(self,data):
+        return ord(data[self.offset])/2.0
+                 
+class Data2c(DataField):
+    """
+    Beispiel für die Berechnung:
+    if ((x & 8000h) == 8000h) // y negativ
+    y = - [dez(High_Byte(!x)) 16 + dez(High_Nibble (Low_Byte (!x)))
+    + (dez(Low_Nibble (Low_Byte (!x))) +1 ) / 16]
+    else                       // y positiv
+    y = dez(High_Byte(x)) 16 + dez(High_ Nibble (Low Byte (x)))
+    + dez(Low_ Nibble (Low Byte (x))) / 16
+    """
+    def value(self,data):
+        highByte = ord(data[self.offset+1])
+        lowByte= ord(data[self.offset])
+        if (0x8000 & (highByte<<8 | lowByte)) == 0x8000:
+            return (-1) * ( (0xff^highByte)*16 + (0xff^(lowByte>>4)) + (0x0ff^(0xf&lowByte) + 1)/16.0 )
+        else:
+            return highByte*16 + (lowByte>>4) + (lowByte&0xf)/16.0
+
+class Data2b(DataField):
+    """
+    if ((x&8000h) == 8000h) // y negativ
+    y = - [dez(High_Byte(!x)) + (dez(Low_Byte(!x)) + 1) / 256]
+    else                        // y positiv
+    y = dez(High_Byte (x)) + dez(Low_Byte (x)) / 256
+    """
+    def value(self,data):
+        highByte = ord(data[self.offset+1])
+        lowByte= ord(data[self.offset])
+        if (0x8000 & (highByte<<8 | lowByte)) == 0x8000:
+            return (-1) * ((0xff^highByte) + (0xff^lowByte+1)/256.0)
+        else:
+            return highByte + lowByte/256.0
+        
+class Bit(DataField):
+    def value(self, data):
+        return ord(data[self.offset]) == 0x1
+    
+class Word(DataField):
+    def value(self, data):
+        lb, hb = data[self.offset], data[self.offset+1]
+        return ord(lb) + (ord(hb)<<8)
+    
+class Bcd(DataField):
+    """
+    y = dez(High_Nibble(x))*10 + dez(Low_Nibble(x))
+    """
+    def value(self, data):
+        byte = ord(data[self.offset])
+        return (byte >> 4) * 10 + (byte & 0xf)
+    
+class Byte(DataField):
+    def value(self, data):
+        return ord(data[self.offset])
+    
+
+class ByteEnum(DataField):
+    def __init__(self, name, offset, values):
+        fields.DataField.__init__(self, name, offset)
+        self.values = values
+
+    def value(self, data):
+        value = ord(data[self.offset])
+        if self.values.has_key(value):
+            return self.values[value]
+        else:
+            return None
diff --git a/ebus/model/sql.py b/ebus/model/sql.py
new file mode 100644
index 0000000..fa9b479
--- /dev/null
+++ b/ebus/model/sql.py
@@ -0,0 +1,128 @@
+# -*- coding: utf-8 -*-
+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
+
+
+class EbusPacketSQL(object):
+    def __init__(self,session):
+        self.session = session
+
+    def get_values(self,packet):
+        packet_description = packet.get_packet_description()
+        sensor_name = packet._get_source_name()
+        sensor = self.session.query(Sensor) \
+            .filter(Sensor.name==sensor_name)
+        
+        if packet is None:
+            return
+
+        for field in packet_description.fields:
+            name = field.name
+            offset = int(field.offset)
+            if not name:
+                continue
+            elif field.tag == "bit":
+                yield ValueInt(sensor, \
+                                   fields.Bit(name, offset).value())
+            elif field.tag == "byte":
+                yield ValueInt(sensor, \
+                                   fields.Byte(name, offset).value())
+            elif field.tag == "word":
+                yield ValueInt(sensor, \
+                                   fields.Word(name, offset).value())
+            elif field.tag == "data1b":
+                yield ValueInt(sensor, \
+                                   fields.Data1b(name, offset).value())
+            elif field.tag == "data1c":
+                yield ValueInt(sensor, \
+                                   fields.Data1c(name, offset).value())
+            elif field.tag == "data2b":
+                yield ValueInt(sensor, \
+                                   fields.Data2b(name, offset).value())
+            elif field.tag == "data2c":
+                yield ValueInt(sensor, 
+                               fields.Data2c(name, offset).value())
+            elif field.tag == "bcd":
+                yield ValueString(sensor,
+                                  fields.Bcd(name, offset).value())
+            elif field.tag == "byteEnum":
+                options = dict(map(lambda opt: ( int(opt.get("value")[2:],16), opt.get("name")),
+                                   field.xpath("./option")))
+                yield ValueString(sensor,
+                                  fields.ByteEnum(name, offset, options).value())
+            else:
+                print "Unsupported field-type %s" % (name)
+