74 lines
1.7 KiB
Python
74 lines
1.7 KiB
Python
|
from nullptr.store import store
|
||
|
|
from nullptr.models.ship import Ship
|
||
|
|
from nullptr.mission import *
|
||
|
|
from random import choice
|
||
|
|
from time import sleep
|
||
|
|
|
||
|
|
class CentralCommand:
|
||
|
|
def __init__(self, api):
|
||
|
|
self.missions = {}
|
||
|
|
self.stopping = False
|
||
|
|
self.api = api
|
||
|
|
self.update_missions()
|
||
|
|
|
||
|
|
def get_ready_missions(self):
|
||
|
|
result = []
|
||
|
|
for ship, mission in self.missions.items():
|
||
|
|
if mission.is_ready():
|
||
|
|
result.append(ship)
|
||
|
|
return result
|
||
|
|
|
||
|
|
def tick(self):
|
||
|
|
missions = self.get_ready_missions()
|
||
|
|
if len(missions) == 0: return
|
||
|
|
ship = choice(missions)
|
||
|
|
mission = self.missions[ship]
|
||
|
|
mission.step()
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
self.update_missions()
|
||
|
|
while not self.stopping:
|
||
|
|
self.tick()
|
||
|
|
self.api.save()
|
||
|
|
sleep(0.5)
|
||
|
|
self.stopping = False
|
||
|
|
|
||
|
|
def stop(self):
|
||
|
|
self.stopping = True
|
||
|
|
|
||
|
|
def set_mission_param(self, ship, nm, val):
|
||
|
|
if ship not in self.missions:
|
||
|
|
print('set a mission for this ship first')
|
||
|
|
return
|
||
|
|
mission = self.missions[ship]
|
||
|
|
params = mission.params()
|
||
|
|
if not nm in params:
|
||
|
|
print(f'{nm} is not a valid param')
|
||
|
|
return
|
||
|
|
param = params[nm]
|
||
|
|
try:
|
||
|
|
parsed_val = param.parse(val)
|
||
|
|
except ValueError as e:
|
||
|
|
print(e)
|
||
|
|
return
|
||
|
|
print('ok')
|
||
|
|
ship.mission_state[nm] = parsed_val
|
||
|
|
|
||
|
|
def update_missions(self):
|
||
|
|
for s in store.all(Ship):
|
||
|
|
if s.mission is None:
|
||
|
|
if s in self.missions:
|
||
|
|
self.stop_mission(s)
|
||
|
|
elif s not in self.missions:
|
||
|
|
self.start_mission(s)
|
||
|
|
|
||
|
|
def start_mission(self, s):
|
||
|
|
mtype = s.mission
|
||
|
|
m = create_mission(mtype, s, self.api)
|
||
|
|
self.missions[s] = m
|
||
|
|
return m
|
||
|
|
|
||
|
|
def stop_mission(self, s):
|
||
|
|
if s in self.missions:
|
||
|
|
del self.missions[s]
|