diff --git a/tradedb.py b/tradedb.py index 27adecbb..14d566ed 100644 --- a/tradedb.py +++ b/tradedb.py @@ -310,6 +310,13 @@ def getShip(self, name): """ Look up a ship by name """ return self.list_search("Ship", name, self.ships, key=lambda item: item.name) + def getTrade(self, src, dst, item): + """ Returns a Trade object describing purchase of item from src for sale at dst. """ + srcStn = self.getStation(src) + dstStn = self.getStation(dst) + trades = srcStn.trades[dstStn.ID] + return trades[item] + def query(self, *args): """ Perform an SQL query on the DB and return the cursor. """ conn = pypyodbc.connect(self.path)