Build time converter so a datetime.datetime object can be passed into…

… get_data

ambient_api/ambientapi.py

@@ -1,4 +1,5 @@
 import datetime
+import pprint
 import time
 
 import requests
@@ -28,9 +29,14 @@ def __str__(self):
         return '%s@%s' % (self.info.get('name'), self.mac_address)
 
     @staticmethod
-    def current_time():
+    def convert_datetime(datetime_object):
+        epoch = datetime.datetime.fromtimestamp(0)
 
-        return int(round(time.time() * 1000))
+        return int((datetime_object - epoch).total_seconds() * 1000.0)
+
+    def current_time(self):
+
+        return self.convert_datetime(datetime.datetime.now())
 
     def get_data(self, **kwargs):
         """
@@ -45,6 +51,9 @@ def get_data(self, **kwargs):
         limit = int(kwargs.get('limit', 288))
         end_date = kwargs.get('end_date', self.current_time())
 
+        if isinstance(end_date, datetime.datetime):
+            end_date = self.convert_datetime(end_date)
+
         if self.mac_address is not None:
             service_address = 'devices/%s' % self.mac_address
 

test.py

@@ -1,4 +1,5 @@
 import pprint
+import datetime
 
 from ambient_api.ambientapi import AmbientAPI
 
@@ -7,9 +8,12 @@
 devices = weather.get_devices()
 
 for device in devices:
+    print('Device')
     print(str(device))
 
+    print('Last Data')
     pprint.pprint(device.last_data)
 
-    pprint.pprint(device.get_data())
+    print('Get Data')
+    pprint.pprint(device.get_data(end_date=datetime.datetime(year=2018, month=10, day=1)))