Module:Political party

local p = {}

local categories = { party_not_in_list = '', shortname_not_in_list = '', color_not_in_list = '', }

local function getFirstLetter(party) 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*$') 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(frame) -- Initialise and populate variables local args = frame.args local party = stripToNil(args[1]) or error('parameter 1 should be a party name') local out_type = stripToNil(args[2]) or error('parameter 2 should be the output type') 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 return args.error or party end local return_value = party_info[out_type] if return_value == "" then if out_type == 'color' then return args.error or 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

return p