Hell 发表于 2023-6-28 16:21

用python实现调用百度图片搜索的API

前言:
这段代码是一个简单的图片爬虫程序
它可以通过输入关键词,在百度图片中搜索相关图片,并返回一张随机的图片。
代码使用Flask框架搭建了一个简单的Web应用,将用户输入的关键词作为参数传递给爬虫程序,然后从百度图片中获取相关图片的URL
并随机选择一张返回给用户。用户可以通过访问Web应用的首页,在输入框中输入关键词进行搜索。如果找到相关图片,则会跳转到图片的URL
如果未找到图片,则会显示无法正常查找的提示信息。

应用场景:
将代码发布到服务器上,就可以用markdown格式
!服务器域名/?word=关键字
来显示搜索的图片。

特别是部署了gpt-web的大佬,可以用角色仓库的方式来实现调用百度图片的api来搜图
以下是案例:




源码:
import argparse
import os
import re
import sys
import urllib
import json
import socket
import urllib.request
import urllib.parse
import urllib.error
import random

# 设置超时
import time

from flask import Flask, redirect, request, make_response
import random

timeout = 5
socket.setdefaulttimeout(timeout)

import ssl
ssl._create_default_https_context = ssl._create_unverified_context


class Crawler:
    # 睡眠时长
    __time_sleep = 0.1
    __amount = 0
    __start_amount = 0
    __counter = 0
    headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64; rv:23.0) Gecko/20100101 Firefox/23.0', 'Cookie': ''}
    __per_page = 30

    # 获取图片url内容等
    # t 下载图片时间间隔
    def __init__(self, t=0.1):
      self.time_sleep = t

    # 获取后缀名
    @staticmethod
    def get_suffix(name):
      m = re.search(r'\.[^\.]*$', name)
      if m.group(0) and len(m.group(0)) <= 5:
            return m.group(0)
      else:
            return '.jpeg'

    @staticmethod
    def handle_baidu_cookie(original_cookie, cookies):
      """
      :param string original_cookie:
      :param list cookies:
      :return string:
      """
      if not cookies:
            return original_cookie
      result = original_cookie
      for cookie in cookies:
            result += cookie.split(';') + ';'
      result.rstrip(';')
      return result

    # 开始获取
    def get_images(self, word):
      search = urllib.parse.quote(word)
      pn = self.__start_amount
      image_urls = []
      while pn < self.__amount:
            url = 'https://image.baidu.com/search/acjson?tn=resultjson_com&ipn=rj&ct=201326592&is=&fp=result&queryWord=%s&cl=2&lm=-1&ie=utf-8&oe=utf-8&adpicid=&st=-1&z=&ic=&hd=&latest=&copyright=&word=%s&s=&se=&tab=&width=&height=&face=0&istype=2&qc=&nc=1&fr=&expermode=&force=&pn=%s&rn=%d&gsm=1e&1594447993172=' % (
                search, search, str(pn), self.__per_page)
            try:
                time.sleep(self.time_sleep)
                req = urllib.request.Request(url=url, headers=self.headers)
                page = urllib.request.urlopen(req)
                self.headers['Cookie'] = self.handle_baidu_cookie(self.headers['Cookie'],
                                                                page.info().get_all('Set-Cookie'))
                rsp = page.read()
                page.close()
            except UnicodeDecodeError as e:
                print(e)
                print('-----UnicodeDecodeErrorurl:', url)
            except urllib.error.URLError as e:
                print(e)
                print("-----urlErrorurl:", url)
            except socket.timeout as e:
                print(e)
                print("-----socket timout:", url)
            else:
                rsp_data = json.loads(rsp, strict=False, object_hook=lambda d: {k: urllib.parse.unquote(v) if isinstance(v, str) else v for k, v in d.items()})
                if 'data' not in rsp_data:
                  continue
                else:
                  for image_info in rsp_data['data']:
                        if 'thumbURL' in image_info:
                            thumb_url = image_info['thumbURL']
                            image_urls.append(thumb_url)
                pn += self.__per_page
                return image_urls

    def start(self, word):
      self.__per_page = 30
      self.__start_amount = 0
      self.__amount = self.__per_page
      return self.get_images(word)


app = Flask(__name__)


@app.route("/")
def index():
    word = request.args.get('word')
    if word:
      crawler = Crawler(0.1)# 抓取延迟为 0.1
      image_urls = crawler.start(word)
      if image_urls:
            image_url = random.choice(image_urls)
            # 返回图片的 URL
            return redirect(image_url)
    return """
      <!DOCTYPE html>
      <html>
      <body>
      <h1>无法正常查找</h1>
      </body>
      </html>
    """


if __name__ == '__main__':
    app.run()

redfieldw 发表于 2023-6-28 22:32

厉害,收藏学习了

忧郁之子 发表于 2023-6-29 00:11

谢谢分享,学习了

laustar 发表于 2023-6-30 16:34

厉害了。学习一下

skylcx 发表于 2024-9-14 13:46

sixsixsix
页: [1]
查看完整版本: 用python实现调用百度图片搜索的API