local p = {}

local default_color = '#F8F9FA'

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

local function create_error(error_message) return string.format('%s', 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, '#', '#') 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