Update analyzer.py, central_command.py and five other files

nullptr/missions/__init__.py

@@ -2,17 +2,22 @@ from nullptr.missions.survey import SurveyMission
 from nullptr.missions.mine import MiningMission
 from nullptr.missions.haul import HaulMission
 from nullptr.missions.travel import TravelMission
+from nullptr.missions.probe import ProbeMission
 
-
-def create_mission(mtype, ship, store, api):
+def get_mission_class( mtype):
   types = {
     'survey': SurveyMission,
     'mine': MiningMission,
     'haul': HaulMission,
-    'travel': TravelMission
+    'travel': TravelMission,
+    'probe': ProbeMission
   }
   if mtype not in types:
-    logging.warning(f'invalid mission type {mtype}')
-    return
-  m = types[mtype](ship, store, api)
+    raise ValueError(f'invalid mission type {mtype}')
+  return types[mtype]
+  
+def create_mission(mtype, ship, store, api):
+  typ = get_mission_class(mtype)
+  m = typ(ship, store, api)
   return m
+

nullptr/missions/base.py

@@ -25,6 +25,8 @@ class MissionParam:
       return str(val)
     elif self.cls == int:
       return int(val)
+    elif self.cls == list:
+      return [i.strip() for i in val.split(',')]
     elif issubclass(self.cls, Base):
       data = store.get(self.cls, val)
       if data is None:

nullptr/missions/probe.py

@@ -0,0 +1,33 @@
+from nullptr.missions.base import BaseMission, MissionParam
+from nullptr.models.waypoint import Waypoint
+
+class ProbeMission(BaseMission):
+  def start_state(self):
+    return 'next-hop'
+  
+  @classmethod
+  def params(cls):
+    return {
+      'hops': MissionParam(list, True),
+      'next-hop': MissionParam(int, True, 0)
+    }
+  
+  def steps(self):
+    return {
+      'next-hop': (self.step_next_hop, 'travel-to'),
+      **self.travel_steps('to', 'site', 'market'),
+      'market': (self.step_market, 'next-hop'),
+ 
+    }
+  
+  def step_market(self):
+    loc = self.ship.location()
+    self.api.marketplace(loc)
+  
+  def step_next_hop(self):
+    hops = self.st('hops')
+    next_hop = self.st('next-hop')
+    hop = hops[next_hop]
+    self.sts('site', hop)
+    self.sts('next-hop', (next_hop+1) % len(hops))
+