Update api.py, command_line.py and five other files

nullptr/models/base.py

@@ -4,6 +4,7 @@ from nullptr.util import sg
 
 @dataclass
 class Base:
+  identifier = 'symbol'
   symbol: str
   store: object
   
@@ -17,11 +18,13 @@ class Base:
   def __eq__(self, other):
     return self.symbol == other.symbol and type(self) == type(other)
     
-  def seta(self, attr,  d, name=None):
+  def seta(self, attr,  d, name=None, interp=None):
     if name is None:
       name = attr
     val = sg(d, name)
     if val is not None:
+      if interp is not None:
+        val = interp(val)
       setattr(self, attr, val)
   
   def setlst(self, attr, d, name, member):
@@ -37,6 +40,9 @@ class Base:
   
   def update(self, d):
     pass
+  
+  def load(self, d):
+    self.__dict__ = d
     
   def dict(self):
     r = {}

nullptr/models/contract.py

@@ -0,0 +1,59 @@
+
+from time import time
+from nullptr.util import *
+from .base import Base
+from typing import List
+
+class Contract(Base):
+  identifier = 'id'
+  type: str
+  deliveries: List
+  accepted: bool
+  fulfilled: bool
+  expires: int
+  expires_str: str
+  pay: int
+  
+  @classmethod
+  def ext(cls):
+    return 'cnt'
+  
+  def path(self):
+    return f'contracts/{self.symbol}.{self.ext()}'
+  
+  def is_expired(self):
+    return time() > self.expires
+  
+  def api_dict(self):
+    return {
+      'id': self.symbol,
+      'expiration': self.expires_str,
+    }
+  
+  def update(self, d):
+    self.seta('expires',d, 'terms.deadline',parse_timestamp)
+    self.seta('expires_str', d,'terms.deadline')
+    self.seta('accepted', d, 'accepted')
+    self.seta('fulfilled', d, 'fulfilled')
+    self.seta('type', d, 'type')
+    self.pay = mg(d, 'terms.payment.onAccepted') + mg(d, 'terms.payment.onFulfilled')
+    deliveries = must_get(d, 'terms.deliver')
+    self.deliveries = []
+    for e in deliveries:
+      delivery = {}
+      delivery['trade_symbol'] = must_get(e, 'tradeSymbol')
+      delivery['units_fulfilled'] = must_get(e, 'unitsFulfilled')
+      delivery['units_required'] = must_get(e, 'unitsRequired')
+      delivery['destination'] = must_get(e, 'destinationSymbol')
+      self.deliveries.append(delivery)
+
+  def f(self, detail=1):
+    hours = int(max(0, self.expires - time()) / 3600)
+    accepted = 'A' if self.accepted else '-'
+    fulfilled = 'F' if self.fulfilled else '-'
+    result = f'{self.symbol} {hours}h {accepted}{fulfilled}'
+    if detail > 1:
+      result += '\n'
+      for d in self.deliveries:
+        result += f"({d['units_fulfilled']} / {d['units_required']}) {d['trade_symbol']} to {d['destination']}"
+    return result

nullptr/models/ship.py

@@ -3,13 +3,6 @@ from time import time
 from nullptr.util import *
 from dataclasses import dataclass
 
-@dataclass
-class InventoryItem:
-  symbol: str
-  units: int
-  def __str__(self):
-    return self.symbol + ': ' + str(self.units)
-   
 class Ship(Base):  
   cargo:dict = {}
   mission_state:dict = {}
@@ -38,7 +31,6 @@ class Ship(Base):
   def update(self, d):
     self.seta('status', d, 'nav.status')
     self.seta('location_str', d, 'nav.waypointSymbol')
-    
     self.seta('cargo_capacity', d, 'cargo.capacity')
     self.seta('cargo_units', d, 'cargo.units')
     self.seta('fuel_capacity', d, 'fuel.capacity')
@@ -46,13 +38,9 @@ class Ship(Base):
     cargo = sg(d, 'cargo.inventory')
     if cargo is not None:
       self.load_cargo(cargo)
-    cooldown = sg(d, 'cooldown.expiration')
-    if cooldown:
-      self.cooldown = parse_timestamp(cooldown)
-    arrival = sg(d, 'nav.route.arrival')
-    if arrival:
-      self.arrival = parse_timestamp(arrival)
-      
+    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'
@@ -66,26 +54,26 @@ class Ship(Base):
   def get_cargo(self, typ):
     if typ not in self.cargo:
       return 0
-    return self.cargo[typ].units
+    return self.cargo[typ]
     
   def load_cargo(self, cargo):
     result = {}
     for i in cargo:
       symbol = must_get(i, 'symbol')
       units = must_get(i, 'units')
-      result[symbol] = InventoryItem(symbol, units)
+      result[symbol] = units
     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)
+      if self.get_cargo(d['trade_symbol']) > 0:
+        result.append(d['trade_symbol'])
     return result
     
   def nondeliverable_cargo(self, contract):
-    cargo = [c.symbol for c in self.cargo.values()]
-    deliveries = [d.trade_symbol for d in contract.deliveries]
+    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