Модул:Infobox

Модул за аутокатегоризацију на основу података прослеђених у инфокутију за предмет. Овај модул такође довлачи потребне податке за рад из модула са подацима.

---

local p = {}

local data = mw.loadData('Module:Infobox/data')

local function cat(cats, name)
	table.insert(cats, '[[Категорија:')
	table.insert(cats, name)
	table.insert(cats, ']]')
end

local function decomposeDepartments(arg)
	local decomposed = {}
	for _, v in pairs(mw.text.split(arg, ', ', true)) do
		decomposed[v] = true
	end
	return decomposed
end

local function catSemester(cats, arg, departments)
	local semester = tonumber(mw.ustring.match(arg, '%d+'))
	if departments['СИ'] then
		cat(cats, data.semesters[semester] .. ' семестар')
	end
	if departments['РТИ'] then
		local matchRTI = mw.ustring.match(arg, '(%d+) %(РТИ%)')
		local semesterRTI = matchRTI and tonumber(matchRTI) or semester
		cat(cats, data.semesters[semesterRTI] .. ' семестар (РТИ)')
	end
end

function p.main(frame)
	local cats = {}
	local args = frame:getParent().args
	local departments = decomposeDepartments(args['одсек'] or 'СИ')
	catSemester(cats, args['семестар'], departments)
	return table.concat(cats)
end

return p