Skip to content

Commit 2f965fe

Browse files
authored
Update playlists.py
1 parent a3095f9 commit 2f965fe

1 file changed

Lines changed: 29 additions & 12 deletions

File tree

tests/async/playlists.py

Lines changed: 29 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -3,29 +3,46 @@
33
import time
44
from youtubesearchpython.__future__ import *
55

6+
TIMEOUT = 60
7+
68
def print_json(data):
79
print(json.dumps(data, indent=2, ensure_ascii=False))
810

911
async def run_get(url):
1012
start = time.perf_counter()
11-
result = await Playlist.get(url)
12-
elapsed = time.perf_counter() - start
13-
print_json(result)
14-
print(f"\n⏱ Playlist.get took {elapsed:.3f} seconds\n{'-'*60}\n")
13+
try:
14+
result = await Playlist.get(url, timeout=TIMEOUT)
15+
elapsed = time.perf_counter() - start
16+
print_json(result)
17+
print(f"\n⏱ Playlist.get took {elapsed:.3f} seconds\n{'-'*60}\n")
18+
except Exception as e:
19+
elapsed = time.perf_counter() - start
20+
print_json({"error": type(e).__name__, "message": str(e)})
21+
print(f"\n⏱ Playlist.get failed after {elapsed:.3f} seconds\n{'-'*60}\n")
1522

1623
async def run_get_info(url):
1724
start = time.perf_counter()
18-
result = await Playlist.getInfo(url)
19-
elapsed = time.perf_counter() - start
20-
print_json(result)
21-
print(f"\n⏱ Playlist.getInfo took {elapsed:.3f} seconds\n{'-'*60}\n")
25+
try:
26+
result = await Playlist.getInfo(url, timeout=TIMEOUT)
27+
elapsed = time.perf_counter() - start
28+
print_json(result)
29+
print(f"\n⏱ Playlist.getInfo took {elapsed:.3f} seconds\n{'-'*60}\n")
30+
except Exception as e:
31+
elapsed = time.perf_counter() - start
32+
print_json({"error": type(e).__name__, "message": str(e)})
33+
print(f"\n⏱ Playlist.getInfo failed after {elapsed:.3f} seconds\n{'-'*60}\n")
2234

2335
async def run_get_videos(url):
2436
start = time.perf_counter()
25-
result = await Playlist.getVideos(url)
26-
elapsed = time.perf_counter() - start
27-
print_json(result)
28-
print(f"\n⏱ Playlist.getVideos took {elapsed:.3f} seconds\n{'-'*60}\n")
37+
try:
38+
result = await Playlist.getVideos(url, timeout=TIMEOUT)
39+
elapsed = time.perf_counter() - start
40+
print_json(result)
41+
print(f"\n⏱ Playlist.getVideos took {elapsed:.3f} seconds\n{'-'*60}\n")
42+
except Exception as e:
43+
elapsed = time.perf_counter() - start
44+
print_json({"error": type(e).__name__, "message": str(e)})
45+
print(f"\n⏱ Playlist.getVideos failed after {elapsed:.3f} seconds\n{'-'*60}\n")
2946

3047
async def main():
3148
url1 = "https://www.youtube.com/playlist?list=PLRBp0Fe2GpgmsW46rJyudVFlY6IYjFBIK"

0 commit comments

Comments
 (0)