Make pretty

soccer/gameplay/play.py

@@ -33,7 +33,6 @@ def is_restart(cls):
     def handles_goalie(cls):
         return False
 
-
     ## Override to allow a play to run during the stopped state
     # This is reserved only for very special plays that need to run during the
     # stopped state.  Most plays should give up control during the stopped state

soccer/gameplay/root_play.py

@@ -81,19 +81,22 @@ def execute_running(self):
             if self.play != None:
                 if self.play.__class__ not in map(lambda tup: tup[0],
                                                   enabled_plays_and_scores):
-                    logging.info("Current play '" + self.play.__class__.
-                                 __name__ + "' no longer enabled, aborting")
+                    logging.info("Current play '" +
+                                 self.play.__class__.__name__ +
+                                 "' no longer enabled, aborting")
                     self.play.terminate()
                     self.play = None
                 elif self.play.is_done_running():
-                    logging.info("Current play '" + self.play.__class__.
-                                 __name__ + "' finished running")
+                    logging.info("Current play '" +
+                                 self.play.__class__.__name__ +
+                                 "' finished running")
                     if self.play.is_restart:
                         self._currently_restarting = False
                     self.play = None
                 elif self.play.__class__.score() == float("inf"):
-                    logging.info("Current play '" + self.play.__class__.
-                                 __name__ + "' no longer applicable, ending")
+                    logging.info("Current play '" +
+                                 self.play.__class__.__name__ +
+                                 "' no longer applicable, ending")
                     self.play.terminate()
                     self.play = None
 

soccer/gameplay/single_robot_composite_behavior.py

@@ -20,7 +20,6 @@ def __init__(self, continuous=False, autorestart=lambda: True):
         composite_behavior.CompositeBehavior.__init__(self,
                                                       continuous=continuous)
         self.autorestart = autorestart
-
 
     @property
     def autorestart(self):

soccer/gameplay/tactics/positions/coach.py

@@ -9,6 +9,7 @@
 import random
 import role_assignment
 
+
 ## Motivates, encourages, and directs the team.
 class Coach(single_robot_composite_behavior.SingleRobotCompositeBehavior):
 
@@ -19,19 +20,18 @@ class Coach(single_robot_composite_behavior.SingleRobotCompositeBehavior):
     TheirScore = 0
 
     STRATEGIZING_IDEAS = [
-        "*incoherent mumbling*",
-        "They'll never see it coming!",
+        "*incoherent mumbling*", "They'll never see it coming!",
         "*wild robot appendage gestures*",
         "And that's when we pull out the soldering irons",
-        "*coach violently hawks a loogie*",
-        "Sweep the Leg",
+        "*coach violently hawks a loogie*", "Sweep the Leg",
         "Is it necesarry for me to drink my own urine? No, but I do it anyway because its sterile and I like the taste",
         "There's nothing in the rules about golf clubs",
         "*coach points violently at the ref*",
         "*coach slyly passes the player an extra battery*",
         "Yes, the soup last night was delicious",
         "Time for another rule in our name...",
-        "THIS... STATEMENT... IS... FALSE"]
+        "THIS... STATEMENT... IS... FALSE"
+    ]
 
     class State(enum.Enum):
         watching = 0
@@ -40,7 +40,7 @@ class State(enum.Enum):
         strategizing = 3
 
     def __init__(self):
-        super().__init__(continuous=True)#self.State != self.State.strategizing)
+        super().__init__(continuous=True)  #self.State != self.State.strategizing)
         self.spin_angle = 0
 
         for state in Coach.State:
@@ -53,24 +53,28 @@ def __init__(self):
         self.add_transition(self.State.watching, self.State.celebrating,
                             lambda: self.our_score_increased(), 'goal scored')
         self.add_transition(self.State.celebrating, self.State.watching,
-                            lambda: self.spin_angle > Coach.MaxSpinAngle, 'done with celebration')
+                            lambda: self.spin_angle > Coach.MaxSpinAngle,
+                            'done with celebration')
 
         self.add_transition(self.State.watching, self.State.motivating,
-                            lambda: self.their_score_increased(), 'Let a goal through')
+                            lambda: self.their_score_increased(),
+                            'Let a goal through')
         self.add_transition(self.State.motivating, self.State.watching,
                             lambda: True, 'done talking')
 
         self.add_transition(self.State.watching, self.State.strategizing,
                             lambda: main.game_state().is_stopped(), 'Timeout')
         self.add_transition(self.State.strategizing, self.State.watching,
-                            lambda: not main.game_state().is_stopped(), 'Timeout over')
+                            lambda: not main.game_state().is_stopped(),
+                            'Timeout over')
 
     def our_score_increased(self):
         if (Coach.OurScore < main.game_state().our_score):
             return True
         else:
             Coach.OurScore = main.game_state().our_score
             return False
+
     def their_score_increased(self):
         if (Coach.TheirScore < main.game_state().their_score):
             return True
@@ -83,8 +87,9 @@ def on_enter_running(self):
         self.add_subbehavior(move, 'coach')
 
     def on_enter_watching(self):
-        self.subbehavior_with_name('coach').pos = robocup.Point(-constants.Field.Width / 2 - constants.Robot.Radius * 2,
-                                   constants.Field.Length / 3)
+        self.subbehavior_with_name('coach').pos = robocup.Point(
+            -constants.Field.Width / 2 - constants.Robot.Radius * 2,
+            constants.Field.Length / 3)
 
     def on_exit_running(self):
         self.remove_all_subbehaviors()
@@ -124,10 +129,15 @@ def on_enter_motivating(self):
 
     def on_enter_strategizing(self):
         #pick a robot to talk to
-        target_bot = random.randint(0, ((len(main.our_robots()) - 1) if (main.our_robots() is not None) else 0));
-        self.subbehavior_with_name('coach').pos = (main.our_robots()[target_bot].pos + robocup.Point(-constants.Robot.Radius * 2, constants.Robot.Radius * 2))
-        self.subbehavior_with_name('coach').threshold = constants.Robot.Radius * 2;
-        print("\n\n Alright Number " + str(target_bot) + " here is the plan:");
+        target_bot = random.randint(0,
+                                    ((len(main.our_robots()) - 1) if
+                                     (main.our_robots() is not None) else 0))
+        self.subbehavior_with_name('coach').pos = (
+            main.our_robots()[target_bot].pos + robocup.Point(
+                -constants.Robot.Radius * 2, constants.Robot.Radius * 2))
+        self.subbehavior_with_name(
+            'coach').threshold = constants.Robot.Radius * 2
+        print("\n\n Alright Number " + str(target_bot) + " here is the plan:")
 
     def execute_strategizing(self):
         #Stops coach from talking too much
@@ -139,9 +149,11 @@ def execute_strategizing(self):
             print(random.choice(Coach.STRATEGIZING_IDEAS))
 
     def on_exit_strategizing(self):
-        print("\n*coach gives the player an invigorating butt slap*\nGo get'em")
+        print(
+            "\n*coach gives the player an invigorating butt slap*\nGo get'em")
 
     @staticmethod
+
     ## Returns a fortune.
     def fortune_wrapper():
         try:
@@ -160,8 +172,7 @@ def fortune_wrapper():
 
     def role_requirements(self):
         reqs = super().role_requirements()
-        for req in role_assignment.iterate_role_requirements_tree_leaves(
-            reqs):
+        for req in role_assignment.iterate_role_requirements_tree_leaves(reqs):
             #There is only one coach! Not just any robot can be coach
             req.robot_change_cost = 30.0
         return reqs