Module:Infobox television season name: Difference between revisions
NeonWabbit (talk | contribs) Created page with "local match = require("Module:String")._match local p = {} --- Returns a formatted link to the list of episodes article. --- @param listOfEpisodesArticle string --- @return string local function getListOfEpisodesLink(listOfEpisodesArticle) local listOfEpisodesPage = mw.title.new(listOfEpisodesArticle, 0) if listOfEpisodesPage and listOfEpisodesPage.exists and listOfEpisodesPage.redirectTarget ~= mw.title.getCurrentTitle() then return string.format("[[%s|List of epi..." |
NeonWabbit (talk | contribs) No edit summary |
||
| Line 1: | Line 1: | ||
require('strict') | |||
local match = require("Module:String")._match | local match = require("Module:String")._match | ||
| Line 39: | Line 40: | ||
end | end | ||
--- Returns the type of word used for "season" in the | --- Returns the type of word used for "season" in the title. | ||
--- | --- | ||
--- The returned value can be one of three options: "season", "series", "story arc" or "specials". | --- The returned value can be one of three options: "season", "series", "story arc" or "specials". | ||
--- @param | --- @param title string The article's title. | ||
--- @return string | --- @return string | ||
local function getSeasonType( | local function getSeasonType(title) | ||
for _, seasonType in pairs({"season", "series", "story arc", "specials"}) do | for _, seasonType in pairs({"season", "series", "story arc", "specials"}) do | ||
if string.find( | if string.find(title, seasonType) then | ||
return seasonType | return seasonType | ||
end | end | ||
| Line 53: | Line 54: | ||
end | end | ||
--- Returns the | --- Returns the season number from the title. | ||
--- @param | --- @param title string The article's title. | ||
--- @return string | --- @return string | nil | ||
local function | local function getSeasonNumber(title) | ||
return match(title , "%d+", 1, -1, false, "") | |||
end | end | ||
| Line 66: | Line 66: | ||
local function getDisambiguation(title) | local function getDisambiguation(title) | ||
local disambiguation = match(title, "%s%((.-)%)", 1, -1, false, "") | local disambiguation = match(title, "%s%((.-)%)", 1, -1, false, "") | ||
if disambiguation == "" then | if disambiguation and disambiguation == "" then | ||
return nil | return nil | ||
end | end | ||
return disambiguation | return disambiguation | ||
end | |||
--- Returns the title without its disambiguation. | |||
--- @param title string The article's title. | |||
--- @return string | nil | |||
local function getTitleWithoutDisambiguation(title) | |||
local disambiguation = getDisambiguation(title) | |||
if disambiguation then | |||
return string.gsub(title, "%(" .. disambiguation .. "%)", "") | |||
end | |||
return title | |||
end | end | ||
| Line 118: | Line 129: | ||
--- @return string | --- @return string | ||
local function getShowName(title) | local function getShowName(title) | ||
local name, _ = mw.ustring.gsub(title, "% | local name, _ = mw.ustring.gsub(title, "season %d*$", "") | ||
name, _ = mw.ustring.gsub(name, "series %d*$", "") | |||
name, _ = mw.ustring.gsub(name, "specials", "") | |||
name, _ = mw.ustring.gsub(name, "story arc %d*$", "") | |||
name = string.match(name, "^%s*(.-)%s*$") -- Trim spaces. | |||
return name | return name | ||
end | end | ||
| Line 164: | Line 179: | ||
--- | --- | ||
--- The following are the supported season naming styles: | --- The following are the supported season naming styles: | ||
--- -- Style: <showName> | --- -- Style: <showName> <seasonType> <seasonNumber> | ||
--- Example: Lost | --- Example: Lost season 2. | ||
--- Example: Doctor Who | --- Example: Doctor Who series 2. | ||
--- -- Style: <showName> (<country> <seasonType> <seasonNumber> | --- -- Style: <showName> (<country> TV series) <seasonType> <seasonNumber> | ||
--- Example: The Office (American season 2) | --- Example: The Office (American TV series) season 2. | ||
--- Example: X Factor (British series 2) | --- Example: Teenage Mutant Ninja Turtles (1987 TV series) season 2 | ||
--- Example: X Factor (British TV series) series 2. | |||
--- Example: Love Island (British TV series) series 2 | |||
--- -- Style: <showName> (<year> TV series) <seasonType> <seasonNumber> | |||
--- Example: Love Island (2015 TV series) series 2 | |||
--- -- Style: <showName> (<country> <seasonType>) | --- -- Style: <showName> (<country> <seasonType>) | ||
--- Example: Big Brother 2 (American season). | --- Example: Big Brother 2 (American season). | ||
--- @param title string The article's title. | --- @param title string The article's title. | ||
--- @param seasonNumberDiff number The number difference between the current season and the other season. | --- @param seasonNumberDiff number The number difference between the current season and the other season. | ||
--- @return string, string | --- @return string, string | ||
local function getArticleTitleAndPipedLink(title, seasonNumberDiff) | local function getArticleTitleAndPipedLink(title, seasonNumberDiff) | ||
local seasonType = getSeasonType(title) | |||
local currentSeasonNumber = getSeasonNumber(title) | |||
if tonumber(currentSeasonNumber) == nil then | |||
return "", nil | |||
end | |||
local seasonNumber = currentSeasonNumber + seasonNumberDiff | |||
local modifiedTitle, numberOfReplacements = string.gsub(title, "%d+$", seasonNumber) | |||
local pipedLink = seasonType:gsub("^%l", string.upper) .. " " .. seasonNumber | |||
local disambiguation = getDisambiguation(title) | local disambiguation = getDisambiguation(title) | ||
-- Titles such as "Big Brother 2 (American season) and Teenage Mutant Ninja Turtles (1987 TV series) season 2". | |||
if disambiguation then | if disambiguation then | ||
local titleWithoutDisambiguation = string.gsub(title, disambiguation, "_DAB_") | |||
modifiedTitle, numberOfReplacements = string.gsub(titleWithoutDisambiguation, "%d+", seasonNumber) | |||
-- Articles, such as "Hawaii Five-0 (2010 TV series) season 2", that have a number | |||
-- as part of their title will need an additional fix in order for that number not to change. | |||
if numberOfReplacements > 1 then | |||
local titleFix = string.match(title, "%d+", 1) | |||
if | modifiedTitle = string.gsub(modifiedTitle, "%d+", titleFix, 1) | ||
end | end | ||
modifiedTitle = string.gsub(modifiedTitle, "_DAB_", disambiguation) | |||
return modifiedTitle, pipedLink | |||
-- Titles such as "Big Brother Brasil 2". | |||
elseif not string.find(title, seasonType) then | |||
return modifiedTitle, nil | |||
-- Invalid usages of TV series articles with the television season infobox. | |||
elseif disambiguation and string.find(disambiguation, "TV series") and not (string.find(disambiguation, ", season") or string.find(disambiguation, ", series")) then | |||
return "", nil | return "", nil | ||
-- Standard titles such as "Lost season 1". | |||
-- Standard titles such as "Lost | |||
else | else | ||
return modifiedTitle, pipedLink | |||
return | |||
end | end | ||
end | end | ||
| Line 232: | Line 241: | ||
local getArgs = require("Module:Arguments").getArgs | local getArgs = require("Module:Arguments").getArgs | ||
local args = getArgs(frame) | local args = getArgs(frame) | ||
if args.italic_title then | |||
return "no" | |||
end | |||
local title = args.title | local title = args.title | ||
if not title then | if not title then | ||
title = mw.title.getCurrentTitle().text | title = mw.title.getCurrentTitle().text | ||
| Line 303: | Line 315: | ||
function p.getSeasonWord(frame) | function p.getSeasonWord(frame) | ||
local title = getTitle(frame) | local title = getTitle(frame) | ||
local | title = getTitleWithoutDisambiguation(title) | ||
if | local seasonType = getSeasonType(title) | ||
return seasonType | |||
end | |||
--- Returns an {{Italic title}} instance if title qualifies or a blank string. | |||
--- @param frame table | |||
--- @return string | |||
function p.getItalicTitle(frame) | |||
local getArgs = require("Module:Arguments").getArgs | |||
local args = getArgs(frame) | |||
-- If italic_title is set then "no" is the only valid value. | |||
-- Don't set an italic title. | |||
if args.italic_title then | |||
return "" | return "" | ||
end | end | ||
local | local title = getTitle(frame) | ||
title = getShowName(getTitleWithoutDisambiguation(title)) | |||
if | |||
-- If the infobox is used on List of articles don't set an italic title. | |||
-- TODO: this can be fixed in the future but current usages use a manual display title. | |||
if string.find(title, "List of") then | |||
return "" | |||
end | end | ||
return | return frame:expandTemplate{title = "Italic title", args = {string = title}} | ||
end | |||
--- Returns the text used for the |above= field of the infobox. | |||
--- | |||
--- @param frame table | |||
--- @return string | |||
function p.getAboveTitle(frame) | |||
local getArgs = require("Module:Arguments").getArgs | |||
local args = getArgs(frame) | |||
local title = getTitle(frame) | |||
title = getShowName(getTitleWithoutDisambiguation(title)) | |||
return title | |||
end | end | ||
--- Returns the | --- Returns the text used for the |subheader= field of the infobox. | ||
--- | --- | ||
--- The text is returned in the format of "Season #" or "Series #", | --- The text is returned in the format of "Season #" or "Series #", | ||
| Line 323: | Line 364: | ||
--- @param frame table The frame invoking the module. | --- @param frame table The frame invoking the module. | ||
--- @return string | nil | --- @return string | nil | ||
function p. | function p.getSubHeader(frame) | ||
local getArgs = require("Module:Arguments").getArgs | local getArgs = require("Module:Arguments").getArgs | ||
local args = getArgs(frame) | local args = getArgs(frame) | ||
| Line 337: | Line 378: | ||
end | end | ||
local title = getTitle(frame) | if not seasonNumber then | ||
local title = getTitle(frame) | |||
local titleWithoutDisambiguation = getTitleWithoutDisambiguation(title) | |||
seasonNumber = getSeasonNumber(titleWithoutDisambiguation) | |||
seasonType = getSeasonType(titleWithoutDisambiguation) | |||
-- For pages like "Doctor Who specials (2008–2010)". | |||
if seasonType == "specials" then | if seasonType == "specials" then | ||
return | local disambiguation = getDisambiguation(title) or "" | ||
return disambiguation .. " " .. seasonType | |||
end | end | ||
seasonType = seasonType:sub(1, 1):upper() .. seasonType:sub(2) | seasonType = seasonType:sub(1, 1):upper() .. seasonType:sub(2) | ||
end | end | ||
| Line 380: | Line 422: | ||
local title = getTitle(frame) | local title = getTitle(frame) | ||
local showName = getShowName(title) | local showName = getShowName(getTitleWithoutDisambiguation(title)) | ||
if showName then | if showName then | ||
local disambiguation = getDisambiguation(title) | local disambiguation = getDisambiguation(title) | ||
if disambiguation then | |||
local listOfEpisodesArticle = string.format("List of %s%s episodes", showName, | disambiguation = " (" .. disambiguation .. ")" | ||
end | |||
local listOfEpisodesArticle = string.format("List of %s%s episodes", showName, disambiguation or "") | |||
return getListOfEpisodesLink(listOfEpisodesArticle) | return getListOfEpisodesLink(listOfEpisodesArticle) | ||
end | end | ||