Refresh .prices file when removing a station that had items listed

eyeonus · Mar 28, 2015 · c81229e · c81229e
1 parent bdb24cb
commit c81229e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/CHANGES.txt b/CHANGES.txt
@@ -3,6 +3,7 @@ TradeDangerous, Copyright (C) Oliver "kfsone" Smith, July 2014
 ==============================================================================
 
 v6.15.0 Mar 27 2015
+. (kfsone) Refresh .prices file after removing a station that had items
 . (kfsone) Maddavo Plugin
    - Added support for "corrections"
      "--opt=corrections" and "--opt=csvs" will read Maddavo's correction data

diff --git a/commands/station_cmd.py b/commands/station_cmd.py
@@ -6,6 +6,7 @@
 from tradedb import System, Station
 from tradedb import TradeDB
 
+import cache
 import csvexport
 import difflib
 import re
@@ -287,6 +288,7 @@ def removeStation(tdb, cmdenv, station):
     db.commit()
     cmdenv.NOTE("{} (#{}) removed from {} database.",
             station.name(), station.ID, tdb.dbPath)
+    cmdenv.stationItemCount = station.itemCount
     return True
 
 
@@ -300,6 +302,12 @@ def checkResultAndExportStations(tdb, cmdenv, result):
 
     lines, csvPath = csvexport.exportTableToFile(tdb, cmdenv, "Station")
     cmdenv.NOTE("{} updated.", csvPath)
+
+    if cmdenv.remove:
+        if cmdenv.stationItemCount:
+            cmdenv.NOTE("Station had items, regenerating .prices file")
+            cache.regeneratePricesFile(tdb, cmdenv)
+
     return None