Module:Table row counter/sandbox

-- This module counts table rows in wikitext.

local p = {} local getArgs

function p.main(frame) if not getArgs then getArgs = require('Module:Arguments').getArgs end return p._main(getArgs(frame, {wrappers = 'Template:Table row counter'})) end

function p._main(args) -- Get the title object. local titleObj do local success success, titleObj = pcall(mw.title.new, args.page) if not success or not titleObj then titleObj = mw.title.getCurrentTitle end end

-- Get the page content. local content = titleObj:getContent if not content then return nil end

-- Find the wikitables on that page. local wikitables = {} do local iWikitable = 0 local s1 = content:match('^({|.-\n|})') if s1 then iWikitable = iWikitable + 1 wikitables[iWikitable] = s1		end for s in content:gmatch('\n({|.-\n|})') do			iWikitable = iWikitable + 1 wikitables[iWikitable] = s		end end

-- Find the wikitable to work on. local wikitable if args.id then for i, s in ipairs(wikitables) do			if s:match('^{|[^\n]*id *= *" *(%w+) *"') == args.id then wikitable = s				break end end else wikitable = wikitables[tonumber(args.tableno) or 1] end if not wikitable then return nil end

-- Count the number of rows. local count do local temp temp, count = wikitable:gsub('\n|%-', '\n|-') end

-- Control for missing row markers at the start. if not wikitable:find('^{|[^\n]*%s*\n|%-') then count = count + 1 end

-- Control for extra row markers at the end. if wikitable:find('\n|%-[^\n]-%s*\n|}$') then count = count - 1 end

-- Subtract the number of rows to ignore, or the number of header -- rows if it's empty, and make sure the result isn't below zero. local headers headers = 0 if not args.include_headers then headers = wikitable:gsub('\n|%-\n!', '\n|-\n!') end if not wikitable:find('^{|[^\n]*%s*\n|%-\n!') then headers = headers + 1 end count = count - (tonumber(args.ignore) or headers) if count < 0 then count = 0 end return count end

return p