Fixed station argument for shipvendor

eyeonus · Jan 26, 2015 · 3421624 · 3421624
1 parent 5102470
commit 3421624
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/commands/shipvendor_cmd.py b/commands/shipvendor_cmd.py
@@ -20,7 +20,7 @@
 epilog=None
 arguments = [
     ParseArgument(
-        'station',
+        'origin',
         help='Specify the full name of the station (SYS NAME/STN NAME is also supported).',
         metavar='STATIONNAME',
         type=str,
@@ -119,10 +119,11 @@ def checkShipPresent(tdb, ship, station):
 
 def run(results, cmdenv, tdb):
 
-    try:
-        station = tdb.lookupStation(cmdenv.station)
-    except LookupError:
-        raise CommandLineError("Unrecognized Station: {}".format(cmdenv.station))
+    station = cmdenv.startStation
+    if not isinstance(station, Station):
+        raise CommandLineError("{} is a system, not a station".format(
+            station.name()
+        ))
 
     try:
         ship = tdb.lookupShip(cmdenv.ship)