Editing Module:Person/Flag

local Flags = require('Module:Flags')
local pFlag = {}
local cargo = mw.ext.cargo

function pFlag.makeFlag(flagArg, playerId, gameCategory) 
	-- Look up Cargo table for nationality string
	if flagArg then
		return Flags.icon( flagArg, gameCategory)
	end
	local tables = 'People'
	local fields = '_pageName, nationality'
	local cargoArgs = {
		where = '_pageName = "' .. playerId .. '"'
	}
	local results = cargo.query(tables, fields, cargoArgs)

	if #results > 0 then
		return Flags.icon( mw.text.split(results[1].nationality, ',')[1], gameCategory)
	else 
		return nil
	end
end

return pFlag