TradeQueryGenerator.lua.rej

diff a/src/Classes/TradeQueryGenerator.lua b/src/Classes/TradeQueryGenerator.lua	(rejected hunks)
@@ -10,6 +10,7 @@ local m_max = math.max
 local s_format = string.format
 local t_insert = table.insert
 local tradeHelpers = LoadModule("Classes/TradeHelpers")
+local utils = LoadModule("Modules/Utils")
 
 -- string are an any type while tables require all fields to be matched with type and subType require both to be matched exactly. [1] type, [2] subType, subType is optional and must be nil if not present.
 local tradeCategoryNames = {
@@ -395,11 +396,6 @@ function TradeQueryGeneratorClass:InitMods()
 		error("Error received from api/trade2/data/stats: "..body.error.message)
 	end
 
-	local f = io.open("./Data/TradeSiteStats.lua", "w")
-	if not f then
-		error("Could not open file for writing trade stat data")
-	end
-
 	for catIdx, _ in ipairs(body.result) do
 		table.sort(body.result[catIdx].entries, function(a, b)
 			if a.text == b.text then
@@ -409,14 +405,8 @@ function TradeQueryGeneratorClass:InitMods()
 		end)
 	end
 
-	local template = [[-- This file is automatically downloaded, do not edit!
--- Trade site stat data (c) Grinding Gear Games
--- https://www.pathofexile.com/api/trade2/data/stats
--- spell-checker: disable
-return %s
--- spell-checker: enable]]
-	f:write(s_format(template, stringify(body.result)))
-	f:close()
+	local description = "This file contains the trade site data from https://www.pathofexile.com/api/trade2/data/stats"
+	utils.saveTableToFile("./Data/TradeSiteStats.lua", body.result, description)
 
 	self.modData = {
 		["Explicit"] = { },
@@ -636,19 +626,12 @@ return %s
 		end
 	end
 
-	local queryModsFile = io.open(queryModFilePath, 'w')
-	queryModsFile:write([[-- This file is automatically generated, do not edit!
--- Stat data (c) Grinding Gear Games
-
--- This file contains categories of stats, mapped from trade hash to details
--- relevant for generating search weights Note that the trade site requires a
--- prefix of e.g. explicit.stat_{hash}. See
--- https://www.pathofexile.com/api/trade2/data/stats for a list of all trade
--- site stats.
-
-]])
-	queryModsFile:write("return " .. stringify(self.modData))
-	queryModsFile:close()
+	local qmDescription = [[This file contains categories of stats, mapped from trade hash to details
+relevant for generating search weights Note that the trade site requires a
+prefix of e.g. explicit.stat_{hash}. See
+TradeSiteStats.lua for a list of all trade
+site stats.]]
+	utils.saveTableToFile(queryModFilePath, self.modData, qmDescription)
 end
 
 function TradeQueryGeneratorClass:GenerateModWeights(modsToTest)