Mô đun:Political party

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
local function getOriginPartyName (pName)
	local vData = mw.loadData('Mô đun:Political party/Data');
	local vPartyName = vData.oName;
	if (pName and vPartyNamepName ~= nil) then
		return vPartyNamepName
	end
	return pName
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 args1 then
		return create_error("parameter 1 should be a party name.")
	end
	if not args2 then
		return create_error("parameter 2 should be the output type.")
	end
	local dParty = stripToNil(args1)
	local party = getOriginPartyName(dParty)
	local out_type = stripToNil(args2)
	if out_type == 'colour' then
		out_type = 'color'
	end
	local index = getFirstLetter(party)
	
	-- Load data from submodule
	local data = mw.loadData('Mô đun:Political party/' .. index)
	local data_all = data.full
	local party_alt = data.alternateparty
	local party_info
	if party_alt then
		if data_allparty_alt then
			party_info = data_allparty_alt
		else
			index = getFirstLetter(party_alt)
			data = mw.loadData('Mô đun:Political party/' .. index)
			party_info = data.fullparty_alt
		end
	else
		party_info = data_allparty
	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 dParty ~= party and dParty or party
		end
	end
	local return_value = party_infoout_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 dParty ~= party and dParty or party
			end
		elseif out_type == 'shortname' then
			if party_info.abbrev ~= "" then
				return party_info.abbrev 
			else
				return dParty ~= party and dParty or party
			end
		else
			return dParty ~= party and dParty or 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