Merge pull request #166 from EJEP/add_print_capability

Add print capability to Timestep, Element, Day and Site

Author: Emlyn Price

datapoint/Day.py

@@ -4,3 +4,20 @@ def __init__(self, api_key=""):
 
         self.date = None
         self.timesteps = []
+
+    def __str__(self):
+        day_str = ''
+
+        date_part = 'Date: ' + str(self.date) + '\n\n'
+        day_str += date_part
+
+        day_str += 'Timesteps: \n\n'
+        try:
+            for timestep in self.timesteps:
+                day_str += str(timestep)
+                day_str += '\n'
+
+        except TypeError:
+            day_str += 'No timesteps'
+
+        return day_str

datapoint/Element.py

@@ -7,3 +7,6 @@ def __init__(self, id=None, value=None, units=None):
 
         # For elements which can also have a text value
         self.text = None
+
+    def __str__(self):
+        return str(self.value) + ' ' + str(self.units)

datapoint/Site.py

@@ -10,3 +10,11 @@ def __init__(self, api_key=""):
         self.nationalPark = None
         self.region = None
         self.unitaryAuthArea = None
+
+    def __str__(self):
+        site_string = ''
+        for attr, value in self.__dict__.items():
+            to_append = attr + ': ' + str(value) + '\n'
+            site_string += to_append
+
+        return site_string

datapoint/Timestep.py

@@ -29,3 +29,10 @@ def elements(self):
         elements = [el[1] for el in self.__dict__.items() if isinstance(el[1], Element)]
 
         return elements
+
+    def __str__(self):
+        timestep_string = ''
+        for attr, value in self.__dict__.items():
+            to_append = attr + ': ' + str(value) + '\n'
+            timestep_string += to_append
+        return timestep_string