diff --git a/cyapi/mixins/_Memory_Protection.py b/cyapi/mixins/_Memory_Protection.py index eb65e2a..31d1dc2 100644 --- a/cyapi/mixins/_Memory_Protection.py +++ b/cyapi/mixins/_Memory_Protection.py @@ -4,6 +4,13 @@ def get_memory_protection_events(self, start_time=None, end_time=None, **kwargs) '''Get a list of Memoryprotection Events''' if end_time and not start_time: raise ValueError("start_time must be set if using end_time parameter") + + '''Parse out Datetime Objects''' + if isinstance(start_time, datetime): + start_time = start_time.timestamp() + if isinstance(end_time, datetime): + end_time = end_time.timestamp() + params = {"start_time": start_time, "end_time": end_time} return self.get_list_items('memoryprotection', params=params, **kwargs)