Module:ISO 3166
Documentation for this module may be created at Module:ISO 3166/doc
local p = {}
local data = mw.loadData("Module:ISO 3166/data/National")
local function cerror(eot,errortext,inputtext)
if eot=="empty" then
return ""
elseif eot=="input" then
return inputtext
else
return mw.html.create("span"):addClass("error"):wikitext(errortext)
end
end
local function strip(text)
text = string.upper(text) --Case insensitivity
text = string.gsub(text,"^THE ","") --Remove definite article
text = string.gsub(text,"[%s%-%,%.%(%)%/%\']","") --Remove spacing and punctuation
--text = string.gsub(text,"[\768-\879]","") --Unicode deaccenting doesn't work
return text
end
local function findname(code,cdata,qry)
local sqry = strip(qry)
if cdata["name"] and sqry==strip(cdata["name"])
or cdata["isoname"] and sqry==strip(cdata["isoname"])
or sqry==code or sqry==cdata["alpha3"] or sqry==cdata["numeric"]
then
return true
end
for _,tname in ipairs(cdata["altnames"] or {}) do
if sqry==strip(tname) then
return true
end
end
return false
end
function p.luaname(args)
local eot = args.error
if not args[1] then
return cerror(eot,"No parameter given",args[2] or "")
end
local code1 = string.upper(args[1] or "")
local code2 = string.upper(args[2] or "")
if string.find(code1,"%-") then
code1, code2 = string.match(code1,"^([^%-]*)%-(.*)$")
end
if --Check if valid code
--No non-alphanumeric characters allowed
string.find(code1..code2,"[^A-Z0-9]")
--3166-1 codes can be two or three letters or three digits;
--3166-2 codes must be two letters (first part) and 1-3 letters or digits (second part)
or not (string.find(code1,"^%u%u%u?$") or string.find(code1,"^%d%d%d$"))
or not string.find(code2,"^%w?%w?%w?$")
or not (string.find(code1,"^%u%u$") or code2=="")
then
if code2=="" then
return cerror(eot,"Invalid ISO 3166-1 code "..code1,code1)
else
return cerror(eot,"Invalid ISO 3166-2 code "..code1.."-"..code2,code1.."-"..code2)
end
end
if string.find(code1,"^%u%u$") then
if code2=="" then
--3166-1 alpha-2 code
if data[code1] then
return args.isoname and data[code1]["isoname"] or data[code1]["name"]
else
return cerror(eot,"Unknown ISO 3166-1 code "..code1,code1)
end
else
--3166-2 code
local sdata
if data[code1] then
sdata = mw.loadData("Module:ISO 3166/data/"..code1)
else
return cerror(eot,"Unknown ISO 3166-1 code "..code1,code1.."-"..code2)
end
if sdata[code2] then
return args.isoname and sdata[code2]["isoname"] or sdata[code2]["name"]
else
for _,_ in pairs(sdata) do
return cerror(eot,"Unknown ISO 3166-2 code "..code1.."-"..code2,code1.."-"..code2)
end
return cerror(eot,"No subdivision codes for "..data[code1]["name"],code1.."-"..code2)
end
end
else
--3166-1 alpha-3 or numeric code
local codetype = string.find(code1,"%d") and "numeric" or "alpha3"
for alpha2,cdata in pairs(data) do
if cdata[codetype]==code1 then
return args.isoname and cdata["isoname"] or cdata["name"]
end
end
return cerror(eot,"Unknown ISO 3166-1 code "..code1,code1)
end
end
function p.name(frame)
local args = require("Module:Arguments").getArgs(frame)
return p.luaname(args)
end
function p.luacode(args)
local eot = args.error
if not args[1] then
return cerror(eot,"No parameter given",args[2] or "")
end
if not args[2] then
--3166-1 code
for alpha2,cdata in pairs(data) do
if findname(alpha2,cdata,args[1]) then
if args["codetype"]=="numeric" or args["codetype"]=="alpha3" then
return cdata[args["codetype"]]
else
return alpha2
end
end
end
return cerror(eot,"Unknown country name "..args[1],args[1])
else
--3166-2 code
for alpha2,cdata in pairs(data) do
if findname(alpha2,cdata,args[1]) then
local sdata = mw.loadData("Module:ISO 3166/data/"..alpha2)
local empty = true
for scode,scdata in pairs(sdata) do
empty = false
if findname(scode,scdata,args[2]) then
return alpha2.."-"..scode
end
end
if empty then
return cerror(eot,"No subdivision codes for "..args[1],args[2])
else
return cerror(eot,"Unknown subdivision name "..args[2],args[2])
end
end
end
return cerror(eot,"Unknown country name "..args[1],args[2])
end
end
function p.code(frame)
local args = require("Module:Arguments").getArgs(frame)
return p.luacode(args)
end
function p.subpagetest(frame)
local code = frame.args[1]
local sdata
if not data[code] then
return "" --Unassigned or reserved 3166-1 code
else
local sname = "Module:ISO 3166/data/"..code
if not mw.title.new(sname).exists then
return "[[sname|-]]" --Data subpage doesn't exist
else
sdata = mw.loadData(sname)
end
end
for _,_ in pairs(sdata) do
return "n" --Non-empty 3166-2 entry
end
return "e" --Empty 3166-2 entry
end
return p