Fixed handling of the case where no links are available in getBestTrade

eyeonus · Aug 6, 2014 · 4e5d6fd · 4e5d6fd
1 parent ef363ac
commit 4e5d6fd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tradecalc.py b/tradecalc.py
@@ -91,9 +91,11 @@ def tryCombinations(self, startCr, tradeList, capacity=None):
 
     def getBestTrade(self, src, dst, startCr, capacity=None):
         if self.debug: print("# %s -> %s with %dcr" % (src, dst, startCr))
+        if not dst in src.stations:
+            raise ValueError("%s does not have a link to %s" % (src, dst))
 
         # Get a list of what we can buy
-        return self.tryCombinations(startCr, src.links[dst.ID], capacity=capacity)
+        return self.tryCombinations(startCr, src.links[dst.ID], capacity)
 
     def getBestHopFrom(self, src, credits, capacity=None):
         """ Determine the best trade run from a given station. """