AUTO-5174: Add support for setting date as well as time

Author: hadiTab

lgsvl/simulator.py

@@ -13,6 +13,8 @@
 
 from collections import namedtuple
 from environs import Env
+from datetime import datetime
+import re
 
 RaycastHit = namedtuple("RaycastHit", "distance point normal")
 
@@ -177,10 +179,29 @@ def weather(self, state):
     def time_of_day(self):
         return self.remote.command("environment/time/get")
 
+    @property
+    def current_datetime(self):
+        date_time_str = self.remote.command("simulator/datetime/get")
+        date_time_arr = list(map(int, re.split('[. :]', date_time_str)))
+        date_time = datetime(
+            date_time_arr[2],
+            date_time_arr[1],
+            date_time_arr[0],
+            date_time_arr[3],
+            date_time_arr[4],
+            date_time_arr[5]
+        )
+        return date_time
+
     @accepts((int, float), bool)
     def set_time_of_day(self, time, fixed=True):
         self.remote.command("environment/time/set", {"time": time, "fixed": fixed})
 
+    @accepts(datetime, bool)
+    def set_date_time(self, date_time, fixed=True):
+        date_time = date_time.__str__()
+        self.remote.command("environment/datetime/set", {"datetime": date_time, "fixed": fixed})
+
     def get_spawn(self):
         spawns = self.remote.command("map/spawn/get")
         return [Spawn.from_json(spawn) for spawn in spawns]

quickstart/19-time-of-day.py

@@ -7,6 +7,7 @@
 
 from environs import Env
 import lgsvl
+from datetime import datetime
 
 env = Env()
 
@@ -40,3 +41,14 @@
 sim.run(5)
 
 print(sim.time_of_day)
+
+input("Press Enter to set date to July 1 2020 and time to 19:00")
+dt = datetime(2020, 7, 1, 19, 0, 0, 0)
+sim.set_date_time(dt, False)
+print(sim.time_of_day)
+print(sim.current_datetime)
+
+sim.run(5)
+
+print(sim.time_of_day)
+print(sim.current_datetime)