当前位置：首页 > article >正文

python爬虫初体验（二）

article 2024/9/21 21:21:22

在Python中，每个模块都有一个内置的变量 name，用于表示当前模块的名称。当一个Python文件被执行时，Python解释器会首先将该文件作为一个模块导入，并执行其中的代码。此时，__name__的值为模块的名称。

作用

模块可被导入也可直接执行：使一个Python脚本既可以作为一个独立的程序运行，也可以被其他脚本作为模块导入。
减少全局污染：当脚本作为模块被导入时，可以避免执行不必要的代码，提高代码的整洁度和可维护性。
测试代码：可以编写一些测试代码，只在直接运行脚本时执行，而不影响作为模块导入时的行为。

使用场景

脚本和模块分离：使脚本既可以独立运行，也可以作为模块导入。
单元测试：在脚本中编写一些测试代码，只在直接运行时执行。
初始化资源：在脚本直接运行时初始化一些资源，例如数据库连接等。

示例

假设有一个Python脚本 example.py：

import csv
import requests
from bs4 import BeautifulSoup

def greet(name):
    print("Hello, {}!".format(name))

def main():
	# 检查命令行参数数量
    if len(sys.argv) < 2:
        print("Usage: python script.py <name>")
        sys.exit(1) # 退出程序，返回错误码1

	# 获取命令行参数中的名字
    name = sys.argv[1]
    # 调用greet函数
    greet(name)
    
def amazon():
  # 目标URL
  url = 'https://www.shanghairanking.cn/rankings/bcur/2024'
  # 请求头，模拟浏览器行为，防止被服务器识别为爬虫。
  headers = {
    'Connection': 'keep-alive',
    'Cache-Control': 'no-store, no-cache, must-revalidate',
    'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36',
    'Accept': '*/*',
    'Accept-Encoding': 'gzip, deflate, br, zstd',
    'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8',
    'Referer': 'https://www.shanghairanking.cn/rankings/bcur/2024',
  }
  # 创建一个会话对象，保持连接
  session = requests.session()

  # 先访问一次首页，模拟用户行为
  session.get('https://www.shanghairanking.cn')

  # 发送带有headers的GET请求并获取响应内容
  resHtml = session.get(url, headers=headers).content.decode('utf-8')

  # 使用BeautifulSoup解析HTML内容
  html_soup = BeautifulSoup(resHtml, 'html.parser')

  # 查找表格中的所有行
  all_goods_li = html_soup.find('table').find_all('tr')

  # 打开CSV文件准备写入
  with open('output.csv', 'wb') as csvfile:
  	# 创建一个csv.writer对象
    csv_writer = csv.writer(csvfile)

    for row in all_goods_li:

      goods_info_list = []

      rank = row.find('div', class_="ranking")
      img_link = row.find('img')['src']
      name = row.find('span', class_='name-cn')
      tags = row.find('p', class_="tags")
      if name and img_link:

        goods_info_list.append(rank.get_text(strip=True).encode('utf-8'))
        goods_info_list.append(img_link)
        # 将name添加到列表中，并转换为UTF-8编码
        goods_info_list.append(name.get_text(strip=True).encode('utf-8'))
        goods_info_list.append(tags.get_text(strip=True).encode('utf-8'))

      csv_writer.writerow(goods_info_list)
 
# 如果此脚本被直接运行，则执行amazon函数
if __name__ == '__main__':
  main()

如果你直接运行这个脚本：

python example.py world

控制台会输出：Hello, World!

作为模块导入如果你在另一个脚本中导入 example.py：

import example
example.greet("Alice")

example.amazon()

则输出Hello, Alice!并执行amazon脚本。example.py 中的 if __name__ == '__main__': 块中的代码不会被执行，因为 example.py 是作为模块被导入的，其 __name__ 的值为 'example'，而不是 '__main__'。