PHP爬蟲 -- 012 做業解析(把書城信息保存到數據庫)

建庫建表

create database bookstore character set utf8;
use bookstore;

CREATE TABLE `category` (
  `cid` int(11) NOT NULL AUTO_INCREMENT,
  `cname` varchar(255) NOT NULL,
  PRIMARY KEY (`cid`)
) ENGINE=MyISAM DEFAULT CHARSET=utf8;

CREATE TABLE `book` (
  `bid` int(11) NOT NULL AUTO_INCREMENT,
  `bname` varchar(255) NOT NULL,
  `bprice` varchar(10) NOT NULL,
  `bcid` int(11) NOT NULL,
  PRIMARY KEY (`bid`)
) ENGINE=MyISAM DEFAULT CHARSET=utf8;
複製代碼

把make_txt()換成save_data(),create_category()和create_book()

/* * @Description: 把整合好的數據, 寫入數據庫 * @param: 整合好的數組 * @return: 沒有返回值 */
function save_data($data) {
    foreach ($data as $key => $value) {
        $bcid = create_category($value['category_name']);
        foreach ($value['books'] as $k => $book) {
            $bname = $book['book_name'];
            $bprice = $book['book_price'];
            create_book($bname, $bprice, $bcid);
        }
    }
}
/* * @Description:往類別表裏插入數據 * @param: 類別名稱 * @return: 沒有返回值 */
function create_category($category_name) {
    global $con;
    $sql = "insert into category(`cname`) values(\"{$category_name}\")";
    print($sql);
    print("\n");
    mysqli_query($con, $sql);
    return mysqli_insert_id($con);
}

/* * @Description: 往圖書表中插入數據 * @param: 圖書名稱 * @param: 圖書價格 * @param: 圖書的分類id * @return: 沒有返回值 */
function create_book($bname, $bprice, $bcid) {
    global $con;
    $sql = "insert into book(`bname`,`bprice`,`bcid`) values(\"{$bname}\",'{$bprice}',{$bcid})";
    print($sql);
    print("\n");
    mysqli_query($con, $sql);
}
複製代碼

最終效果

完整代碼

<?php

/** * _ooOoo_ * o8888888o * 88" . "88 * (| -_- |) * O\ = /O * ____/`---'\____ * .' \\| |// `. * / \\||| : |||// \ * / _||||| -:- |||||- \ * | | \\\ - /// | | * | \_| ''\---/'' | | * \ .-\__ `-` ___/-. / * ___`. .' /--.--\ `. . __ * ."" '< `.___\_<|>_/___.' >'"". * | | : `- \`.;`\ _ /`;.`/ - ` : | | * \ \ `-. \_ __\ /__ _/ .-` / / * ======`-.____`-.___\_____/___.-`____.-'====== * `=---=' * ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ * 佛祖保佑 永無BUG */

require 'vendor/autoload.php';
use QL\QueryList;
// 生成一個querylist對象
$ql = new QueryList();
// 生成一個數據庫鏈接
$con = mysqli_connect('localhost', 'root', 'root', 'bookstore');
/* * @Description:獲取分類 * @params: 主頁url * @return: 二維數組, 包括分類名稱, 分類url */
function get_category($url) {
    global $ql;

    $data = $ql->get($url)->rules([
        "category_name" => ['#default > div > div > div > aside > div.side_categories > ul > li > ul > li > a', 'text'],
        "category_url" => ['#default > div > div > div > aside > div.side_categories > ul > li > ul > li > a', 'href'],
    ])->queryData();
    // 把分類的連接地址補全
    foreach ($data as $key => $value) {
        $value['category_url'] = $url . $value['category_url'];
        $data[$key] = $value;
    }
    return $data;
}

/* * @Description: 獲取分類下的圖書信息, 若是有下一頁, 遞歸獲取 * @param: 分類的url * @return: 二維數組, 包括圖書名稱, 圖書價格 */
function get_book($url) {
    global $ql;
    echo $url . "\n";
    $data = $ql->get($url)->rules([
        "book_name" => ['#default > div > div > div > div > section > div:nth-child(2) > ol > li > article > h3 > a', 'title'],
        "book_price" => ['#default > div > div > div > div > section > div:nth-child(2) > ol > li> article > div.product_price > p.price_color', 'text'],
    ])->queryData();
    // 獲取下一頁按鈕的href, 用來拼接下一頁的完整url
    $next = has_next($url);
    if ($next) {
        // 生成完整url
        $tmp_arr = explode('/', $url);
        $tmp_arr[count($tmp_arr) - 1] = $next;
        $next_url = implode('/', $tmp_arr);
        // 調用get_book(), 把返回的數據和當前數據合併
        $data = array_merge($data, get_book($next_url));
    }
    return $data;
}
/* * @Description: 判斷有沒有下一頁 * @param: 當前url * @return: 若是有返回按鈕的href, 若是沒有, 返回空字符串 */
function has_next($url) {
    global $ql;
    $res = $ql->get($url)->find('#default > div > div > div > div > section > div:nth-child(2) > div > ul > li.next > a')->href;
    return $res;
}
/* * @Description: 生成最終的數組 * @param: 從分類獲取的數組 * @return: 最終整合了圖書信息的數組 */
function make_array($data) {
    foreach ($data as $key => $value) {
        $value['books'] = get_book($value['category_url']);
        $data[$key] = $value;
    }
    return $data;
}

/* * @Description: 把整合好的數據, 寫入數據庫 * @param: 整合好的數組 * @return: 沒有返回值 */
function save_data($data) {
    foreach ($data as $key => $value) {
        $bcid = create_category($value['category_name']);
        foreach ($value['books'] as $k => $book) {
            $bname = $book['book_name'];
            $bprice = $book['book_price'];
            create_book($bname, $bprice, $bcid);
        }
    }
}
/* * @Description:往類別表裏插入數據 * @param: 類別名稱 * @return: 沒有返回值 */
function create_category($category_name) {
    global $con;
    $sql = "insert into category(`cname`) values(\"{$category_name}\")";
    print($sql);
    print("\n");
    mysqli_query($con, $sql);
    return mysqli_insert_id($con);
}

/* * @Description: 往圖書表中插入數據 * @param: 圖書名稱 * @param: 圖書價格 * @param: 圖書的分類id * @return: 沒有返回值 */
function create_book($bname, $bprice, $bcid) {
    global $con;
    $sql = "insert into book(`bname`,`bprice`,`bcid`) values(\"{$bname}\",'{$bprice}',{$bcid})";
    print($sql);
    print("\n");
    mysqli_query($con, $sql);
}

// 爬取並整合數據
$data = make_array(get_category('http://books.toscrape.com/'));
// 把數據寫入books.txt
save_data($data);
複製代碼

下一節

• PHP爬蟲 -- 013 PHP+Medoo