使用jsoup将表格内容展开 方便Regex进行内容定位

背景

最近在做的项目,需要从网络上抓取部分数据,在使用正则对数据进行匹配、提取时发现,表格类的数据因表格头和数据部分分离,直接定位这样的数据,有一定的困难(好吧,承认了,是自己没有好的解决思路_),所以简化下在解析前对内容进行预处理,方便后续的正则解析,比如:通过冗余,将表格头内容附加到内容前,使用特殊符号进行分隔,这样正则就能准确定位提取内容了。

一小步

思路有了,第一个问题就是,表格的行、列合并(rowspan、colspan)问题,那就需要把表格展开,代码是一个二维表格展开的方法,但是在生产环境下还得解决像并发这样的问题,这里就不展开了

上代码

使用库 JSoup,具体使用可以参考官方文档

表格转换

import java.io.FileInputStream;
import java.io.InputStream;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element;
import org.jsoup.parser.Tag;
import org.jsoup.select.Elements;

public class TableConvert {
/**
     * 将表格进行二维展开
     * @param table
     * @return
     */
    public Element[][] toTable(Element table) {
        if (!table.nodeName().equals("table")) {
            return null;
        }

        Elements tableRows = table.getElementsByTag("tr");
        int tableHeight = tableRows.size();

        //找 展开的最大列数,存在问题:如果某一列 全部使用 colspan 且其值都 大于2,有可能出错
        int tableWidth = 0;
        for (int tr_idx = 0; tr_idx < tableHeight; tr_idx++) {
            Elements tds = tableRows.get(tr_idx).select("td, th");
            int td_size = tds.size();
            if (td_size > tableWidth)
                tableWidth = td_size;
        }

        System.out.println("tableHeight:"+tableHeight+";tableWidth:"+tableWidth);
        
        if (tableHeight < 2 || tableWidth < 2)
            return null;

        //定义二维数组
        Element[][] result = new Element[tableHeight][tableWidth];

        //使用canreplace 来占位
        for(int i=0;i<tableHeight;i++) {
            for(int j=0;j<tableWidth;j++) {
                result[i][j]=new Element(Tag.valueOf("canreplace"),"");
            }
        }
        
        
        //出现不规范的 colspan 则会出现   实际列数 > tableWidth ,直接抛出异常
        try {
            for (int rowIndex = 0; rowIndex < tableHeight; rowIndex++) {
                Elements colCells = tableRows.get(rowIndex).select("td, th");

                System.out.println("row"+rowIndex+":\n"+colCells);              
                int pointIndex = 0;//列的索引
                for (int colIndex=0; colIndex < colCells.size();colIndex++) {                   
                    Element currentCell=colCells.get(colIndex);
                    //放到二维数组
                    if(result[rowIndex][colIndex].tagName().equalsIgnoreCase("canreplace"))
                    {
                        result[rowIndex][colIndex] = currentCell;
                        pointIndex=colIndex;
                    }else {
                        pointIndex=colIndex+1;
                        //查找可放置 一直找到一个可替换
                        while(!result[rowIndex][pointIndex].tagName().equalsIgnoreCase("canreplace") && pointIndex< tableWidth ) {
                            pointIndex++;                   
                            System.out.println("===rowIndex==="+pointIndex+"====tempColIndex==="+pointIndex+"==="+result[rowIndex][pointIndex].tagName());
                        }
                        if(pointIndex < tableWidth && result[rowIndex][pointIndex].tagName().equalsIgnoreCase("canreplace") ) {
                            result[rowIndex][pointIndex] = currentCell;
                        }else {
                            throw new Exception("table格式有错误!");
                        }
                    }
                    
                    
                    // 检查 colspan
                    int colspan = 1;
                    if (currentCell.hasAttr("colspan")) {
                        colspan = Integer.valueOf(currentCell.attr("colspan"));
                        currentCell.removeAttr("colspan");
                    }                   
                
                    //复制表格内容
                    if (colspan > 1) {      
                        for(int emptyColindex =1;emptyColindex < colspan  ;emptyColindex++)
                        {
                            pointIndex++;                           
                            while(!result[rowIndex][pointIndex].tagName().equalsIgnoreCase("canreplace") && pointIndex< tableWidth ) {
                                pointIndex++;                   
                                System.out.println("===rowIndex==="+pointIndex+"====tempColIndex==="+pointIndex+"==="+result[rowIndex][pointIndex].tagName());
                            }
                            if(pointIndex < tableWidth && result[rowIndex][pointIndex].tagName().equalsIgnoreCase("canreplace") ) {
                                result[rowIndex][pointIndex] = currentCell;
                            }else {
                                throw new Exception("table格式有错误!");
                            }
                        }
                    }

                    // 检查rowspan
                    int rowspan = 1;
                    if (currentCell.hasAttr("rowspan")) {
                        rowspan = Integer.valueOf(currentCell.attr("rowspan"));
                        currentCell.removeAttr("rowspan");
                    }

                    if (rowspan > 1) {
                        for (int i = 1; i < rowspan; i++) {
                            if (i >= tableHeight)  break; // ignore bad rowspans
                            System.out.println("===rowIndex==="+pointIndex+"====tempColIndex==="+pointIndex+"==="+result[rowIndex][pointIndex].tagName());
                            result[rowIndex+i][colIndex] = currentCell;//new Element(invalidTag, "");
                        }
                    }
                }
            }
        }       catch (Exception e) {
            e.printStackTrace();
            return null;
        }

        return result;
    }

辅助打印

    public void printTable(Element[][] table) {
        if (table == null) return;

        System.out.println("==================");
        for (int rowIndex = 0; rowIndex < table.length; rowIndex++) {
            System.out.print("|");
            for (int colIndex = 0; colIndex < table[rowIndex].length; colIndex++) {
                if (table[rowIndex][colIndex] == null) {
                    System.out.print("  ");
                } else {
                    System.out.print(table[rowIndex][colIndex].text());
                }
                System.out.print(" |");
            }
            System.out.println();
        }
        System.out.println("==================");
    }

运行测试

public static void main(String[] args) {
        String url = "d:\\Untitled-2.html";
        InputStream in;
        TableConvert tableConvert = new TableConvert();
        try {
            in = new FileInputStream(url);;
            Document doc = Jsoup.parse(in, null, "");

            for (Element aTable : doc.getElementsByTag("table")) {
                Elements subtables = aTable.getElementsByTag("table");
                subtables.remove(aTable);
                if(subtables.size() == 0) {
                    System.out.println("converting table...");
                    Element[][] result = tableConvert.toTable(aTable);
                    if (null != result)
                        tableConvert.printTable(result);
                    else
                        System.out.println("Could not convert table.");
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

测试文件内容

<table>
    <tr>
        <td>序号</td>
        <td>内容</td>
    </tr>
    <tr>
            <td colspan="2" rowspan="2">1</td>
    </tr>
    <tr>
            <td>英文</td>
    </tr>
    <tr>
            <td>2</td>
            <td>法文</td>
    </tr>
</table>
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,193评论 6 498
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,306评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 162,130评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,110评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,118评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,085评论 1 295
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,007评论 3 417
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,844评论 0 273
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,283评论 1 310
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,508评论 2 332
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,667评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,395评论 5 343
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,985评论 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,630评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,797评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,653评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,553评论 2 352

推荐阅读更多精彩内容

  • 内容样式 Content Types 通常,内容元素用于从一个网页中提取数据,并在输出数据中有相应的数据字段。内容...
    游侠儿evil阅读 942评论 0 0
  • 第一部分 HTML&CSS整理答案 1. 什么是HTML5? 答:HTML5是最新的HTML标准。 注意:讲述HT...
    kismetajun阅读 27,472评论 1 45
  • 转自链接 3.项目实践 3.1基于.xls模板生成Excel文件 3.2生成九九乘法表 3.3生成一张工资单 3....
    腿毛裤阅读 3,448评论 0 0
  • HTML 5 HTML5概述 因特网上的信息是以网页的形式展示给用户的,因此网页是网络信息传递的载体。网页文件是用...
    阿啊阿吖丁阅读 3,887评论 0 0
  • 上节课,使用table做出了一个最基本的表格,学会了设置表格的属性,调整宽度和边框。这节课,来学习如何更多tabl...
    学哥量化交易学习阅读 2,571评论 0 5