Module:ISO 3166: Difference between revisions

3,178 bytes added ,  2 years ago
m
1 revision imported
m (1 revision imported)
m (1 revision imported)
 
(One intermediate revision by one other user not shown)
Line 1:
-- to enable us to replicate the current functioning of Country extract, we need to deal with:
-- 2 {{<name>}} DONE!
-- 3 [[<name>]] DONE!
-- 4 [[<name>|<junk>]] DONE!
-- 5 [[image:flag of <country>.[svg|gif|png|jpg]|\d+px]] DONE!
 
local p = {}
local getArgs = require("Module:Arguments").getArgs
 
local data = mw.loadData("Module:ISO 3166/data/National")
 
--[[----------F I N D N A M E----------]]-- -- Finds the name in the database
local function cerror(eot,errortext,inputtext)
if eot=="empty" then
return nil
elseif eot=="input" then
return inputtext
else
return mw.html.create("span"):addClass("error"):wikitext("Error in [[Module:ISO 3166]]: "..errortext)
end
end
 
local function findname(code,cdata,qry)
local sqry = p.strip(qry)
if cdata["name"] and sqry==p.strip(cdata["name"])
or cdata["isoname"] and sqry==p.strip(cdata["isoname"])
or not cdata["nocode"] and sqry==code
or sqry==cdata["alpha3"] or sqry==cdata["numeric"]
then
return true
end
for _,tname in pairs(cdata["isonames"] or {}) do
if sqry==p.strip(tname) then
return true
end
end
for _,tname in pairs(cdata["altnames"] or {}) do
if sqry==p.strip(tname) then
return true
end
end
return false
end
 
--[[----------I S O N A M E----------]]-- -- Find the ISO name of a country/region
 
local function isoname(data,code,lang)
if data[code]["isonames"] then
local name = data[code]["isonamesisodisplaynames"][lang orand data[code]["defaultlangisodisplaynames"] or data["defaultlang"] or "en"lang]
or data[code]["isonames"][data[code]["defaultlang"] or data["defaultlang"] or "en"lang]
or data[code]["isodisplaynames"] and data[code]["isodisplaynames"][data[code]["defaultlang"] or data["defaultlang"]]
or data[code]["isonames"]["en"]
or data[code]["isonames"][data[code]["defaultlang"] or data["defaultlang"]]
if name then return name end
or data[code]["isodisplaynames"] and data[code]["isodisplaynames"]["en"]
for _,iname in pairs(data[code]["isonames"]) do return iname end
returnor data[code]["isodisplaynameisonames"] or data[code]["isonameen"]
if name then return name end
else
for _,iname in return pairs(data[code]["isodisplaynameisonames"]) ordo data[code]["isoname"]return iname end
return data[code]["isodisplayname"] or data[code]["isoname"]
end
else
return data[code]["isodisplayname"] or data[code]["isoname"]
end
end
 
--[[----------S T R I P----------]]-- -- Removes junk from the input
 
function p.strip(text)
local accents = {["À"]="A",["Á"]="A",["Â"]="A",["Ã"]="A", -- accent list
text = mw.ustring.upper(text) --Case insensitivity
["Ä"]="A",["Å"]="A",["Ç"]="C",["È"]="E",["É"]="E",
text = string.gsub(text,"^THE ","") --Remove definite article
["Ê"]="E",["Ë"]="E",["Ì"]="I",["Í"]="I",["Î"]="I",
text = string.gsub(text,"[%s%-%,%.%(%)%/%\']","") --Remove spacing and punctuation
local accents = {["À"]="A",["Á"]="A",["Â"]="A",["Ã"]="A", ["ÄÏ"]="AI",["ÅÑ"]="AN",["ÇÒ"]="CO",["ÈÓ"]="EO",["ÉÔ"]="EO",
["ÊÕ"]="EO",["ËÖ"]="EO",["ÌØ"]="IO",["ÍÙ"]="IU",["ÎÚ"]="I",["Ï"]="I",["Ñ"]="N",["Ò"]="O",["Ó"]="OU",
["Ô"]="O",["Õ"]="O",["Ö"]="O",["Ø"]="O",["Ù"]="U",["Ú"]="U", ["Û"]="U",["Ü"]="U",["Ý"]="Y"}
}
text = mw.ustring.gsub(text,"[À-Ý]",accents) --Deaccent
local remove = {"NATION OF","COUNTRY OF","TERRITORY OF", -- text to be removed list
return text
"FLAG OF","FLAG","KINGDOM OF","STATE OF"," STATE ",
"PROVINCE OF","PROVINCE","TERRITORY"
}
local patterns = {[".+:"]="",["|.+"]="",["%(.-%)"]="", -- patterns to follow (order may matter)
["%..*"]="",["^THE "]="",["%_"]=" ",["%-"]=" ",
["%d%d?%d?PX"]="",
}
text = mw.ustring.upper(text) -- Case insensitivity
text = mw.ustring.gsub(text,"[À-Ý]",accents) -- Deaccent
for pattern,value in pairs(patterns) do -- Follow patterns
text = mw.ustring.gsub(text,pattern,value)
end
for _,words in pairs(remove) do -- Remove unneeded words
text = mw.ustring.gsub(text,words,"")
end
text = mw.ustring.gsub(text,"%W","") -- Remove non alpha-numeric
return text
end
 
--[[----------P . C A L L S T R I P ---------]]-- -- Calls P.strip but using Module:Arguments
function p.luaname(args)
 
function p.callstrip(frame)
local eot = args.error
local args = getArgs(frame)
return p.strip(args[1]) or ""
 
if not args[1] then
return cerror(eot,"No parameter given",args[2] or "")
end
 
--[[----------P . L U A C O D E---------]]-- -- Makes the ISO code of a country
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
 
function p.luacode(args)
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(code1args[1] or '',"^%u%u$-") then
args[1], args[2] = string.match(args[1] or '',"^([^%-]*)%-(.*)$")
if code2=="" then
end
--3166-1 alpha-2 code
if data[code1] then
if args[1] then args[1] = p.strip(args[1]) end
return (args.isoname or args.lang)
if args[2] then args[2] = p.strip(args[2]) end
and isoname(data,code1,args.lang)
or (data[code1]["displayname"] or data[code1]["name"])
if args["codetype"]=="3" then
else
args["codetype"]="alpha3"
return cerror(eot,"Unknown ISO 3166-1 code "..code1,code1)
end
else
local eot = args.error or ""
--3166-2 code
local sdata
local catnocountry = (args.nocat and args.nocat == 'true') and ''
if data[code1] then
or '[[Category:Wikipedia page with obscure country]]'
sdata = mw.loadData("Module:ISO 3166/data/"..code1)
local catnosubdivision = (args.nocat and args.nocat == 'true') and ''
else
or '[[Category:Wikipedia page with obscure subdivision]]'
return cerror(eot,"Unknown ISO 3166-1 code "..code1,code1.."-"..code2)
end
if not if sdataargs[code21] then
if mw.title.getCurrentTitle().namespace ~= 0 then catnocountry = '' end
return (args.isoname or args.lang)
return catnocountry, '<span style="font-size:100%" class="error">"No parameter for the country given"</span>'
and isoname(sdata,code2,args.lang)
end
or (sdata[code2]["displayname"] or sdata[code2]["name"])
else
if not args[2] then --3166-1 code
for _,_ in pairs(sdata) do
for alpha2,cdata in pairs(data) do
return cerror(eot,"Unknown ISO 3166-2 code "..code1.."-"..code2,code1.."-"..code2)
if findname(alpha2,cdata,args[1]) then
end
if args["codetype"]=="numeric" or args["codetype"]=="alpha3" then
return cerror(eot,"No subdivision codes for "..data[code1]["name"],code1.."-"..code2)
return cdata[args["codetype"]]
end
else
end
return alpha2
else
end
--3166-1 alpha-3 or numeric code
end
local codetype = string.find(code1,"%d") and "numeric" or "alpha3"
end
for alpha2,cdata in pairs(data) do
if mw.title.getCurrentTitle().namespace ~= 0 then catnocountry = '' end
if cdata[codetype]==code1 then
return catnocountry
return (args.isoname or args.lang)
else --3166-2 code
and isoname(data,alpha2,args.lang)
for alpha2,cdata in pairs(data) do
or (cdata["displayname"] or cdata["name"])
if findname(alpha2,cdata,args[1]) then
end
if mw.ustring.match(alpha2,"GB") then -- For England, Wales etc.
end
alpha2 = "GB"
return cerror(eot,"Unknown ISO 3166-1 code "..code1,code1)
end
local sdata = mw.loadData("Module:ISO 3166/data/"..alpha2)
local empty = true
for scode,scdata in pairs(sdata) do
if type(scdata)=="table" then
empty = false
if findname(scode,scdata,args[2]) then
return alpha2.."-"..scode
end
end
end
if mw.title.getCurrentTitle().namespace ~= 0 then catnosubdivision = '' end
return catnosubdivision
end
end
if mw.title.getCurrentTitle().namespace ~= 0 then catnocountry = '' end
return catnocountry
end
 
end
 
--[[----------P . C O D E---------]]-- -- Calls P.Luacode but using Module:Arguments
function p.name(frame)
 
function p.code(frame)
local args = require("Module:Arguments").getArgs(frame)
 
return p.luaname(args) or ""
return p.luacode(getArgs(frame)) or ""
 
end
 
--[[----------P . N U M E R I C---------]]-- -- Calls P.Luacode but using Module:Arguments and setting it to output a numeric value
function p.luacode(args)
 
function p.numeric(frame)
local eot = args.error
 
local args = getArgs(frame)
if not args[1] then
return cerror(eot,"No parameter given",args[2] or "")
args["codetype"]="numeric"
return p.luacode(args) or ""
end
--[[----------P . L U A N A M E---------]]-- -- Makes the ISO/common name of a country
function p.luaname(args)
 
local code1 = p.luacode(args)
if not args[2] then
local code2 = ''
--3166-1 code
for alpha2,cdata in pairs(data) do
if findnamestring.find(alpha2,cdatacode1,args[1]"%-") then
code1, code2 = string.match(code1,"^([^%-]*)%-(.*)$")
if args["codetype"]=="numeric" or args["codetype"]=="alpha3" then
end
return cdata[args["codetype"]]
else
if string.find(code1,"^%u%u$") then
return alpha2
if code2=="" then --3166-1 alpha-2 code
end
if data[code1] then
end
return (args.isoname or args.lang) and isoname(data,code1,args.lang)
end
or (data[code1]["displayname"] or data[code1]["name"])
return cerror(eot,"Unknown country name "..args[1],args[1])
else
return '[[Category:Wikipedia page with obscure country]]'
--3166-2 code
end
for alpha2,cdata in pairs(data) do
else --3166-2 code
if findname(alpha2,cdata,args[1]) then
local sdata
local sdata = mw.loadData("Module:ISO 3166/data/"..alpha2)
if data[code1] then
local empty = true
sdata = mw.loadData("Module:ISO 3166/data/"..code1)
for scode,scdata in pairs(sdata) do
else
if type(scdata)=="table" then
return '[[Category:Wikipedia page with obscure country]]'
empty = false
end
if findname(scode,scdata,args[2]) then
if sdata[code2] then
return alpha2.."-"..scode
return (args.isoname or args.lang) and isoname(sdata,code2,args.lang)
end
or (sdata[code2]["displayname"] or sdata[code2]["name"])
end
else
end
return '[[Category:Wikipedia page with obscure country]]'
if empty then
end
return cerror(eot,"No subdivision codes for "..args[1],args[2])
end
else
end
return cerror(eot,"Unknown subdivision name "..args[2],args[2])
end
end
end
return cerror(eot,"Unknown country name "..args[1],args[2])
end
 
--[[----------P . N A M E---------]]-- -- Calls P.Luaname but using Module:Arguments
end
 
function p.codename(frame)
 
local return args = require("Module:Arguments")p.luaname(getArgs(frame)) or ""
return p.luacode(args) or ""
 
end
--[[----------P . G E O C O O R D I N S E R T---------]]-- -- Wrapper for Module:Coordinates.coordinsert
function p.geocoordinsert(frame)
-- {{#invoke:ISO 3166|geocoordinsert|{{coord|...}}
-- |country=..|subdivision1=...|subdivision2=...
-- |type=...|scale=...|dim=...|source=...|globe=...
-- }}
local args = frame.args
local subdivisionqueried = false
local catnocountry = (args.nocat and args.nocat == 'true') and ''
or '[[Category:Wikipedia page with obscure country]]'
local catnosubdivision = (args.nocat and args.nocat == 'true') and ''
or '[[Category:Wikipedia page with obscure subdivision]]' or ''
local tracking = ''
local targs = {}
targs[1] = args[1] or ''
for i, v in pairs(args) do
if i == 'country' and not mw.ustring.find(targs[1], 'region:') then
local country = v
local k, region = 1, ''
-- look for a valid subdivision
while region == '' and k < 3 do
local subdivision = args['subdivision' .. k] or ''
if subdivision ~= '' then
region = p.luacode({country, subdivision, nocat = 'true'})
subdivisionqueried = true
end
k = k + 1
end
-- subdivision lookup failed or never attempted, try country only
if region == '' then
region = p.luacode({country, nocat = 'true'})
if mw.title.getCurrentTitle().namespace ~= 0 then catnocountry, catnosubdivision = '', '' end
if region == '' then
tracking = tracking .. catnocountry
elseif subdivisionqueried == true then
tracking = tracking .. catnosubdivision
end
end
-- something worked, add it to the targs
if region ~= '' then
targs[#targs + 1] = 'region:' .. region
end
elseif i == 'type' or i == 'scale' or i == 'dim'
or i == 'source' or i == 'globe' then
targs[#targs + 1] = i .. ':' .. v
end
end
-- call Module:Coordinates.coordinsert if there is something to insert
if #targs > 1 then
local coordinsert = require('Module:Coordinates').coordinsert
return coordinsert({args = targs}) .. tracking
end
-- otherwise, just return the coordinates
return targs[1] .. tracking
end