Module:ISO 3166/sandbox: Difference between revisions

Content deleted Content added
No edit summary
sync
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")
 
local catnocountry = '[[Category:Wikipedia page with obscure country]]'
--[[----------F I N D N A M E----------]]-- -- Finds the name in the database
local catnosubdivision = '[[Category:Wikipedia page with obscure subdivision]]'
 
--[[----------S T R I P----------]]--
local function findname(code,cdata,qry)
-- Remove unnecessary text
local functionsqry = p.strip(textqry)
if cdata["name"] and sqry==p.strip(cdata["name"])
local accents = {
or cdata["isoname"] and sqry==p.strip(cdata["isoname"])
["À"]="A", ["Á"]="A", ["Â"]="A", ["Ã"]="A", ["Ä"]="A", ["Å"]="A",
or not cdata["nocode"] and sqry==code
["Ç"]="C",
["È"]or sqry="E", =cdata["Éalpha3"]="E", ["Ê"]or sqry="E", =cdata["Ënumeric"]="E",
then
["Ì"]="I", ["Í"]="I", ["Î"]="I", ["Ï"]="I",
["Ñ"]="N",
["Ò"]="O", ["Ó"]="O", ["Ô"]="O", ["Õ"]="O", ["Ö"]="O", ["Ø"]="O",
["Ù"]="U", ["Ú"]="U", ["Û"]="U", ["Ü"]="U",
["Ý"]="Y"
}
local patterns = {
["%_"]=" ",
["%-"]=" "
}
local remove = {
"%d+PX",
"<.*>",
".+:",
"|.+",
"%(.-%)",
"%..*",
"^THE",
",.*$",
"%W",
"^[0]+",
"KINGDOMOF",
"OF",
"^STATE",
"STATE$",
"REPUBLIC$",
"NATION",
"COUNTRY",
"TERRITORY",
"FLAG",
"PROVINCE"
}
text = mw.ustring.upper(text)
if mw.ustring.match(text,"[À-Ý]") then
text = mw.ustring.gsub(text,"[À-Ý]",accents)
end
text = mw.text.unstrip(text)
for find, replace in pairs(patterns) do
text = mw.ustring.gsub(text,find,replace)
end
for _, find in ipairs(remove) do
text = mw.ustring.gsub(text,find,"")
end
return text
end
--[[----------F I N D C O D E----------]]--
-- Check to find the code or name
local function findcode(code,cdata,qry)
local sqry = strip(qry)
if sqry == code
or sqry == cdata["alpha3"]
or cdata["numeric"] and sqry == mw.ustring.gsub(cdata["numeric"], '^[0]+','')
or sqry == strip(cdata["isoname"] or '')
or sqry == strip(cdata["name"] or '')
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
Line 79 ⟶ 32:
return false
end
 
--[[----------N A M E O R C O D E---------]]--
--[[----------I S O N A M E----------]]-- -- Find the ISO name of a country/region
--Determine what name or alpha2 code to return
 
local function nameorcode(args,data,ldata,alpha2,dash2)
local function isoname(data,code,lang)
if args.type == "name" then
if args.isoname or args.langdata[code]["isonames"] then
local langname = args.lang or ldatadata[code]["defaultlangisodisplaynames"] orand data["defaultlang"code] or ["enisodisplaynames"][lang]
return ldataor data["isonames"code] and ldata["isonames"][lang] or ldata["isoname"] or ''
or data[code]["isodisplaynames"] and data[code]["isodisplaynames"][data[code]["defaultlang"] or data["defaultlang"]]
end
or data[code]["isonames"][data[code]["defaultlang"] or data["defaultlang"]]
return ldata["name"]
or data[code]["isodisplaynames"] and data[code]["isodisplaynames"]["en"]
elseif dash2 then
or data[code]["isonames"]["en"]
return alpha2.."-"..dash2
if name then return name end
for _,iname in pairs(data[code]["isonames"]) do return iname end
return data[code]["isodisplayname"] or data[code]["isoname"]
else
return data[code]["isodisplayname"] or data[code]["isoname"]
return alpha2
end
end
 
--[[----------G E T C O D E---------]]--
--[[----------S T R I P----------]]-- -- Removes junk from the input
-- Return the code or name
 
local function getcode(args)
function p.strip(text)
catnocountry = args.nocat == 'true' and '' or catnocountry
local accents = {["À"]="A",["Á"]="A",["Â"]="A",["Ã"]="A", -- accent list
catnosubdivision = args.nocat == 'true' and '' or catnosubdivision
["Ä"]="A",["Å"]="A",["Ç"]="C",["È"]="E",["É"]="E",
if not args[1] then return catnocountry end
["Ê"]="E",["Ë"]="E",["Ì"]="I",["Í"]="I",["Î"]="I",
if string.find(args[1], "^%s*(%w%w?%w?)%-(%w%w?%w?)") then
["Ï"]="I",["Ñ"]="N",["Ò"]="O",["Ó"]="O",["Ô"]="O",
args[1], args[2] = string.match(args[1],"^%s*(%w%w?%w?)%-(%w%w?%w?)")
["Õ"]="O",["Ö"]="O",["Ø"]="O",["Ù"]="U",["Ú"]="U",
["Û"]="U",["Ü"]="U",["Ý"]="Y"
}
local remove = {"NATION OF","COUNTRY OF","TERRITORY OF", -- text to be removed list
"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
if args[1] then args[1] = strip(args[1]) end
for _,words in pairs(remove) do -- Remove unneeded words
if args[2] then args[2] = strip(args[2]) end
text = mw.ustring.gsub(text,words,"")
if args["codetype"] == "3" then args["codetype"] = "alpha3" end
end
for alpha2,cdata in pairs(data) do
if findcode(alpha2,cdata,args[1]) then
text = mw.ustring.gsub(text,"%W","") -- Remove non alpha-numeric
if args.type == "numeric" or args.type == "alpha3" then
return cdata[args.type]
return text
else
local scode = nil
end
if args[2] then
 
local scdata = mw.loadData("Module:ISO 3166/data/"..alpha2)
--[[----------P . C A L L S T R I P ---------]]-- -- Calls P.strip but using Module:Arguments
for scode,sdata in pairs(scdata) do
 
if findcode(scode,sdata,args[2]) then
function p.callstrip(frame)
return nameorcode(args,scdata,sdata,alpha2,scode)
local args = getArgs(frame)
return p.strip(args[1]) or ""
 
end
 
--[[----------P . L U A C O D E---------]]-- -- Makes the ISO code of a country
 
function p.luacode(args)
 
if string.find(args[1] or '',"%-") then
args[1], args[2] = string.match(args[1] or '',"^([^%-]*)%-(.*)$")
end
if args[1] then args[1] = p.strip(args[1]) end
if args[2] then args[2] = p.strip(args[2]) end
if args["codetype"]=="3" then
args["codetype"]="alpha3"
end
local eot = args.error or ""
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]]'
if not args[1] then
if mw.title.getCurrentTitle().namespace ~= 0 then catnocountry = '' end
return catnocountry, '<span style="font-size:100%" class="error">"No parameter for the country given"</span>'
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
if mw.title.getCurrentTitle().namespace ~= 0 then catnocountry = '' end
return catnocountry
else --3166-2 code
for alpha2,cdata in pairs(data) do
if findname(alpha2,cdata,args[1]) then
if mw.ustring.match(alpha2,"GB") then -- For England, Wales etc.
alpha2 = "GB"
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
return catnosubdivision
else
return nameorcode(args,data,cdata,alpha2)
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
 
return catnocountry
end
 
--[[----------M A K E F U N C---------]]--
--[[----------P . C O D E---------]]-- -- Calls P.Luacode but using Module:Arguments
-- Make a function for the p table
 
local function makefunc(params)
function p.code(frame)
params = params or {}
 
return function(frame)
return p.luacode(getArgs(frame)) or ""
local args = {}
 
if type(frame.args) == 'table' then
end
args = frame.args
 
elseif type(frame) == 'table' then
--[[----------P . N U M E R I C---------]]-- -- Calls P.Luacode but using Module:Arguments and setting it to output a numeric value
args = frame
 
end
function p.numeric(frame)
if params.type then
 
args.type = params.type
local args = getArgs(frame)
end
if params.func == 'strip' then
args["codetype"]="numeric"
return strip(args[1] or args.text) or ''
else
return getcodep.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)
local code2 = ''
if string.find(code1,"%-") then
code1, code2 = string.match(code1,"^([^%-]*)%-(.*)$")
end
if string.find(code1,"^%u%u$") then
if code2=="" then --3166-1 alpha-2 code
if data[code1] then
return (args.isoname or args.lang) and isoname(data,code1,args.lang)
or (data[code1]["displayname"] or data[code1]["name"])
else
return '[[Category:Wikipedia page with obscure country]]'
end
else --3166-2 code
local sdata
if data[code1] then
sdata = mw.loadData("Module:ISO 3166/data/"..code1)
else
return '[[Category:Wikipedia page with obscure country]]'
end
if sdata[code2] then
return (args.isoname or args.lang) and isoname(sdata,code2,args.lang)
or (sdata[code2]["displayname"] or sdata[code2]["name"])
else
return '[[Category:Wikipedia page with obscure country]]'
end
end
end
end
 
--[[----------P . N A M E---------]]-- -- Calls P.Luaname but using Module:Arguments
 
function p.name(frame)
 
return p.luaname(getArgs(frame)) or ""
 
end
--[[----------E X P O R T E D F U N C T I O N S---------]]--
--[[----------P . G E O C O O R D I N S E R T---------]]-- -- Wrapper for Module:Coordinates.coordinsert
-- Functions in the p table
p.strip = makefunc({func='strip'})
p.code = makefunc()
p['alpha2'] = makefunc()
p['alpha3'] = makefunc({type='alpha3'})
p.numeric = makefunc({type='numeric'})
p.name = makefunc({type='name'})
--[[----------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|...}}
Line 166 ⟶ 235:
local args = frame.args
local subdivisionqueried = false
local catnocountry = (args.nocat and args.nocat == 'true') and '' or catnocountry
or '[[Category:Wikipedia page with obscure country]]'
catnosubdivision = args.nocat == 'true' and '' or catnosubdivision
local catnosubdivision = (args.nocat and args.nocat == 'true') and ''
or '[[Category:Wikipedia page with obscure subdivision]]' or ''
local tracking = ''
local targs = {}
Line 187 ⟶ 258:
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
Line 210 ⟶ 282:
return targs[1] .. tracking
end
 
return p