Python进阶：多线程处理案例

原创「爱情、让人受尽委屈。」 2025-03-03 14:51 37阅读 0赞

在Python中，多线程是通过`threading`模块实现的。下面我们将通过一个实际案例来演示如何使用多线程。

案例：模拟网络下载文件

假设我们有一个任务，需要下载一个大文件，并将其保存在一个特定的位置。这个过程可以分解为多个子任务：

1. 检查文件是否可用
2. 如果不可用，尝试下载
3. 下载完成后，将文件移动到目标位置

现在我们将每个子任务转化为一个线程，然后使用`Thread.join()`方法等待所有线程完成。

```python
import threading
import requests

# 子任务：检查文件并下载
def download_file(url, target_path):
    try:
        response = requests.get(url, stream=True)
        if response.status_code == 200:
            with open(target_path, 'wb')) as f:
                for chunk in response.iter_content(1024):
                    f.write(chunk)
            return True
        else:
            print(f"Failed to download file: {url}. Status code: {response.status_code}")
    except requests.exceptions.RequestException as e:
        print(f"Error occurred while downloading file: {e}")

# 子任务：将文件移动到目标位置
def move_file(file_path, target_path):
    try:
        if file_path != target_path:
            os.rename(file_path, target_path))
            print(f"File moved successfully to: {target_path}")
        else:
            print("Same source and destination path. No action taken.")
    except Exception as e:
        print(f"Error occurred while moving file: {e}")

# 主线程：创建多个线程执行任务
def main():
    url = "http://example.com/large_file.zip"
    target_path = "./downloaded_file.zip"

threads = []
    
    # 任务1：检查文件是否可用
    t1 = threading.Thread(target=download_file, args=(url, target_path + ".exists")))
    threads.append(t1)

# 任务2：如果不可用，尝试下载
    if not t1.is_alive():
        t2 = threading.Thread(target=download_file, args=(url, target_path), daemon=True))  # 创建后台线程
        threads.append(t2)

# 任务3：将下载的文件移动到目标位置
    for t in threads:
        t.join()  # 等待所有线程完成

if __name__ == "__main__":
    main()
```

这个例子展示了如何使用Python多线程来处理网络下载大文件的任务。通过合理分配任务，可以有效地利用多核处理器的优势。