from .base import Base from time import time from nullptr.util import * from nullptr.models import Waypoint import os class Ship(Base): def define(self): self.cargo:dict = {} self.mission_state:dict = {} self.status:str = '' self.cargo_capacity:int = 0 self.cargo_units:int = 0 self.location = None self.cooldown:int = 0 self.arrival:int = 0 self.fuel_current:int = 0 self.fuel_capacity:int = 0 self.mission:str = None self.mission_status:str = 'init' self.role = None self.frame = '' self.speed = "CRUISE" self._log_file = None def log(self, m): if self._log_file is None: fn = os.path.join(self.store.data_dir, f'{self.symbol}.{self.ext()}.log') self._log_file = open(fn, 'a') ts = int(time()) m = m.strip() self._log_file.write(f'{ts} {m}\n') self._log_file.flush() @classmethod def ext(self): return 'shp' def range(self): if self.fuel_capacity == 0: return 100000 return self.fuel_capacity def update(self, d): self.seta('status', d, 'nav.status') self.seta('speed', d, "nav.flightMode") self.seta('frame', d, 'frame.name') getter = self.store.getter(Waypoint, create=True) self.seta('location', d, 'nav.waypointSymbol', interp=getter) self.seta('cargo_capacity', d, 'cargo.capacity') self.seta('cargo_units', d, 'cargo.units') self.seta('fuel_capacity', d, 'fuel.capacity') self.seta('fuel_current', d,'fuel.current') cargo = sg(d, 'cargo.inventory') if cargo is not None: self.load_cargo(cargo) self.seta('cooldown', d, 'cooldown.expiration', parse_timestamp) self.seta('arrival', d, 'nav.route.arrival', parse_timestamp) def tick(self): if self.status == 'IN_TRANSIT' and self.arrival < time(): self.status = 'IN_ORBIT' def is_cooldown(self): return self.cooldown > time() def is_travelling(self): return self.status == 'IN_TRANSIT' def set_mission_state(self, nm, val): self.mission_state[nm] = val self.store.dirty(self) def get_cargo(self, typ): if typ not in self.cargo: return 0 return self.cargo[typ] def take_cargo(self, typ, amt): if typ not in self.cargo: return if self.cargo[typ] <= amt: del self.cargo[typ] else: self.cargo[typ] -= amt self.cargo_units = sum(self.cargo.values()) def load_cargo(self, cargo): result = {} total = 0 for i in cargo: symbol = must_get(i, 'symbol') units = must_get(i, 'units') result[symbol] = units total += units self.cargo_units = total self.cargo = result def deliverable_cargo(self, contract): result = [] for d in contract.deliveries: if self.get_cargo(d['trade_symbol']) > 0: result.append(d['trade_symbol']) return result def nondeliverable_cargo(self, contract): cargo = self.cargo.keys() deliveries = [d['trade_symbol'] for d in contract.deliveries] garbage = [c for c in cargo if c not in deliveries] return garbage def update_timers(self): if self.status == 'IN_TRANSIT' and self.arrival < time(): self.status = 'IN_ORBIT' def f(self, detail=1): self.update_timers() arrival = int(self.arrival - time()) cooldown = int(self.cooldown - time()) role = self.role if role is None: role = 'none' mstatus = self.mission_status if mstatus == 'error': mstatus = mstatus.upper() status = self.status.lower() if status.startswith('in_'): status = status[3:] if detail < 2: r = self.symbol elif detail == 2: symbol = self.symbol.split('-')[1] r = f'{symbol:<2} {role:7} {mstatus:8} {str(self.location):11}' if self.is_travelling(): r += f' [A: {arrival}]' if self.is_cooldown(): r += f' [C: {cooldown}]' else: r = f'== {self.symbol} {self.frame} ==\n' r += f'Role: {role}\n' r += f'Mission: {self.mission} ({mstatus})\n' for k, v in self.mission_state.items(): if type(v) == list: v = f'[{len(v)} items]' r += f' {k}: {v}\n' adj = 'to' if self.status == 'IN_TRANSIT' else 'at' r += f'Status {self.status} {adj} {self.location}\n' r += f'Fuel: {self.fuel_current}/{self.fuel_capacity}\n' r += f'Speed: {self.speed}\n' r += f'Cargo: {self.cargo_units}/{self.cargo_capacity}\n' for res, u in self.cargo.items(): r += f' {res}: {u}\n' if self.is_travelling(): r += f'Arrival: {arrival} seconds\n' if self.is_cooldown(): r += f'Cooldown: {cooldown} seconds \n' return r