From c52cb9b2bab63322322ddbb8fe4a3811b2540011 Mon Sep 17 00:00:00 2001
From: NikolajDanger <nikolajgade@live.dk>
Date: Wed, 7 Sep 2022 09:20:32 +0200
Subject: [PATCH] square

---
 square.py | 24 ++++++++++++++++++------
 1 file changed, 18 insertions(+), 6 deletions(-)

diff --git a/square.py b/square.py
index ada28f7..d4e072a 100644
--- a/square.py
+++ b/square.py
@@ -1,15 +1,27 @@
 from time import sleep
 import robot
 
+POWER = 64
+TURN_TIME = 0.75
+DRIVE_TIME = 2.5
+
 def main():
     arlo = robot.Robot()
 
-    turn_time = 3
-    left_speed = 64
-    right_speed = 64
-    print(arlo.go_diff(left_speed, right_speed, 1, 1))
-    sleep(turn_time)
-    print(arlo.stop())
+    try:
+        while True:
+            # Driving forward
+            arlo.go_diff(POWER, POWER, 1, 1)
+            sleep(DRIVE_TIME)
+            arlo.stop()
+
+            # Turning 90 degrees
+            arlo.go_diff(POWER, POWER, 1, 0)
+            sleep(TURN_TIME)
+            arlo.stop()
+
+    except KeyboardInterrupt:
+        arlo.stop()
 
 
 if __name__ == "__main__":