Added optional capcaity parameter to getBestHopFrom

eyeonus · Aug 6, 2014 · a6803b5 · a6803b5
1 parent ced8c73
commit a6803b5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tradecalc.py b/tradecalc.py
@@ -95,13 +95,13 @@ def getBestTrade(self, src, dst, startCr, capacity=None):
         # Get a list of what we can buy
         return self.tryCombinations(startCr, src.links[dst.ID], capacity=capacity)
 
-    def getBestHopFrom(self, src, credits):
+    def getBestHopFrom(self, src, credits, capacity=None):
         """ Determine the best trade run from a given station. """
         if isinstance(src, str):
             src = self.tdb.getStation(src)
         bestDst, bestLoad, bestGainCr = None, None, 0
         for dst in src.stations:
-            trade = self.getBestTrade(src, dst, credits)
+            trade = self.getBestTrade(src, dst, credits, capacity=capacity)
             if trade and trade[1] > bestGainCr:
                 bestDst, bestLoad, bestGainCr = dst, trade[0], trade[1]
         return bestDst, bestLoad, bestGainCr