Complete Module 2

Author: johnjbarry

sensor/house_info.py

@@ -0,0 +1,22 @@
+from datetime import date, datetime
+
+
+class HouseInfo:
+    def __init__(self, data):
+        self.data = data
+
+    def get_data_by_area(self, field, rec_area=0):
+        field_data = []
+        for record in self.data:
+            if rec_area == 0:
+                field_data.append(record[field])
+            elif rec_area == int(record['area']):
+                field_data.append(record[field])
+        return field_data
+
+    def get_data_by_date(self, field, rec_date=date.today()):
+        field_data = []
+        for record in self.data:
+            if rec_date.strftime("%m/%d/%y") == record['date']:
+                field_data.append(record[field])
+        return field_data

sensor/sensor_app.py

@@ -1,5 +1,7 @@
 # Runner script for all modules
 from load_data import load_sensor_data
+from house_info import HouseInfo
+from datetime import date, datetime
 
 ##############################
 # Do not remove these two lines
@@ -11,10 +13,18 @@
 # Module 1 code here:
 data = load_sensor_data()
 print("Loaded records: {}".format(len(data)))
+
 # Module 2 code here:
+house_info = HouseInfo(data)
+test_area = 1
+recs = house_info.get_data_by_area("id", rec_area=test_area)
+print("\nHouse sensor records for area {} = {}".format(test_area, len(recs)))
+test_date = datetime.strptime("5/9/20", "%m/%d/%y")
+recs = house_info.get_data_by_date("id", rec_date=test_date)
+print("\nHouse sensor records for date: {} = {}".format(test_date.strftime("%m/%d/%y"), len(recs)))
 
 # Module 3 code here:
 
 # Module 4 code here:
 
-# Module 5 code here:
+# Module 5 code here: