保留页面结构爬取wiki数据

最近有个爬取各国领导人信息的奇怪需求，要求百度和维基两种版本的数据，最要命的还要保持数据的结构不变。正好印象中隐约记得维基有专门的领导人列表页，不考虑爬取下来的格式不变的话应该很好爬的样子。

首先思路是通过列表页把每个领导人的信息页链接爬取下来，然后再逐个去解析信息页就OK了，思路很简单。

那么准备好爬取入口，在wiki上有一个各国领导人信息的列表页：https://zh.wikipedia.org/wiki/各国领导人列表

打开这个页面是这样的：

简直就是专为爬取设计的入口页，看了下页面代码结构，也是好爬的不行，查看具体的人物信息页链接是这样的格式：

https://zh.wikipedia.org/wiki/+名字

也就是这一页爬名字统统爬取下来，再凑好链接，加入任务列表就好了。

这是爬取列表页生成任务队列的代码：

 1 /**
 2          * 1.拿到列表页，得到所有领导人的姓名
 3          * 2.用姓名拼凑出链接，加入爬取队列
 4          */
 5         //得到信息列表并遍历列表
 6         List<Selectable> peopleTable = page.getHtml().xpath("//div[@id='mw-content-text']/div/table").nodes();
 7         for (Selectable table : peopleTable) {
 8             if (count != 0) {
 9                 break;
10             }
11             //拼凑链接加入爬取队列
12             List<Selectable> peopleLine = table.xpath("//tr").nodes();
13             for (Selectable line : peopleLine) {
14                 String name = line.xpath("//td[3]/a/text()").get();
15                 if (null == name || "".equals(name)) {
16                     continue;
17                 }
18                 String country = line.xpath("//td[1]/a/text()").get();
19                 //System.out.println("名字:"+name+"-------------"+"链接:"+url);
20                 countrys.put(name, country);
21                 String url = "https://zh.wikipedia.org/wiki/" + name;
22                 page.addTargetRequest(url);
23             }
24         }

接下来就去人物信息页查看页面结构，发现人物信息页的数据结构不太好爬，是这样子的：

 

 全是p标签和h标签的结构，看着脑袋疼，不过发现了一个有意思的东西，就是页面中有个这样的目录：

那么就根据目录来定位内容进行爬取，正好保证了数据的结构不变。

仔细查看目录发现目录跟后面的内容还是有联系点的：

 

 ------------------------------------------------------------------------------------------------------------------------------------------------------------------------------

 

很明显的联系，那就根据目录来爬取了。

在这之前需要先设计好数据，考虑到之后可能不止存领导人的信息而且每个人物的目录层次变动性很大，所以使用mongodb存数据

设计表如下，尽可能把页面上的数据都朗阔进去，虽然不一定都爬下来：

 

 接下来就开始写解析代码将数据抽取出来了，这一步需要用到jsoup来解析页面，里面有个很好用的方法就是定位到标签后可以拿到它的前一个或后一个兄弟标签，非常适用于这种文本段落没明显分层的页面。下面的是解析代码：

 1 /**
 2          * 1.解析详细人物信息页
 3          * 2.编写抽取规则，进行数据抽取
 4          */
 5         leader = new Leader();
 6         //从链接得到所解析人物的姓名，再得到国籍
 7         String leaderUrl = page.getUrl().get();
 8         String leaderName = leaderUrl.replace("https://zh.wikipedia.org/wiki/", "");
 9         String leaderCountry = countrys.get(leaderName);
10         System.out.println(leaderName + "--------" + leaderCountry);
11         leader.setId("" + count);
12         leader.setName(leaderName);
13         leader.setNationality(leaderCountry);
14         leader.setType(new String[] { "军事", "政治" });
15 //        testa = leaderName;        
16         
17         //解析页面，得到页面的导航目录，通过目录爬取具体内容
18         List<Selectable> lists = page.getHtml().xpath("//div[@id='toc']/ul/li").nodes();
19         List<Item> leaderDetail = new ArrayList<>();
20         for (Selectable list : lists) {
21             Item item = new Item();//一级目录
22             List<SubItems> subItems = new ArrayList<SubItems>();//二级目录
23             List<Selectable> secondLists = list.xpath("//ul//li").nodes();
24             //如果一级目录下记录不为0，说明当前目录下存在二级标题。否则只存在一级标题
25             if (secondLists.size() != 0) {
26                 String firstItem = list.xpath("//span[2]/text()").get();
27                 //抽取二级目录数据
28                 for (Selectable secondList : secondLists) {
29                     SubItems subItem = new SubItems();
30                     String[] secondItem = getText(secondList, page);
31                     subItem.setName(secondItem[0]);
32                     subItem.setValue(secondItem[1]);
33                     subItems.add(subItem);
34                 }
35                 item.setName(firstItem);
36                 item.setValue("");
37                 item.setSubItems(subItems);
38                 leaderDetail.add(item);
39             } else {
40                 //抽取一级目录数据
41                 String[] secondItem = getText(list, page);
42                 item.setName(secondItem[0]);
43                 item.setValue(secondItem[1]);
44                 leaderDetail.add(item);
45             }
46         }

其中的二级目录内容获取方法：

 1 public String[] getText(Selectable select, Page page) {
 2         String itemId = select.xpath("//a/@href").get();
 3         itemId = itemId.replace("#", "");
 4 
 5         Document doc = Jsoup.parse(page.getHtml().get());
 6         String citiao = doc.getElementById(itemId).text();
 7         Element elt = doc.getElementById(itemId).parent();
 8 
 9         StringBuffer sb = new StringBuffer();
10         while (true) {
11             elt = elt.nextElementSibling();
12             if (elt == null || "h2".equals(elt.tagName()) || "h3".equals(elt.tagName())) {
13                 break;
14             }
15             sb.append(elt.text()).append("\r\n");
16         }
17         return new String[] { citiao, sb.toString() };
18     }

这里面主要的就是要保持爬取下来的数据要按目录结构存储，一级目录包含二级目录，对应关系要把持住，在设计数据库的时候就需要考虑到。

然后是一些相关bean：

 1 public class Leader {
 2     private String id;
 3     private String name;
 4     private String gender;
 5     private String nationality;
 6     private Date birthday;
 7     private String birthPlace;
 8     private String[] type;
 9     private List<Item> info;
10     private List<Item> details;
11     public String getId() {
12         return id;
13     }
14     public void setId(String id) {
15         this.id = id;
16     }
17     public String getName() {
18         return name;
19     }
20     public void setName(String name) {
21         this.name = name;
22     }

 1 public class Item {
 2     private String name;
 3     private String value;
 4     private List<SubItems> subItem;
 5     public String getName() {
 6         return name;
 7     }
 8     public void setName(String name) {
 9         this.name = name;
10     }
11     public String getValue() {
12         return value;
13     }
14     public void setValue(String value) {
15         this.value = value;
16     }
17     public List<SubItems> getSubItems() {
18         return subItem;
19     }
20     public void setSubItems(List<SubItems> subItem) {
21         this.subItem = subItem;
22     }
23     
24     
25 }

 1 public class SubItems {
 2     private String name;
 3     private String value;
 4     public String getName() {
 5         return name;
 6     }
 7     public void setName(String name) {
 8         this.name = name;
 9     }
10     public String getValue() {
11         return value;
12     }
13     public void setValue(String value) {
14         this.value = value;
15     }
16     
17 }

最后测试的时候，发现一个很严重的问题，就是有些叫不出名字的国家，他们的领导人页面没有目录（就寥寥几句简介）。。。。。。导致信息爬取不出。

这是最后无目录的领导人个数：

这是数据库的数据：

只能说成功了一半。~~|