ColoScript2: Version 2.0 transfer p2p

 p2p = Planet 2 Planet allows you to have a TO and FROM,
 planets and transfer folks.

flexible.py

@@ -2491,12 +2491,12 @@ class PlanetUpScript(object):
             self.observer.load(self.save)
 
 class ColoScript(object):
-    """ Colonise Script
+    """ Colonise Script 1.0
 
         Macro:
         ntel^^n1els^^1^q
 
-        Ask what planet we want to use,
+        Ask what planet we want to load up on,
         Then Move to Terra,
         load up,
         Move to target planet,
@@ -2510,6 +2510,10 @@ class ColoScript(object):
         4 = load people, then move to sector with planet
         5 = init land
         6 = Decide to loop (jump to 3), Unload people
+
+        TODO:
+          * Make it compatable so we can also pull people from other planets,
+            besides Terra (sector 1).
     """
     def __init__(self, game):
         self.game = game
@@ -2582,7 +2586,7 @@ class ColoScript(object):
                 # Are we really getting this? Yup
                 #log.debug("Planet Number: {0} Sector: {1}".format(self.planet_number, self.planet_sector))
                 ask1 = PlayerInput(self.game)
-                d1 = ask1.prompt("How many times ", 3, name="rolls", digits=True)
+                d1 = ask1.prompt("How many times ", 5, name="rolls", digits=True)
                 d1.addCallback(self.loop_chosen)
             else:
                 self.deactivate(True)
@@ -2595,11 +2599,12 @@ class ColoScript(object):
             if self.loops == 0:
                 self.loops = 1
             self.maxloops = self.loops
+            #ask2 = PlayerInput(self.game)
+            #d2 = ask2.prompt("From? ", 5, name="from", digits=True)
+            #d2.addCallback(self.from_chosen)
             self.state = 3
             self.game.to_player = False
-            self.send2game("I") # A neutral command, that won't hose up sending Express Mode? (Yes)
-            # In fact this should allow us to see how many holds are avalible/empty
-            # Jump to state 3 = Move to Terra
+            self.send2game("I")
 
     def game_prompt(self, prompt: str):
         log.debug("P {0} | {1}".format(self.state, prompt))
@@ -2797,6 +2802,349 @@ class ColoScript(object):
                 self.planets[number]['org'] = details[6]
                 self.planets[number]['equ'] = details[7]
 
+class ColoScript2(object):
+    """ ColoScript 2.0
+
+        Goal:
+          * Allow a player to move people from anywhere to anywhere,
+            only planets of course. (Acheived!)
+        
+        States:
+        1 = Computer talks with us giving corp and personal planet info
+        2 = Grab's data and asks series of questions, TO, FROM, LOOPS
+        3 = Move to FROM
+        4 = Identify Population Categories / Is it terra?
+        5 = Load People, then move to planet TO
+        6 = Init Land on TO
+        7 = Decide to loop (Jump back to 3), Unload People
+    """
+    def __init__(self, game):
+        self.game = game
+        self.queue_game = game.queue_game
+        self.queue_player = game.queue_player
+        self.observer = game.observer
+        self.r = Style.RESET_ALL
+        self.c = merge(Style.BRIGHT + Fore.YELLOW)
+        self.nl = "\n\r"
+
+        # My Pants
+        self.state = 1
+        self.corp = True
+        self.planets = {}
+        self.loops = 0
+        self.maxloops = 0
+        self.TO = 0
+        self.FROM = 0
+        
+        # Activate
+        self.prompt = game.buffer
+        self.save = self.observer.save()
+        self.observer.connect('player', self.player)
+        self.observer.connect("prompt", self.game_prompt)
+        self.observer.connect("game-line", self.game_line)
+
+        self.defer = None
+        self.to_player = self.game.to_player
+        self.game.to_player = False
+        self.send2game("TLQ")
+
+    def whenDone(self):
+        self.defer = defer.Deferred()
+        return self.defer
+
+    def deactivate(self, andExit=False):
+        self.state = 0
+        log.debug("ColoScript2.deactivate()")
+        self.game.to_player = True
+        assert(not self.save is None)
+        self.observer.load(self.save)
+        self.save = None
+        if self.defer:
+            if andExit:
+                self.defer.callback({'exit':True})
+            else:
+                self.defer.callback('done')
+            self.defer = None
+
+    def player(self, chunk: bytes):
+        self.deactivate(True)
+
+    def send2game(self, txt):
+        self.queue_player.put(txt)
+
+    def send2player(self, txt):
+        self.queue_game.put(txt)
+
+    def to_chosen(self, choice: str):
+        if choice.strip() == '':
+            self.deactivate(True)
+        else:
+            self.to_number = int(choice)
+            if self.to_number in self.planets:
+                # Ok, this'll work
+                self.TO = self.planets[self.to_number]['sector']
+                self.planet_name = self.planets[self.to_number]['name']
+                # Are we really getting this? Yup
+                #log.debug("Planet Number: {0} Sector: {1}".format(self.planet_number, self.planet_sector))
+                ask1 = PlayerInput(self.game)
+                d1 = ask1.prompt("From: (1 = Terra) ", 3, name="from", digits=True)
+                d1.addCallback(self.from_chosen)
+            else:
+                self.deactivate(True)
+    
+    def from_chosen(self, choice: str):
+        if choice.strip() == '':
+            self.deactivate(True)
+        else:
+            self.from_number = int(choice)
+            if self.from_number in self.planets:
+                # Ok, this'll work
+                self.FROM = self.planets[self.from_number]['sector']
+                self.planet_name = self.planets[self.from_number]['name']
+                # Are we really getting this? Yup
+                #log.debug("Planet Number: {0} Sector: {1}".format(self.planet_number, self.planet_sector))
+                ask1 = PlayerInput(self.game)
+                d1 = ask1.prompt("How many times ", 3, name="rolls", digits=True)
+                d1.addCallback(self.loop_chosen)
+            elif self.from_number == 1:
+                # Yup handle if the user picks to pull from Terra
+                self.FROM = 1
+                ask1 = PlayerInput(self.game)
+                d1 = ask1.prompt("How many times ", 3, name="rolls", digits=True)
+                d1.addCallback(self.loop_chosen)
+            else:
+                self.deactivate(True)
+
+    def loop_chosen(self, choice: str):
+        if choice.strip() == '':
+            self.deactivate(True)
+        else:
+            self.loops = abs(int(choice))
+            if self.loops == 0:
+                self.loops = 1
+            self.maxloops = self.loops
+            #ask2 = PlayerInput(self.game)
+            #d2 = ask2.prompt("From? ", 5, name="from", digits=True)
+            #d2.addCallback(self.from_chosen)
+            self.state = 3
+            self.game.to_player = False
+            self.send2game("I")
+
+    def game_prompt(self, prompt: str):
+        log.debug("P {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))
+            # self.queue_game.put(pformat(self.planets).replace("\n", self.nl) + self.nl)
+
+            if len(self.planets) == 0:
+                # Ok, this is easy.
+                self.queue_game.put(self.nl + Boxes.alert("You don't have any planets?  You poor, poor dear.", base="red"))
+                self.deactivate()
+                return
+
+            # I need this to know if I can just land, or need to move to the planet.
+            # Get current sector from the prompt
+            # Command [TL=00:00:00]:[10202] (?=Help)? :
+            _, _, part = prompt.partition(']:[')
+            sector, _, _ = part.partition(']')
+            self.current_sector = int(sector)
+
+            # A better default is to ask which planet to upgrade.
+
+            tc = merge(Style.BRIGHT + Fore.YELLOW + Back.BLUE)
+            c1 = merge(Style.BRIGHT + Fore.WHITE + Back.BLUE)
+            c2 = merge(Style.BRIGHT + Fore.CYAN + Back.BLUE)
+
+            box = Boxes(44, color=tc)
+            self.queue_game.put(box.top())
+            self.queue_game.put(box.row(tc + "{0:3} {1:6} {2:33}".format(" # ", "Sector", "Planet Name")))
+            self.queue_game.put(box.middle())
+
+            def planet_output(number, sector, name):
+                row = "{0}{1:^3} {2:6} {3}{4:33}".format(c1, number, sector, c2, name)
+                self.queue_game.put(box.row(row))
+
+            for s in sorted(self.planets.keys()):
+                planet_output(s, self.planets[s]['sector'], self.planets[s]['name'])
+
+            self.queue_game.put(box.bottom())
+
+            ask = PlayerInput(self.game)
+            d = ask.prompt("To: ", 3, name="planet", digits=True)
+            d.addCallback(self.to_chosen)
+        elif self.state == 3:
+            # Initalize moving to sector 1, Terra
+            if prompt.startswith('Do you want to engage the TransWarp drive? '):
+                self.queue_player.put("N")
+            elif prompt.startswith('Engage the Autopilot? (Y/N/Single step/Express) [Y]'):
+                self.queue_player.put("E")
+            elif prompt.startswith('Stop in this sector (Y,N,E,I,R,S,D,P,?) (?=Help) [N] ?'):
+                self.queue_player.put("N")
+            elif re.match(r"Command \[TL=.* \(\?=Help\)\? :", prompt):             
+                self.state = 4
+                self.game.to_player = True
+                self.send2game("{0}\r".format(self.FROM))
+                # Move to planet FROM
+        elif self.state == 4:
+            # Are we there yet? (NNY)
+            if prompt.startswith('Do you want to engage the TransWarp drive? '):
+                self.queue_player.put("N")
+            elif prompt.startswith('Engage the Autopilot? (Y/N/Single step/Express) [Y]'):
+                self.queue_player.put("E")
+            elif prompt.startswith('Stop in this sector (Y,N,E,I,R,S,D,P,?) (?=Help) [N] ?'):
+                self.queue_player.put("N")
+            elif re.match(r"Command \[TL=.* \(\?=Help\)\? :", prompt):             
+                self.state = 5
+                self.send2game("L")
+        elif self.state == 5:
+            if self.FROM == 1: # Is our FROM planet Terra?
+                if prompt.startswith('Land on which planet <Q to abort> ?'):
+                    self.send2game("1\r\r\r{0}\r".format(self.TO))
+                    self.state = 6
+                    # Planetary Scanner Detected, Move to sector with planet
+                elif prompt.startswith('Do you wish to (L)eave or (T)ake Colonists? [T] (Q to leave)'):
+                    self.send2game("\r\r{0}\r".format(self.TO))
+                    self.state = 6
+                    # No Planetary Scanner Detected, Move to sector with planet
+                elif prompt.startswith('Engage the Autopilot? (Y/N/Single step/Express) [Y]'):
+                    self.send2game("E")
+                    self.state = 4
+                    # We are not at terra, go back
+            else:
+                if prompt.startswith('Land on which planet <Q to abort> ?'):
+                    self.send2game("{0}\rs\rt1\rq{1}\r".format(self.from_number, self.TO))
+                    self.state = 6
+                    # Planetary Scanner Detected, Move to sector with planet
+                elif prompt.startswith('Planet command (?=help) [D] '):
+                    self.send2game("s\rt1\rq{0}\r".format(self.TO))
+                    self.state = 6
+                elif prompt.startswith('Engage the Autopilot? (Y/N/Single step/Express) [Y]'):
+                    self.send2game("E")
+                    self.state = 4
+                    # We are not at terra, go back
+        elif self.state == 6:
+            if prompt.startswith('Do you want to engage the TransWarp drive? '):
+                self.queue_player.put("N")
+            elif prompt.startswith('Engage the Autopilot? (Y/N/Single step/Express) [Y]'):
+                self.queue_player.put("E")
+            elif prompt.startswith('Stop in this sector (Y,N,E,I,R,S,D,P,?) (?=Help) [N] ?'):
+                self.queue_player.put("N")
+            elif re.match(r"Command \[TL=.* \(\?=Help\)\? :", prompt):             
+                self.state = 6
+                self.send2game("L")
+            elif prompt.startswith('Land on which planet <Q to abort> ?'):
+                self.state = 7
+                self.send2game("{0}\r".format(self.to_number))
+                # Planetary Scanner Detected selecting planet number
+            elif prompt.startswith('Planet command (?=help) [D] '):
+                self.state = 7
+                self.send2game("D")
+                # No Planetary Scaner skip on
+        elif self.state == 7:
+            if prompt.startswith('Engage the Autopilot? (Y/N/Single step/Express) [Y]'):
+                self.state = 6
+                self.send2game("E")
+                # Missed moving jump back to state 6
+            if prompt.startswith('Land on which planet <Q to abort> ?'):
+                self.queue_player.put("{0}\r".format(self.to_number))
+                # Planetary Scanner Detected selecting planet number
+            if prompt.startswith('Planet command (?=help) [D] '):
+                # Unload people and process the loop
+                self.loops -= 1
+                if self.loops:
+                    self.state = 3
+                    self.send2game("S\r\r1\rQ")
+                    # Jump to state 3 we are not done
+                else:
+                    self.send2game("S\r\r1\rQ")
+                    self.send2player("\r" + Boxes.alert("Completed ({0})".format(self.maxloops)))
+                    self.deactivate(True)
+                    # Ok we are done
+    
+    def game_line(self, line: str):
+        log.debug("L {0} | {1}".format(self.state, line))
+        # IF at any state we see turns left lets grab it
+        if 'turns left' in line:
+                work = line[19:].replace(' turns left.', '').strip()
+                self.turns = work
+                log.debug("TURNS LEFT: {0}".format(self.turns))
+                if int(self.turns) < 200:
+                    self.send2player("\r" + Boxes.alert("Low Turns! ({0})".format(self.turns)))
+                    self.deactivate(True)
+        # IF at any state we see how many holds avalible let's get that
+        if 'Total Holds' in line:
+            work = line[16:].replace('-', '').replace('=', ' ').split()
+            self.total_holds = int(work[0])
+            count = 0
+            for w in work:
+                if(w != 'Empty'):
+                    count += 1
+                elif(w == 'Empty'):
+                    count += 1
+                    self.holds = int(work[count])
+                    log.debug("EMPTY HOLDS = {0}".format(self.holds))
+                    self.holds_percent = int((self.holds / self.total_holds) * 100.0)
+                    log.debug("HOLDS PERCENT = {0}%".format(self.holds_percent))
+                    if(self.holds < self.total_holds):
+                        self.send2player("\r" + Boxes.alert("You need {0} holds empty! ({1} Empty)".format(self.total_holds, self.holds)))
+                        self.deactivate(True)
+        if "There aren't that many on the planet!" in line:
+            self.send2player("\r" + Boxes.alert("We're missing people!"))
+            self.deactivate(True)
+        if "There isn't room on the planet for that many!" in line:
+            self.send2player("\r" + Boxes.alert("We have to many people!"))
+            self.deactivate(True)
+
+        # Now back to our scheduled program
+        if self.state == 1:
+            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
+            # 10202   #3    Home of Bugz             Class M, Earth Type        No Citadel]
+            if '#' in line:
+                # Ok, we have a planet detail line.
+                # There's an extra T when the planet is maxed out.
+                if line[7] == 'T':
+                    line = line[:7] + ' ' + line[8:]
+
+                detail, _, _ = line.partition('Class')
+                detail = detail.strip() # Sector  #X  Name of planet
+                sector, number, name = re.split(r'\s+', detail, 2)
+                sector = int(sector)
+                number = int(number[1:])
+                self.last_seen = number
+                self.planets[number] = {"sector": sector, "name": name}
+                log.info("Planet # {0} in {1} called {2}".format( number, sector, name))
+            if '---' in line:
+                number = self.last_seen
+                # Ok, take the last_seen number, and use it for this line
+                # [ Sector  Planet Name    Ore  Org  Equ   Ore   Org   Equ   Fighters    Citadel]
+                # [ Shields Population    -=Productions=-  -=-=-=-=-On Hands-=-=-=-=-    Credits]
+                # [ 10202   #3    Home of Bugz             Class M, Earth Type        No Citadel]
+                # [   ---   (1M)           144   49   26   145    75    12         10          0]
+                details = re.split(r"\s+", line.strip())
+                # OK:  Likely, I'm not going to use these numbers AT ALL.  
+                self.planets[number]['population'] = details[1].replace('(', '').replace(')', '')
+                # Ok, there's going to have to be some sort of modifier (K, M, etc.)
+                # to these numbers.  Beware!
+                self.planets[number]['ore'] = details[5]
+                self.planets[number]['org'] = details[6]
+                self.planets[number]['equ'] = details[7]
+
+    
+
+
 class ProxyMenu(object):
     """ Display ProxyMenu 
     
@@ -3418,7 +3766,8 @@ class ProxyMenu(object):
         menu_item("2", "Explore (Strange new sectors)")
         menu_item("3", "Space... the broken script...")
         menu_item("4", "Upgrade Planet")
-        menu_item("5", "Colonize Planet")
+        #menu_item("5", "Colonize Planet")
+        menu_item("5", "Colonize Planet v2.0")
         menu_item("X", "eXit")
         self.queue_game.put(box.bottom())
         self.queue_game.put("   " + c1 + "-=>" + self.r + " ")
@@ -3483,7 +3832,8 @@ class ProxyMenu(object):
             return
         elif key == '5':
             self.queue_game.put(self.c + key + self.r + self.nl)
-            colo = ColoScript(self.game)
+            #colo = ColoScript(self.game)
+            colo = ColoScript2(self.game)
             d = colo.whenDone()
             d.addCallback(self.deactivate_scripts_menu)
             d.addErrback(self.deactivate_scripts_menu)