Module:HeroData/headcount

From Deadlock Wiki
Revision as of 02:11, 25 October 2024 by Sur (talk | contribs) (ported from HeroData module; now supports released and hero lab heroes separately)
(diff) ← Older revision | Latest revision (diff) | Newer revision → (diff)
Jump to navigation Jump to search

Documentation for this module may be created at Module:HeroData/headcount/doc

local p = {}

-- {{Heroes count}} access point
function p.get_headcount(frame)
	local hero_type = frame.args[1]
	
	local iter = 0
	for _, v in pairs(heroes_data) do
		if hero_type == 'Released' then
			if (v["InDevelopment"] == false and v["IsDisabled"] == false and v["Name"] ~= nil) then
				iter = iter + 1	
			end
		elseif hero_type == 'HeroLabs' then
			if ((v["InDevelopment"] == true or v["InHeroLabs"] == true) and v["Name"] ~= nil) then
				iter = iter + 1	
			end
		else
			return "get_headcount()'s hero_type parameter must be either 'Released' or 'HeroLabs"
		end
	end
	return iter
end

return p