Add main.lua for political party module functionality

Author: Bapham12

Political party/main.lua

@@ -0,0 +1,116 @@
+local p = {}
+
+local default_color = '#F8F9FA'
+
+local categories = {
+	party_not_in_list = '[[Category:Pages using Political party with unknown party]]',
+	shortname_not_in_list = '[[Category:Pages using Political party with missing shortname]]',
+	color_not_in_list = '[[Category:Pages using Political party with missing color]]',
+}
+
+local function create_error(error_message)
+	return string.format('<strong class="error">%s</strong>', error_message)
+end
+
+local function getFirstLetter(party)
+	local index = mw.ustring.sub(party, 1, 1)
+	-- Set index for non-A-Z starts
+	if string.match(index, '%A') then
+		return '1'
+	end
+	return string.upper(index)
+end
+
+local function stripToNil(text)
+	-- If text is a string, return its trimmed content, or nil if empty.
+	-- Otherwise return text (which may, for example, be nil).
+	if type(text) == 'string' then
+		text = text:match('(%S.-)%s*$')
+		local delink = require('Module:Delink')._delink
+		text = delink({text, wikilinks = "target"})
+	end
+	return text
+end
+
+-- Example of having all the data - color and names - in one table. Requires one page to be edited instead of two when adding a new party.
+function p._fetch(args)
+	if not args[1] then
+		return create_error("parameter 1 should be a party name.")
+	end
+
+	if not args[2] then
+		return create_error("parameter 2 should be the output type.")
+	end
+
+ 	local party = stripToNil(args[1])
+	local out_type = stripToNil(args[2])
+	if out_type == 'colour' then
+		out_type = 'color'
+	end
+	local index = getFirstLetter(party)
+	
+	-- Load data from submodule
+	local data = mw.loadData('Module:Political party/' .. index)
+	local data_all = data.full
+
+	local party_alt = data.alternate[party]
+	local party_info
+	if party_alt then
+		if data_all[party_alt] then
+			party_info = data_all[party_alt]
+		else
+			index = getFirstLetter(party_alt)
+			data = mw.loadData('Module:Political party/' .. index)
+			party_info = data.full[party_alt]
+		end
+	else
+		party_info = data_all[party]
+	end
+
+	-- Check if database value exists
+	-- * Not even in database - return given error or input
+	-- * No color - return error
+	-- * No shortname/abbrev - return first non-blank of abbrev->shortname->input
+	if not party_info then
+		if out_type == 'color' then
+			return args.error or default_color
+		else
+			return args.error or party
+		end
+	end
+	local return_value = party_info[out_type]
+	if return_value == "" then
+		if out_type == 'color' then
+			return args.error or create_error("Value not in template. Please request that it be added.")
+		elseif out_type == 'abbrev' then
+			if party_info.shortname ~= "" then
+				return party_info.shortname
+			else
+				return party
+			end
+		elseif out_type == 'shortname' then
+			if party_info.abbrev ~= "" then
+				return party_info.abbrev 
+			else
+				return party
+			end
+		else
+			return party
+		end
+	end
+
+	if out_type == 'color' and string.find(return_value, '#') then
+		return_value = string.gsub(return_value, '#', '&#35;')
+	end
+	return return_value	
+end
+
+function p.fetch(frame)
+	-- Initialise and populate variables
+	local getArgs = require("Module:Arguments").getArgs
+	local args = getArgs(frame)
+	
+	return p._fetch(args)
+end
+
+return p