获取网页文章里的图片

业务需求是获取到网页文章里的图片
这里就用公众号来举个例子:
声明一个wkwebview

  lazy var wkwebview:WKWebView = {

        return WKWebView(frame: CGRect(x: 0, y: 0, width: UIScreen.main.bounds.width, height: UIScreen.main.bounds.height - 160))

    }()

加载

  let url = "https://mp.weixin.qq.com/s/eoQsC4bR2KWnvr3HIL6-DA"

        //"https://mp.weixin.qq.com/s/JumTmzPMKPzwylC3SDn7MQ"

        //"http://www.baidu.com"

        let request = URLRequest(url: URL(string: url)!)

        self.wkwebview.load(request)

获取图片,这里要声明一下:页面load完成跟资源加载完成是不一样的。所以在didFinish之后立即调用这个函数或许是拿不到图片的。最后是延时几秒执行。要解决这个问题,应该需要监听js的window.onload事件。

  func loadWebView(){
        //获取网页里面积较大的且靠前的图片,这里有个问题,在WKWebView, didFinish执行完实际上是没法立即拿到网页内的资源的。
        let jsStr = "function getImgArr(){const arr=[];const imgs=document.querySelectorAll(\"img\");Array.prototype.forEach.call(imgs,function(el,index){arr.push({src:el.src,index:index,size:el.width*el.height})});arr.sort(function(a,b){return b.size-a.size});return arr}function getBigImg(arr){if(arr.length){if(arr.length>1){return arr[0].index>arr[1].index?arr[1].src:arr[0].src}}};"

        self.wkwebview.evaluateJavaScript(jsStr, completionHandler: nil)
        self.wkwebview.evaluateJavaScript("getBigImg(getImgArr())") { (result, error) in

            print(result ?? "000")
            print(error ?? "sure")
            if let resultStr = result as? String, resultStr != ""{
                if let imageUrl = URL(string: resultStr) {
                    do{
                        if let imageData = try? Data(contentsOf: imageUrl) {
                            self.imgVie.image = UIImage(data: imageData)
                        }
                    }catch let error as NSError {
                        print(error)
                    }
                }
            }
        }
    }

你可能感兴趣的:(获取网页文章里的图片)