forked from x4nth055/pythoncode-tutorials
- Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmultiple_threads.py
More file actions
Latest commit
35 lines (28 loc) · 1.14 KB
/
Copy pathmultiple_threads.py
File metadata and controls
35 lines (28 loc) · 1.14 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
importrequests
fromconcurrent.futuresimportThreadPoolExecutor
fromtimeimportperf_counter
# number of threads to spawn
n_threads=5
# read 1024 bytes every time
buffer_size=1024
defdownload(url):
# download the body of response by chunk, not immediately
response=requests.get(url, stream=True)
# get the file name
filename=url.split("/")[-1]
withopen(filename, "wb") asf:
fordatainresponse.iter_content(buffer_size):
# write data read to the file
f.write(data)
if__name__=="__main__":
urls= [
"https://cdn.pixabay.com/photo/2018/01/14/23/12/nature-3082832__340.jpg",
"https://cdn.pixabay.com/photo/2013/10/02/23/03/dawn-190055__340.jpg",
"https://cdn.pixabay.com/photo/2016/10/21/14/50/plouzane-1758197__340.jpg",
"https://cdn.pixabay.com/photo/2016/11/29/05/45/astronomy-1867616__340.jpg",
"https://cdn.pixabay.com/photo/2014/07/28/20/39/landscape-404072__340.jpg",
] *5
t=perf_counter()
withThreadPoolExecutor(max_workers=n_threads) aspool:
pool.map(download, urls)
print(f"Time took: {perf_counter() -t:.2f}s")