Fixes PlanetUp to work with corp planets.

flexible.py

@@ -1971,14 +1971,24 @@ class PlanetUpScript(object):
         # Hide what's happening from the player
         self.game.to_player = False
 
-        self.queue_player.put("CYQ")  # Computer -> Your Planets -> Quit
+        # self.queue_player.put("CYQ")  # Computer -> Your Planets -> Quit
+        self.queue_player.put("TLQ")  # Team/Corp -> List Corp Planets -> Quit
+        self.corp = True
         self.state = 1
         # self.warpdata = {}
         self.queue_game.put(Boxes.alert("Let me see what I can see here..."))  
 
     def game_prompt(self, prompt):
         log.info("prompt {0} : {1}".format(self.state, prompt))
+        if self.state == 1 and re.match(r"Command \[TL=.* \(\?=Help\)\? :", prompt):
+            # Ok, you're not on a team.  :P
+            self.queue_player.put("CYQ")
         if self.state == 2 and re.match(r"Command \[TL=.* \(\?=Help\)\? :", prompt):
+            if self.corp:
+                self.corp = False
+                self.state = 1
+                self.queue_player.put("CYQ")
+                return
             self.game.to_player = True
             # For now we output the information, and exit
             # self.queue_game.put("{0}\r\n".format(self.planets))
@@ -2234,7 +2244,7 @@ class PlanetUpScript(object):
     def game_line(self, line):
         log.info("line {0} : {1}".format(self.state, line))        
         if self.state == 1:
-            if 'Personal Planet Scan' in line:
+            if 'Personal Planet Scan' in line or 'Corporate Planet Scan' in line:
                 self.state = 2
         elif self.state == 2:
             # Ok, we're in the planet scan part of this