imported>Notacardoor at 19:14, 9 September 2021

2021-09-09T19:14:52Z

New page

local Error = require('Module:Error')

local p = {}

local aliasesQ = {
RottenTomatoes = "Q105584",
RottenTomatoesScore = "Q108403393",
RottenTomatoesAverage = "Q108403540",
Fandango = "Q5433722",
}

local aliasesP = {
RottenTomatoesId = "P1258",
reviewScore = "P444",
reviewScoreBy = "P447",
numberOfReviews = "P7887",
pointInTime = "P585",
determinationMethod = "P459",
author = "P50",
publisher = "P123",
statedIn = "P248",
language = "P407",
retrieved = "P813",
referenceURL = "P854",
archiveURL = "P1065",
title = "P1476",
formatterURL = "P1630",
archiveDate = "P2960",
}

-- Helper functions ------------------------------------------------------------
function mw.ustring.startswith(s, pattern, plain)
return mw.ustring.find(s, pattern, 1, plain) == 1
end

-- 0, nil, empty string, or empty table
local function falsy(x)
if x == nil or x == '' or x == '' then
return true
elseif type(x) == 'table' and next(x) == nil then
return true
end
return false
end

-- copied from Module:wd
local function parseDate(dateStr, precision)
precision = precision or "d"

local i, j, index, ptr
local parts = {nil, nil, nil}

if dateStr == nil then
return parts[1], parts[2], parts[3] -- year, month, day
end

-- 'T' for snak values, '/' for outputs with '/Julian' attached
i, j = dateStr:find("[T/]")

if i then
dateStr = dateStr:sub(1, i-1)
end

local from = 1

if dateStr:sub(1,1) == "-" then
-- this is a negative number, look further ahead
from = 2
end

index = 1
ptr = 1

i, j = dateStr:find("-", from)

if i then
-- year
parts[index] = tonumber(mw.ustring.gsub(dateStr:sub(ptr, i-1), "^\+(.+)$", "%1"), 10) -- remove '+' sign (explicitly give base 10 to prevent error)

if parts[index] == -0 then
parts[index] = tonumber("0") -- for some reason, 'parts[index] = 0' may actually store '-0', so parse from string instead
end

if precision == "y" then
-- we're done
return parts[1], parts[2], parts[3] -- year, month, day
end

index = index + 1
ptr = i + 1

i, j = dateStr:find("-", ptr)

if i then
-- month
parts[index] = tonumber(dateStr:sub(ptr, i-1), 10)

if precision == "m" then
-- we're done
return parts[1], parts[2], parts[3] -- year, month, day
end

index = index + 1
ptr = i + 1
end
end

if dateStr:sub(ptr) ~= "" then
-- day if we have month, month if we have year, or year
parts[index] = tonumber(dateStr:sub(ptr), 10)
end

return parts[1], parts[2], parts[3] -- year, month, day
end

-- nil dates precede all other (reasonable) dates since year becomes 1
local function datePrecedesDate(aY, aM, aD, bY, bM, bD)
aY, aM, aD = aY or 1, aM or 1, aD or 1
bY, bM, bD = bY or 1, bM or 1, bD or 1
if aY < bY then return true end
if aY > bY then return false end
if aM < bM then return true end
if aM > bM then return false end
if aD < bD then return true end
return false
end

--------------------------------------------------------------------------------
-- Returns either QID, true, or ErrorString, false
local function getEntityId(frame)
local entityId = frame.args.qid
local title = frame.args.title
if falsy(entityId) then
if falsy(title) then
local currentPageEntityId = mw.wikibase.getEntityIdForCurrentPage()
if currentPageEntityId then
return currentPageEntityId, true
end
return Error.error({'No Wikidata item connected to current page. Need qid or title parameter.'}), false
else
if not mw.title.makeTitle(0, title).exists then
return Error.error({'Article ' .. title .. ' does not exist.'}), false
end
entityId = mw.wikibase.getEntityIdForTitle(title)
if not entityId then
return Error.error({'Article "' .. title .. '" has no connected Wikidata item.'}), false
end
return entityId, true
end
end
--At this point we should have an entityId. Check if valid.
if not mw.wikibase.isValidEntityId(entityId) then
return Error.error({'Invalid Q-identifier.'}), false
end
if not mw.wikibase.entityExists(entityId) then
return Error.error({'Wikidata item ' .. entityId .. ' does not exist.'}), false
end
return entityId, true

end

local function date_from_statement(statement)
local pointintime = statement.qualifiers[aliasesP.pointInTime]
if not falsy(pointintime) then
pointintime = pointintime[1].datavalue.value.time
return parseDate(pointintime)
end
if statement.references[1] then
local accessdate = statement.references[1].snaks[aliasesP.retrieved]
if not falsy(accessdate) then
accessdate = accessdate[1].datavalue.value.time
return parseDate(accessdate)
end
end
return nil, nil, nil
end

local function reviewedby_RT(statement)
local x = statement.qualifiers[aliasesP.reviewScoreBy]
if falsy(x) then return false end
x = x[1].datavalue.value.id
if x ~= aliasesQ.RottenTomatoes then return false end
return true
end

-- statement should be a review score (P444) statement
local function score_type(statement)
local x = statement.qualifiers[aliasesP.determinationMethod]
if falsy(x) then return nil end
x = x[1].datavalue.value.id
if x == aliasesQ.RottenTomatoesScore then return 'percent' end
if x == aliasesQ.RottenTomatoesAverage then return 'average' end
local y = statement.mainsnak.datavalue.value
if string.match(y, '^0%%$') or string.match(y, '^[1-9][0-9]%%$') or string.match(y, '^100%%$') then
return 'percent'
end
if string.match(y, '^0 percent$') or string.match(y, '^[1-9][0-9] percent$') or string.match(y, '^100 percent$') then
return 'percent'
end
if string.match(y, '^%d/10$') or string.match(y, '^%d.%d%d?/10$') then
return 'average'
end
if string.match(y, '^%d out of 10$') or string.match(y, '^%d.%d%d? out of 10$') then
return 'average'
end
return nil
end

local function most_recent_score_statement(entityId, scoretype)
scoretype = scoretype or 'percent'
local score_statements = mw.wikibase.getAllStatements(entityId, aliasesP.reviewScore)
if falsy(score_statements) then
return nil
end
local newest = nil
local nY, nM, nD = nil, nil, nil
for i, v in ipairs(score_statements) do
local Y, M, D = date_from_statement(v)
if v.rank ~= 'deprecated' and reviewedby_RT(v) and score_type(v)==scoretype and not datePrecedesDate(Y, M, D, nY, nM, nD) then
nY, nM, nD = Y, M, D
newest = v
end
end
return newest
end

local function get_score(entityId, scoretype)
scoretype = scoretype or 'percent'
local x = most_recent_score_statement(entityId, scoretype)
if x then return x.mainsnak.datavalue.value end
return nil
end

local function get_count(entityId)
local x = most_recent_score_statement(entityId)
local y = x.qualifiers[aliasesP.numberOfReviews]
if falsy(y) then
return nil
end
return string.match(y[1].datavalue.value.amount, '%d+') -- dont get sign
end

local function get_rtid(entityId)
local x = mw.wikibase.getBestStatements(entityId, aliasesP.RottenTomatoesId)
if falsy(x) then return nil end
return x[1].mainsnak.datavalue.value
end

local function get_url(entityId)
local rtid = get_rtid(entityId)
if rtid == nil then return nil end
local x = mw.wikibase.getBestStatements(aliasesP.RottenTomatoesId, aliasesP.formatterURL)
subbed, k = string.gsub(x[1].mainsnak.datavalue.value, '$1', rtid)
return subbed
end

local function get_date(entityId, part)
local Y, M, D = date_from_statement(most_recent_score_statement(entityId))
local months = {'January', 'February', 'March', 'April', 'May', 'June',
'July', 'August', 'September', 'October', 'November', 'December'}
if part == 'year' then
if Y then
return Y
else
return ''
end
end
if part == 'month' then
if M then
return months[M]
else
return ''
end
end
if part == 'day' then
if D then
return D
else
return ''
end
end
local s = ''
if Y then
s = Y
if M then
s = M .. ' ' .. s
if D then
s = D .. ' ' .. s
end
end
end
return s
end

function p.test(frame)
mw.log('starting')
local entityId, is_good = getEntityId(frame)
if not is_good then
return entityId -- which is the error message
end
local command = frame.args[1]
if falsy(command) then
return Error.error({'Missing command.'})
end
local retval = ''
if command == 'score' then
retval = get_score(entityId, 'percent')
elseif command == 'average' then
retval = get_score(entityId, 'average')
elseif command == 'count' then
retval = get_count(entityId)
elseif command == 'rtid' then
retval = get_rtid(entityId)
elseif command == 'url' then
retval = get_url(entityId)
elseif command == 'date' or command == 'year' or command == 'month' or command == 'day' then
retval = get_date(entityId, command)
else
return Error.error({'Invalid command.'})
end
if falsy(retval) then
return Error.error({'Data for ' .. command .. ' unavailable.'})
end
return retval
end

function p.test2(frame)
end

return p

Module:Sandbox/Notsniwiast/learn - Revision history

imported>Notacardoor at 19:14, 9 September 2021