Module:Learnsets
Jump to navigation
Jump to search
Documentation for this module may be created at Module:Learnsets/doc
local p = {}
local function getArgs(frame)
local args = {}
local parent = frame:getParent()
if parent then
for k, v in pairs(parent.args) do
if v ~= '' then
args[k] = v
end
end
end
for k, v in pairs(frame.args) do
if v ~= '' then
args[k] = v
end
end
return args
end
local moveDataCache = nil
local function loadMoveData()
if moveDataCache then
return moveDataCache
end
local csvData = mw.ext.externalData.getExternalData({
url = 'https://wiki.pokemonrevolution.net/index.php?title=Special:GetData/RawMovesList',
format = 'CSV with header'
})
moveDataCache = {}
if csvData and csvData.Name then
for i, name in ipairs(csvData.Name) do
moveDataCache[name] = {
type = csvData.Type[i] or 'Normal',
category = csvData.Category[i] or 'Status',
power = csvData.Power[i],
accuracy = csvData.Accuracy[i] or '—',
pp = csvData.PP[i],
broken = csvData.Broken[i] or 'N',
typeless = csvData.Typeless[i] or 'N',
priority = csvData.Priority[i] or '0',
tm = csvData.TM[i],
move_tutor = csvData['Move Tutor'][i],
egg = csvData.Egg[i]
}
end
end
return moveDataCache
end
local function loadLearnsets()
local dataPage = mw.title.new('Data:Learnsets.json')
if not dataPage or not dataPage.exists then
return {}
end
local content = dataPage:getContent()
local success, data = pcall(mw.text.jsonDecode, content)
if success then
return data
else
return {}
end
end
local learnsetCache = nil
local function getCachedLearnsets()
if not learnsetCache then
learnsetCache = loadLearnsets()
end
return learnsetCache
end
local typeColors = {
Normal = { bg = "A8A878", border = "6D6D4E" },
Fire = { bg = "F08030", border = "9C531F" },
Water = { bg = "6890F0", border = "445E9C" },
Electric = { bg = "F8D030", border = "A1871F" },
Grass = { bg = "78C850", border = "4E8234" },
Ice = { bg = "98D8D8", border = "638D8D" },
Fighting = { bg = "C03028", border = "7D1F1A" },
Poison = { bg = "A040A0", border = "682A68" },
Ground = { bg = "E0C068", border = "927D44" },
Flying = { bg = "A890F0", border = "6D5E9C" },
Psychic = { bg = "F85888", border = "A13959" },
Bug = { bg = "A8B820", border = "6D7815" },
Rock = { bg = "B8A038", border = "786824" },
Ghost = { bg = "705898", border = "493963" },
Dragon = { bg = "7038F8", border = "4924A1" },
Dark = { bg = "705848", border = "49392F" },
Steel = { bg = "B8B8D0", border = "787887" },
Fairy = { bg = "EE99AC", border = "9B6470" },
Shadow = { bg = "5A5A5A", border = "2D2D2D" }
}
local categoryColors = {
Physical = { bg = "C92112", border = "7D1F1A" },
Special = { bg = "4F5870", border = "353B51" },
Status = { bg = "8C888C", border = "5A5A5A" }
}
local function isSTAB(moveType, primaryType, secondaryType)
return moveType == primaryType or moveType == secondaryType
end
local function getBorderColor(primaryType)
return typeColors[primaryType] and typeColors[primaryType].border or "000000"
end
function p.renderLevelUp(frame)
local args = getArgs(frame)
local pokemonName = args[1] or args.pokemon
local primaryType = args.type1 or ""
local secondaryType = args.type2 or ""
local learnsets = getCachedLearnsets()
local moveData = loadMoveData()
if not learnsets[pokemonName] then
return "Error: Pokemon '" .. pokemonName .. "' not found."
end
local pokemon = learnsets[pokemonName]
local moves = pokemon.level_up or {}
local html = mw.html.create('div')
local table = html:tag('table')
:css('width', '77%')
:css('text-align', 'center')
:css('margin', 'auto')
:css('border-radius', '15px')
:css('border', '3px solid #' .. getBorderColor(primaryType))
:css('background-color', '#F5F5F5')
:css('padding', '7px')
local header = table:tag('tr')
:css('background-color', '#E0E0E0')
header:tag('th'):wikitext('Level'):css('width', '14%'):css('border-radius', '15px 1px 1px 1px')
header:tag('th'):wikitext('Move'):css('width', '17%')
header:tag('th'):wikitext('Type'):css('width', '22%')
header:tag('th'):wikitext('Category'):css('width', '21%')
header:tag('th'):wikitext('Base Power'):css('width', '26%')
header:tag('th'):wikitext('Accuracy'):css('width', '13%')
header:tag('th'):wikitext('PP'):css('width', '5%'):css('border-radius', '1px 15px 1px 1px')
for _, move in ipairs(moves) do
local moveName = move.move
local level = move.level
local moveInfo = moveData[moveName] or {}
local row = table:tag('tr'):css('background-color', '#FFFFFF')
row:tag('td'):wikitext(tostring(level))
local moveLink = '[[' .. moveName .. ']]'
if isSTAB(moveInfo.type, primaryType, secondaryType) and moveInfo.category ~= "Status" then
moveLink = "'''" .. moveLink .. "'''"
end
if moveInfo.broken == "Y" then
moveLink = "''" .. moveLink .. "''"
end
row:tag('td'):wikitext(moveLink)
local typeColor = typeColors[moveInfo.type] or typeColors.Normal
row:tag('td')
:css('background-color', '#' .. typeColor.bg)
:css('border', '2px solid #' .. typeColor.border)
:wikitext('[[' .. (moveInfo.type or 'Normal') .. '_(type)|<span style="color:#FFFFFF;">' .. (moveInfo.type or 'Normal') .. '</span>]]')
local catColor = categoryColors[moveInfo.category] or categoryColors.Status
row:tag('td')
:css('background-color', '#' .. catColor.bg)
:css('border', '2px solid #' .. catColor.border)
:wikitext('[[' .. (moveInfo.category or 'Status') .. ' moves|<span style="color:#FFFFFF;">' .. (moveInfo.category or 'Status') .. '</span>]]')
row:tag('td'):wikitext(moveInfo.power or '—')
row:tag('td'):wikitext(moveInfo.accuracy or '—')
row:tag('td'):wikitext(moveInfo.pp or '—')
end
table:tag('tr'):tag('td'):attr('colspan', '7')
:css('text-align', 'left')
:css('background-color', '#E0E0E0')
:css('padding', '10px')
:css('border-radius', '1px 1px 25px 25px')
:wikitext([=[
* '''Emboldened''' moves denote this Pokémon avails from a STAB bonus for it.
* ''Italicized'' moves are [[Broken_Moves_%26_Abilities|broken]]; see their individual pages for more information.
* All moves subsumed on this list are relearnable via the [[Move Relearner]]s if lost.
]=])
return tostring(html) .. '<br clear="all">'
end
function p.renderTMsHMs(frame)
local args = getArgs(frame)
local pokemonName = args[1] or args.pokemon
local primaryType = args.type1 or ""
local secondaryType = args.type2 or ""
local learnsets = getCachedLearnsets()
local moveData = loadMoveData()
if not learnsets[pokemonName] then
return "Error: Pokemon '" .. pokemonName .. "' not found."
end
local pokemon = learnsets[pokemonName]
local tms = pokemon.tms or {}
local hms = pokemon.hms or {}
local html = mw.html.create('div')
local table = html:tag('table')
:css('width', '77%')
:css('text-align', 'center')
:css('margin', 'auto')
:css('border-radius', '25px')
:css('border', '3px solid #' .. getBorderColor(primaryType))
:css('background-color', '#F5F5F5')
:css('padding', '7px')
local header = table:tag('tr')
:css('background-color', '#E0E0E0')
header:tag('th'):attr('colspan', '2'):wikitext('TM'):css('width', '11%'):css('border-top-left-radius', '25px')
header:tag('th'):wikitext('Move'):css('width', '18%')
header:tag('th'):wikitext('Type'):css('width', '17%')
header:tag('th'):wikitext('Category'):css('width', '19%')
header:tag('th'):wikitext('Base Power'):css('width', '26%')
header:tag('th'):wikitext('Accuracy'):css('width', '20%')
header:tag('th'):wikitext('PP'):css('width', '7%'):css('border-top-right-radius', '25px')
for _, moveName in ipairs(tms) do
local moveInfo = moveData[moveName] or {}
local row = table:tag('tr'):css('background-color', '#FFFFFF')
local tmNum = moveInfo.tm or ''
row:tag('td'):attr('colspan', '2'):wikitext(tmNum ~= '' and tmNum or 'TM')
local moveLink = '[[' .. moveName .. ']]'
if isSTAB(moveInfo.type, primaryType, secondaryType) and moveInfo.category ~= "Status" then
moveLink = "'''" .. moveLink .. "'''"
end
if moveInfo.broken == "Y" then
moveLink = "''" .. moveLink .. "''"
end
row:tag('td'):wikitext(moveLink)
local typeColor = typeColors[moveInfo.type] or typeColors.Normal
row:tag('td')
:css('background-color', '#' .. typeColor.bg)
:css('border', '2px solid #' .. typeColor.border)
:wikitext('[[' .. (moveInfo.type or 'Normal') .. '_(type)|<span style="color:#FFFFFF;">' .. (moveInfo.type or 'Normal') .. '</span>]]')
local catColor = categoryColors[moveInfo.category] or categoryColors.Status
row:tag('td')
:css('background-color', '#' .. catColor.bg)
:css('border', '2px solid #' .. catColor.border)
:wikitext('[[' .. (moveInfo.category or 'Status') .. ' moves|<span style="color:#FFFFFF;">' .. (moveInfo.category or 'Status') .. '</span>]]')
row:tag('td'):wikitext(moveInfo.power or '—')
row:tag('td'):wikitext(moveInfo.accuracy or '—')
row:tag('td'):wikitext(moveInfo.pp or '—')
end
for _, moveName in ipairs(hms) do
local moveInfo = moveData[moveName] or {}
local row = table:tag('tr'):css('background-color', '#FFFFCC')
row:tag('td'):attr('colspan', '2'):wikitext('HM'):css('font-weight', 'bold')
local moveLink = '[[' .. moveName .. ']]'
if isSTAB(moveInfo.type, primaryType, secondaryType) and moveInfo.category ~= "Status" then
moveLink = "'''" .. moveLink .. "'''"
end
if moveInfo.broken == "Y" then
moveLink = "''" .. moveLink .. "''"
end
row:tag('td'):wikitext(moveLink)
local typeColor = typeColors[moveInfo.type] or typeColors.Normal
row:tag('td')
:css('background-color', '#' .. typeColor.bg)
:css('border', '2px solid #' .. typeColor.border)
:wikitext('[[' .. (moveInfo.type or 'Normal') .. '_(type)|<span style="color:#FFFFFF;">' .. (moveInfo.type or 'Normal') .. '</span>]]')
local catColor = categoryColors[moveInfo.category] or categoryColors.Status
row:tag('td')
:css('background-color', '#' .. catColor.bg)
:css('border', '2px solid #' .. catColor.border)
:wikitext('[[' .. (moveInfo.category or 'Status') .. ' moves|<span style="color:#FFFFFF;">' .. (moveInfo.category or 'Status') .. '</span>]]')
row:tag('td'):wikitext(moveInfo.power or '—')
row:tag('td'):wikitext(moveInfo.accuracy or '—')
row:tag('td'):wikitext(moveInfo.pp or '—')
end
table:tag('tr'):tag('td'):attr('colspan', '8')
:css('text-align', 'left')
:css('background-color', '#E0E0E0')
:css('padding', '10px')
:css('border-radius', '1px 1px 25px 25px')
:wikitext([=[
* '''Emboldened''' moves denote that this Pokémon avails from a STAB bonus for it.
* ''Italicized'' moves are [[Broken_Moves_%26_Abilities|broken]]; see their individual pages for more information.
* Refer to [[TMs and HMs]] for information on each move's obtainability.
]=])
return tostring(html) .. '<br clear="all">'
end
function p.renderMoveTutors(frame)
local args = getArgs(frame)
local pokemonName = args[1] or args.pokemon
local primaryType = args.type1 or ""
local secondaryType = args.type2 or ""
local learnsets = getCachedLearnsets()
local moveData = loadMoveData()
if not learnsets[pokemonName] then
return ""
end
local pokemon = learnsets[pokemonName]
local moves = pokemon.move_tutors or {}
if #moves == 0 then
return ""
end
local html = mw.html.create('div')
local table = html:tag('table')
:css('width', '77%')
:css('text-align', 'center')
:css('margin', 'auto')
:css('border-radius', '25px')
:css('border', '3px solid #' .. getBorderColor(primaryType))
:css('background-color', '#F5F5F5')
:css('padding', '7px')
local header = table:tag('tr'):css('background-color', '#E0E0E0')
header:tag('th'):wikitext('Move'):css('border-top-left-radius', '25px')
header:tag('th'):wikitext('Type')
header:tag('th'):wikitext('Category')
header:tag('th'):wikitext('Base Power')
header:tag('th'):wikitext('Accuracy')
header:tag('th'):wikitext('PP'):css('border-top-right-radius', '25px')
for _, moveName in ipairs(moves) do
local moveInfo = moveData[moveName] or {}
local row = table:tag('tr'):css('background-color', '#FFFFFF')
local moveLink = '[[' .. moveName .. ']]'
if isSTAB(moveInfo.type, primaryType, secondaryType) and moveInfo.category ~= "Status" then
moveLink = "'''" .. moveLink .. "'''"
end
if moveInfo.broken == "Y" then
moveLink = "''" .. moveLink .. "''"
end
row:tag('td'):wikitext(moveLink)
local typeColor = typeColors[moveInfo.type] or typeColors.Normal
row:tag('td')
:css('background-color', '#' .. typeColor.bg)
:css('border', '2px solid #' .. typeColor.border)
:wikitext('[[' .. (moveInfo.type or 'Normal') .. '_(type)|<span style="color:#FFFFFF;">' .. (moveInfo.type or 'Normal') .. '</span>]]')
local catColor = categoryColors[moveInfo.category] or categoryColors.Status
row:tag('td')
:css('background-color', '#' .. catColor.bg)
:css('border', '2px solid #' .. catColor.border)
:wikitext('[[' .. (moveInfo.category or 'Status') .. ' moves|<span style="color:#FFFFFF;">' .. (moveInfo.category or 'Status') .. '</span>]]')
row:tag('td'):wikitext(moveInfo.power or '—')
row:tag('td'):wikitext(moveInfo.accuracy or '—')
row:tag('td'):wikitext(moveInfo.pp or '—')
end
return tostring(html) .. '<br clear="all">'
end
function p.renderEggMoves(frame)
local args = getArgs(frame)
local pokemonName = args[1] or args.pokemon
local primaryType = args.type1 or ""
local secondaryType = args.type2 or ""
local learnsets = getCachedLearnsets()
local moveData = loadMoveData()
if not learnsets[pokemonName] then
return ""
end
local pokemon = learnsets[pokemonName]
local moves = pokemon.egg_moves or {}
if #moves == 0 then
return ""
end
local html = mw.html.create('div')
local table = html:tag('table')
:css('width', '77%')
:css('text-align', 'center')
:css('margin', 'auto')
:css('border-radius', '25px')
:css('border', '3px solid #' .. getBorderColor(primaryType))
:css('background-color', '#F5F5F5')
:css('padding', '7px')
local header = table:tag('tr'):css('background-color', '#E0E0E0')
header:tag('th'):wikitext('Move'):css('width', '15%'):css('border-top-left-radius', '25px')
header:tag('th'):wikitext('Type'):css('width', '13%')
header:tag('th'):wikitext('Category'):css('width', '14%')
header:tag('th'):wikitext('Base Power'):css('width', '15%')
header:tag('th'):wikitext('Accuracy'):css('width', '17%')
header:tag('th'):wikitext('PP'):css('width', '18%'):css('border-top-right-radius', '25px')
for _, moveName in ipairs(moves) do
local moveInfo = moveData[moveName] or {}
local row = table:tag('tr'):css('background-color', '#FFFFFF')
local moveLink = '[[' .. moveName .. ']]'
if isSTAB(moveInfo.type, primaryType, secondaryType) and moveInfo.category ~= "Status" then
moveLink = "'''" .. moveLink .. "'''"
end
if moveInfo.broken == "Y" then
moveLink = "''" .. moveLink .. "''"
end
row:tag('td'):wikitext(moveLink)
local typeColor = typeColors[moveInfo.type] or typeColors.Normal
row:tag('td')
:css('background-color', '#' .. typeColor.bg)
:css('border', '2px solid #' .. typeColor.border)
:wikitext('[[' .. (moveInfo.type or 'Normal') .. '_(type)|<span style="color:#FFFFFF;">' .. (moveInfo.type or 'Normal') .. '</span>]]')
local catColor = categoryColors[moveInfo.category] or categoryColors.Status
row:tag('td')
:css('background-color', '#' .. catColor.bg)
:css('border', '2px solid #' .. catColor.border)
:wikitext('[[' .. (moveInfo.category or 'Status') .. ' moves|<span style="color:#FFFFFF;">' .. (moveInfo.category or 'Status') .. '</span>]]')
row:tag('td'):wikitext(moveInfo.power or '—')
row:tag('td'):wikitext(moveInfo.accuracy or '—')
row:tag('td'):wikitext(moveInfo.pp or '—')
end
table:tag('tr'):tag('td'):attr('colspan', '6')
:css('text-align', 'left')
:css('background-color', '#E0E0E0')
:css('padding', '10px')
:css('border-radius', '1px 1px 25px 25px')
:wikitext([=[
* '''Emboldened''' moves denote that this Pokémon avails from a STAB bonus for it.
* ''Italicized'' moves are [[Broken_Moves_%26_Abilities|broken]]; see their individual pages for more information.
* For more information on egg-move-tutors, refer to: [[Egg Moves]].
]=])
return tostring(html) .. '<br clear="all">'
end
return p