Changeset 367

Timestamp:

09/02/10 19:38:16

Author:

cbc

Message:

Created Data, Device, Sensor, and Point classes in parse module.

Files:

• spongenet/trunk/spongenet/parse.py (modified) (5 diffs)

spongenet/trunk/spongenet/parse.py

@@ -22 +22 @@
-import sys
-import os
+import re
-import glob
-import hashlib
@@ -31 +32 @@
-USAGE = "\n".join(__doc__.splitlines()[3:8])
-TEST_PATH = os.path.join("tests", "parse")
+XMLNS_PATTERN = re.compile(r"(\{.*\})(.*)")
-
-
@@ -164 +166 @@
-    Verify the test document matches the reference data.
-
-
-
-
-
-
+
+
+
+
+
+
-    True
-    >>> doc_hash = hashlib.md5()
@@ -183 +186 @@
-
-
+class Point(object):
+    """
+    A data point for a sponge sensor sample."
+    """
+
+    def __init__(self, point):
+        for key, value in point.attrib.items():
+            key = key.lower()
+            super(Point, self).__setattr__(key, value)
+        for elem in point.getchildren():
+            tag = XMLNS_PATTERN.search(elem.tag).groups()[1].lower()
+            super(Point, self).__setattr__(tag, elem.text)
+
+
+class Sensor(object):
+    """
+    A collection of data points for a sponge sensor sample.
+    """
+
+    def __init__(self, sensor, xmlns):
+        for key, value in sensor.attrib.items():
+            key = key.lower()
+            super(Sensor, self).__setattr__(key, value)
+        for elem in sensor.getchildren():
+            tag = XMLNS_PATTERN.search(elem.tag).groups()[1].lower()
+            if tag == "parameters":
+                self.points = [Point(point) for point
+                                in elem.findall(xmlns + "Point")]
+            else:
+                super(Sensor, self).__setattr__(tag, elem.text)
+
+
+class Device(object):
+    """
+    Data from a collection of sponge sensors for a single time sample.
+    """
+
+    def __init__(self, device, xmlns):
+        for key, value in device.attrib.items():
+            key = key.lower()
+            super(Device, self).__setattr__(key, value)
+        for elem in device.getchildren():
+            tag = XMLNS_PATTERN.search(elem.tag).groups()[1].lower()
+            if tag == "siteinfo":
+                for subelem in elem.getchildren():
+                    tag = XMLNS_PATTERN.search(subelem.tag).groups()[1].lower()
+                    super(Device, self).__setattr__(tag, subelem.text)
+            elif tag == "data":
+                for key, value in elem.attrib.items():
+                    key = key.lower()
+                    if key == "time":
+                        key = "data_time"
+                    elif key == "sessionid":
+                        key = "data_sessionid"
+                    super(Device, self).__setattr__(key, value)
+                self.sensors = [Sensor(sensor, xmlns) for sensor
+                                in elem.findall(xmlns + "SensorData")]
+            else:
+                super(Device, self).__setattr__(tag, elem.text)
+
+
+class Data(object):
+    """
+    A collection of sponge data samples from a collection of sensors.
+    """
+
+    def __init__(self, _xmldoc):
+        """
+        Initialize a new sponge data tree.
+        """
+
+        tree = ET.XML(_xmldoc)
+        self.xmlns = XMLNS_PATTERN.search(
+                         tree.getchildren()[0].tag).groups()[0]
+        self.devices = [Device(device, self.xmlns) for device
+                        in tree.findall(self.xmlns + "Device")]
+
+
-def _main():
-    """
@@ -191 +272 @@
-    >>> from parse import _main
-    """
+
+    data = None
-
-    _xmldoc_path = xmldoc_path()
-    if _xmldoc_path:
-        _xmldoc = xmldoc(_xmldoc_path)
-print len
-
-
+data = Data
+
+ data
-
-if __name__ == "__main__":
-
+
+