Module:WikiProject assessment progression and Module:WikiProject assessment progression/sandbox: Difference between pages

Content deleted Content added

@@ Line 1: / Line 1: @@
 require('strict')
-local classes = {
+local assessedClasses = {
 	'List','Stub','Start','C','B',
 	'GA','A','FA','FL','FM',
@@ Line 10: / Line 10: @@
 local p = {}
 local getArgs = require('Module:Arguments').getArgs
+local errorMessage = require('Module:Error')
 local function categoryCount(category, project)
@@ Line 36: / Line 37: @@
 local function calculateProjectTotal(project)
-	return countCategoryArticles(classes, project) + categoryCount('Unassessed', project)
+	return countCategoryArticles(assessedClasses, project) + categoryCount('Unassessed', project)
 end
-local function arg_or_default(args, from_arg, default)
+local function countHigherCategories(targetClass, project)
+	local higherCategories = {}
-	if args[from_arg] and args[from_arg] ~= '' then
+	local classKey = getKeyForValue(targetClass, assessedClasses)
-		return args[from_arg]
+	for k, v in ipairs(assessedClasses) do
+		if k > classKey then
+			higherCategories[k - classKey] = v
+		end
+	end
+	return countCategoryArticles(higherCategories, project)
+end
+local function getKeyForValue(value, enclosingTable)
+	for k, v in ipairs(enclosingTable) do
+		if v == value then
+			return k
+		end
+	end
+	return nil
+end
+local function setClasses(args)
+	local classes = {}
+	if args['class'] then
+		classes = {args['class']}
 	else
+		classes = assessedClasses
-		return default
 	end
+	return classes
 end
@@ Line 53: / Line 76: @@
 function p._main(args)
-	local project = arg_or_default(args, "project", nil)
+	local project = args['project']
+	local classes = setClasses(args)
 	local classCount = countCategoryArticles(classes, project)
 	local total = calculateProjectTotal(project)
+	if total == 0 then
-	return percentComplete(classCount, total)
+		return errorMessage('Error: project not found for project name "'..project..'"')
+	else
+		return percentComplete(classCount, total)
+	end
 end