lua-htmlparser/doc/sample.lua

89 lines
2.2 KiB
Lua
Raw Permalink Normal View History

--require("luarocks.loader")
2013-03-28 11:24:10 +00:00
-- Omit next line in actual module clients; it's only to support development of the module itself
package.path = "../src/?.lua;" .. package.path
2013-03-19 19:07:52 +00:00
local htmlparser = require("htmlparser")
local io = require("io")
2013-03-28 11:24:10 +00:00
local file = io.input("./sample.html")
local text = io.read("*a") file:close()
2013-03-19 19:07:52 +00:00
local root = htmlparser.parse(text)
-- print the tree
local function p(n)
local space = string.rep(" ", n.level)
local s = space .. n.name
for k,v in pairs(n.attributes) do
s = s .. " " .. k .. "=[[" .. v .. "]]"
end
print(s)
for i,v in ipairs(n.nodes) do
p(v)
end
end
p(root)
print("\nchapters")
2013-03-19 21:32:24 +00:00
local sel, chapters = root("ol.chapters > li"), {}
for e in pairs(sel) do
table.insert(chapters, e:getcontent())
end
-- print
for i,v in ipairs(chapters) do
print(i, v)
end
print("\ncontacts")
local sel, contacts = root("ul.contacts span[class]"), {}
for e in pairs(sel) do
local id = e.parent.parent.id -- li > a > span
2013-03-20 08:38:43 +00:00
contacts[id] = contacts[id] or {}
contacts[id][e.classes[1]] = e:getcontent()
end
-- print
for k,v in pairs(contacts) do
print(k)
for fk,fv in pairs(v) do
print(fk, fv)
end
end
2013-03-20 08:38:43 +00:00
print("\nmicrodata")
local sel, scopes = root("[itemprop]"), {}
for prop in pairs(sel) do
if prop.attributes["itemscope"] then goto nextprop end
local descendantscopes, scope = {}, prop
while true do
repeat
scope = scope.parent
until scope.attributes["itemscope"]
if not scope.attributes["itemprop"] then break end
table.insert(descendantscopes, 1, scope)
end
scopes[scope] = scopes[scope] or {}
local entry = scopes[scope]
for _,v in ipairs(descendantscopes) do
entry[v] = entry[v] or {}
entry = entry[v]
end
local k, v = prop.attributes["itemprop"], prop:getcontent()
entry[k] = v
::nextprop::
end
-- print
local function printscope(node, table, level)
level = level or 1
local scopeprop = node.attributes["itemprop"] or ""
print(string.rep(" ", level - 1) .. node.attributes["itemtype"], scopeprop)
for prop,v in pairs(table) do
if type(prop) == "table" then
printscope(prop, v, level + 1)
else
print(string.rep(" ", level) .. prop .. "=[" .. v .. "]")
end
end
end
for node,table in pairs(scopes) do
printscope(node, table)
end