Zoran: Pywikibot 6.4.0

2021-07-16T07:19:27Z

Pywikibot 6.4.0

New page

-- This module implements {{sports rivalry series table}}
local p = {}
local root = nil
local lang =mw.getContentLanguage()

local function formatnumR(num)
num = mw.ustring.gsub(num, '^[^%d]*([%d,%.]-)[^%d]*$', '%1')
return lang:parseFormattedNumber(num)
end

local function isnotempty(s)
return s and s:match( '^%s*(.-)%s*$' ) ~= ''
end
local function addheader(header1, header2, series_summary, compact, winner_only,
compact_score, nonumber, nolocation, trophy_series, notes, notes_label,
date_width, location_width)
-- create the header row
local row = root:tag('tr')
if(not nonumber) then
row:tag('th'):wikitext('No.')
end
row:tag('th'):css('width', date_width):wikitext('Date')
if(not nolocation) then
row:tag('th'):css('width', location_width):wikitext('Location')
end
if((not compact) and (not winner_only) and (not compact_score)) then
row:tag('th'):attr('colspan', 2):wikitext(header1 or 'Winning team')
row:tag('th'):attr('colspan', 2):wikitext(header2 or 'Losing team')
if(not series_summary) then
row:tag('th'):wikitext('Series')
end
else
if compact_score then
row:tag('th'):wikitext(header1 or 'Winning team')
row:tag('th'):wikitext(header2 or 'Losing team')
else
row:tag('th'):wikitext(header1 or 'Winner')
end
row:tag('th'):wikitext('Score')
if( (winner_only or compact_score) and (not series_summary)) then
row:tag('th'):wikitext('Series')
end
end
if(trophy_series) then
row:tag('th'):wikitext('Trophy series')
end
if(notes) then
row:tag('th'):wikitext(notes_label)
end
end

local function series_text(team1name, team1wins, team2name, team2wins, ties, leads)
local res = ''
local t1 = mw.ustring.gsub(team1name, '%s*/.*', '')
local t2 = mw.ustring.gsub(team2name, '%s*/.*', '')
if (team1wins > team2wins) then
res = t1 .. ' ' .. (leads and 'leads ' or '') .. team1wins .. '–' .. team2wins .. ( (ties > 0) and '–' .. ties or '')
elseif (team2wins > team1wins) then
res = t2 .. ' ' .. (leads and 'leads ' or '') .. team2wins .. '–' .. team1wins .. ( (ties > 0) and '–' .. ties or '')
else
res = 'Tied ' .. team1wins .. '–' .. team2wins .. ( (ties > 0) and '–' .. ties or '')
end
return res
end

local function small_rank(team)
team = mw.ustring.gsub(team or '', '(%(.-%))', '%1')
team = mw.ustring.gsub(team or '', '([Nn]o%.%s*[0-9][0-9]*)', '%1')
team = mw.ustring.gsub(team or '', '#([0-9][0-9]*%s*)([A-Z%)])', '<abbr title="Number">#</abbr>%1%2')
team = mw.ustring.gsub(team or '', '(<abbr[^<>]*>[^<>]*</abbr>%s*[0-9][0-9]*)', '%1')
team = mw.ustring.gsub(team or '', '(<small[^<>]*>%()<small[^<>]*>(.-)(%))', '%1%2%3')
return team
end

local function get_name(team)
team = mw.text.trim(team or '')
team = mw.ustring.gsub(team, '%[%[[^%[%]|]*|(.-)%]%]', '%1')
team = mw.ustring.gsub(team, '%([^%(%)]*%)', '')
team = mw.ustring.gsub(team, '<abbr[^<>]*>[^<>]*</abbr>', '')
team = mw.ustring.gsub(team, '^[Nn]o%.%s*[0-9][0-9]*', '')
team = mw.ustring.gsub(team, '^[^A-Za-z%.]*(.-)[^A-Za-z%.]*$', '%1')
return team
end

local function ismultiequal(s1, s2)
for k1, a1 in pairs( mw.text.split(s1, '[%s]/[%s]') ) do
for k2, a2 in pairs( mw.text.split(s2, '[%s]/[%s]') ) do
if a1 == a2 then
return 1
end
end
end
return nil
end

function p.table(frame)
local args = (frame.args[3] ~= nil) and frame.args or frame:getParent().args
local compact = (args['format'] or ''):lower() == 'compact'
local winner_only = (args['format'] or ''):lower() == 'winner only'
local compact_score = (args['format'] or ''):lower() == 'compact score'
local no_number = isnotempty(args['no_number'])
local no_location = isnotempty(args['no_location'])
local notes = isnotempty(args['notes']) or isnotempty(args['notes_label'])
local notes_label = isnotempty(args['notes_label']) and args['notes_label'] or 'Notes'
local team1style = args['team1style'] or ''
local team1name = mw.text.trim(args['team1'] or '')
local team1abbr = isnotempty(args['team1abbr']) and mw.text.trim(args['team1abbr'] or '') or team1name
local team2style = args['team2style'] or ''
local team2name = mw.text.trim(args['team2'] or '')
local team2abbr = isnotempty(args['team2abbr']) and mw.text.trim(args['team2abbr'] or '') or team2name
local team1wins = tonumber(args['team1win_start']) or 0
local team2wins = tonumber(args['team2win_start']) or 0
local ties = tonumber(args['tie_start']) or 0
local series_summary = isnotempty(args['series_summary'])
local legend = (args['legend'] and args['legend'] ~= 'no') or (args['legend'] == nil)
local cols = tonumber(args['cols'] or '') or 0
local tiestyle = args['tiestyle'] or 'background-color:#DDD; color:#000;'
local nowinstyle = args['nowinstyle'] or ''
local number_start = tonumber(args['number_start'] or '1') or 1
local tseries_start = tonumber(args['trophy_series_start'] or '0') or 0
local tseries_end = tonumber(args['trophy_series_end'] or '0') or 0
local tsteam1wins = 0
local tsteam2wins = 0
local tsties = 0

local res = ''
local topres = ''

if (cols < 1 ) then cols = 1 end

-- compute the maximum cell index
local cellcount = 0
for k, v in pairs( args ) do
if type( k ) == 'number' then
cellcount = math.max(cellcount, k)
end
end
-- datacols
local datacols = 6
if notes then datacols = datacols + 1 end
if no_location then datacols = datacols - 1 end

-- dataoffsets
local doffsets = no_location
and { ['date'] = 1, ['t1'] = 2, ['s1'] = 3, ['t2'] = 4, ['s2'] = 5, ['note'] = 6}
or { ['date'] = 1, ['loc'] = 2, ['t1'] = 3, ['s1'] = 4, ['t2'] = 5, ['s2'] = 6, ['note'] = 7}

-- compute the number of rows
local rows = math.ceil(cellcount / datacols)

-- compute the number of rows per column
local totalrows = rows
if (series_summary ) then totalrows = totalrows + 1 end
if (isnotempty(args['note'])) then totalrows = totalrows + 1 end
local percol = math.ceil( totalrows / cols )

if( (tseries_start > 0) or (tseries_end > 0) ) then
if( tseries_start < 1 ) then
tseries_start = 1
end
if( tseries_end < 1 ) then
tseries_end = rows
end
end

-- generate the legend
if( legend ) then
local legendtable = mw.html.create('table')
local col3 = args['legend_tie_text'] or 'Tie games'
local col4 = args['legend_forfeit_text'] or ''
local lcc = 2
if isnotempty(col3) then lcc = lcc + 1 else col3 = nil end
if isnotempty(col4) then lcc = lcc + 1 else col4 = nil end
local w12 = math.floor(100/lcc + 0.5)
local w34 = 0
if lcc > 2 then w34 = (100 - 2*w12)/(lcc - 2) end
legendtable
:addClass('wikitable')
:css('text-align', 'center')
:css('font-size', '90%')
:css('white-space', 'nowrap')
:cssText(args['style'])
local lrow = legendtable:tag('tr')
local t1 = mw.ustring.gsub(team1name, '%s*/.*', '')
local t2 = mw.ustring.gsub(team2name, '%s*/.*', '')
lrow:tag('td')
:cssText(team1style)
:css('font-weight', 'bold')
:css('width', w12 .. '%')
:wikitext(t1 .. ' victories')
lrow:tag('td')
:cssText(team2style)
:css('font-weight', 'bold')
:css('width', w12 .. '%')
:wikitext(t2 .. ' victories')
if( col3 ) then
lrow:tag('td')
:cssText(tiestyle)
:css('width', w34 .. '%')
:wikitext(col3)
end
if( col4 ) then
lrow:tag('td')
:css('width', w34 .. '%')
:wikitext(col4)
end
topres = topres .. tostring(legendtable)
end

-- build the table content
for j=1,rows do
if(math.fmod(j - 1, percol) == 0 ) then
-- create the root table
res = res .. (root and tostring(root) or '')
root = mw.html.create('table')
root:addClass('wikitable')
if isnotempty(args['sortable']) then root:addClass('sortable') end
root:css('text-align', 'center')
:css('font-size', '90%')
root:cssText(args['style'])
if(cols > 1) then
root:css('float', 'left')
root:css('margin-right', '1em')
end
addheader(args['header1'], args['header2'], series_summary,
compact, winner_only, compact_score, no_number, no_location,
(tseries_start > 0), notes, notes_label,
args['date_width'], args['location_width'])
end
-- start a new row
row = root:tag('tr')
row:css('vertical-align', 'top')
-- Number
if (not no_number) then row:tag('td'):wikitext(j - 1 + number_start) end
-- Date
row:tag('td'):wikitext(args[datacols*(j-1)+doffsets['date']] or '')
-- Location
if(not no_location) then
row:tag('td'):wikitext(args[datacols*(j-1)+doffsets['loc']] or '')
end
-- Team1 / Team2 / Score1 / Score2
local team1 = get_name(args[datacols*(j-1)+doffsets['t1']])
local score1 = mw.ustring.gsub(args[datacols*(j-1)+doffsets['s1']] or '', '^%s*(.-)%s*$', '%1')
local team2 = get_name(args[datacols*(j-1)+doffsets['t2']])
local score2 = mw.ustring.gsub(args[datacols*(j-1)+doffsets['s2']] or '', '^%s*(.-)%s*$', '%1')
local shade1 = nil
local shade2 = nil
local win = mw.ustring.gsub(args['win' .. (j - 1 + number_start)] or '', '^%s*(.-)%s*$', '%1') or ''
if( isnotempty(win) ) then
topres = topres .. '[[Category:Pages using sports rivalry series table with a win parameter]]'
end
if( isnotempty(win) or (score1 ~= '' and score2 ~= '') ) then
local score1num = tonumber(formatnumR(score1)) or 0
local score2num = tonumber(formatnumR(score2)) or 0
if( isnotempty(win) ) then
score1num = -1
score2num = -1
end
if ( (win == team1) or (score1num > score2num) ) then
if( ismultiequal(team1, team1name) ) then
shade1 = 'font-weight:bold;' .. team1style
if isnotempty(win) then shade2 = nowinstyle end
team1wins = team1wins + 1
if (j >= tseries_start and j <= tseries_end) then tsteam1wins = tsteam1wins + 1 end
elseif ( ismultiequal(team1, team2name) ) then
shade1 = 'font-weight:bold;' .. team2style
if isnotempty(win) then shade2 = nowinstyle end
team2wins = team2wins + 1
if (j >= tseries_start and j <= tseries_end) then tsteam2wins = tsteam2wins + 1 end
end
elseif ( (win == team2) or (score2num > score1num) ) then
if( ismultiequal(team2, team1name) ) then
shade2 = 'font-weight:bold;' .. team1style
if isnotempty(win) then shade1 = nowinstyle end
if (j >= tseries_start and j <= tseries_end) then tsteam1wins = tsteam1wins + 1 end
team1wins = team1wins + 1
elseif ( ismultiequal(team2, team2name) ) then
shade2 = 'font-weight:bold;' .. team2style
if isnotempty(win) then shade1 = nowinstyle end
team2wins = team2wins + 1
if (j >= tseries_start and j <= tseries_end) then tsteam2wins = tsteam2wins + 1 end
end
elseif ( (win == 'tie') or (not isnotempty(win) ) ) then
shade1 = tiestyle
shade2 = tiestyle
if not args['header1'] then
args[datacols*(j-1)+doffsets['t1']] = 'Tie'
args[datacols*(j-1)+doffsets['t2']] = 'Tie'
end
ties = ties + 1
if (j >= tseries_start and j <= tseries_end) then tsties = tsties + 1 end
end
end
shaderow = args['style' .. (j - 1 + number_start)]
if( (not compact) and (not winner_only) and (not compact_score)) then
-- Team 1
row:tag('td')
:cssText(shaderow or shade1)
:wikitext(small_rank(args[datacols*(j-1)+doffsets['t1']]))
-- Team 1 score
row:tag('td')
:cssText(shaderow or shade1)
:wikitext(score1)
-- Team 2
row:tag('td')
:cssText(shaderow or shade2)
:wikitext(small_rank(args[datacols*(j-1)+doffsets['t2']]))
-- Team 2 score
row:tag('td')
:cssText(shaderow or shade2)
:wikitext(score2)
-- Series
if(not series_summary) then
local seriescell = row:tag('td')
if( score1 ~= '' and score2 ~= '') then
seriescell:wikitext(series_text(team1abbr, team1wins, team2abbr, team2wins, ties, nil))
end
end
if(tseries_start > 0) then
local seriescell = row:tag('td')
if( score1 ~= '' and score2 ~= '' and j >= tseries_start and j <= tseries_end) then
seriescell:wikitext(series_text(team1abbr, tsteam1wins, team2abbr, tsteam2wins, tsties, nil))
end
end
else
if( isnotempty(win) or (score1 ~= '' and score2 ~= '') ) then
local score1num = tonumber(formatnumR(score1)) or 0
local score2num = tonumber(formatnumR(score2)) or 0
if(score1num > score2num) then
-- Winner
row:tag('td')
:cssText(shaderow or shade1)
:wikitext(small_rank(args[datacols*(j-1)+doffsets['t1']]))
if compact_score then
-- Loser
row:tag('td')
:wikitext(small_rank(args[datacols*(j-1)+doffsets['t2']]))
end
-- Score
row:tag('td')
:wikitext(score1 .. '–' .. score2)
elseif(score2num > score1num) then
if compact_score then
-- Loser
row:tag('td')
:wikitext(small_rank(args[datacols*(j-1)+doffsets['t1']]))
end
-- Winner
row:tag('td')
:cssText(shaderow or shade2)
:wikitext(small_rank(args[datacols*(j-1)+doffsets['t2']]))
-- Score
row:tag('td')
:wikitext(score2 .. '–' .. score1)
else
if compact_score then
row:tag('td'):cssText(shaderow or tiestyle):attr('colspan',2):wikitext('Tie')
else
-- Winner
row:tag('td'):cssText(shaderow or tiestyle):wikitext('Tie')
end
-- Score
row:tag('td')
:wikitext(score1 .. '–' .. score2)
end
if( (winner_only or compact_score) and (not series_summary)) then
local seriescell = row:tag('td')
if( score1 ~= '' and score2 ~= '') then
seriescell:wikitext(series_text(team1abbr, team1wins, team2abbr, team2wins, ties, nil))
end
end
if(tseries_start > 0) then
local seriescell = row:tag('td')
if( score1 ~= '' and score2 ~= '' and j >= tseries_start and j <= tseries_end) then
seriescell:wikitext(series_text(team1abbr, tsteam1wins, team2abbr, tsteam2wins, tsties, nil))
end
end
end
end
if(notes) then row:tag('td'):wikitext(args[datacols*(j-1)+doffsets['note']] or '') end
end

if( series_summary and root) then
local ftext = '\'\'\'Series:\'\'\' '
local ftext = ftext .. series_text(team1name, team1wins, team2name, team2wins, ties, 1)
if(args['footnote']) then
ftext = ftext .. args['footnote']
end
row = root:tag('tr'):addClass('sortbottom')
row:tag('td')
:attr('colspan', 9)
:css('background-color', '#f0f0f0')
:wikitext(ftext)
end

if(isnotempty(args['note']) and root) then
row = root:tag('tr'):addClass('sortbottom')
row:tag('td')
:attr('colspan', 9)
:wikitext(args['note'])
end

res = res .. (root and tostring(root) or '')

if (cols > 1 ) then
root = mw.html.create('table')
root:attr('role', 'presentation')
row = root:tag('tr')
row:tag('td'):wikitext(res)
res = tostring(root)
end

-- return the root table
return topres .. res
end

return p

Module:Sports rivalry series table - Revision history

Zoran: Pywikibot 6.4.0