python-piped-api-client/tests/test_videos.py
2022-02-27 08:40:58 +01:00

63 lines
1.6 KiB
Python

import typing as t
from tests import CLIENT
from datetime import datetime
def test_video(video_id: str='dQw4w9WgXcQ') -> None:
"""
Prints out information about a video.
"""
video = CLIENT.get_video(video_id)
short_description = video.description[:100].replace('\n', '')
print(f"""
Video ID: {video_id}
Title: {video.title}
Description: {short_description}...
Views: {video.views}
Uploaded by: {video.uploader}
Uploaded on: {video.upload_date} ({datetime.now().year - video.upload_date.year} years ago)
Duration: {video.duration}
FPS: {video.get_streams('video')[0].fps}
""")
def test_trending(country_codes: t.List[str]=['US', 'SK', 'CN']) -> None:
"""
Prints out trending videos for a specific country.
"""
for country_code in country_codes:
videos = CLIENT.get_trending(country_code)
# Nothing ever trends in China's YouTube:
if country_code == 'CN':
assert len(videos) == 0
print("\nYes, empty list works.")
for video in videos:
print(f"{video.uploader_name} >> {video.title} ({video.views} views)")
def test_get_audio(video_id: str='dQw4w9WgXcQ') -> None:
"""
Prints out the first audio stream URL for a video.
"""
video = CLIENT.get_video(video_id)
audio_stream = video.get_streams('audio')[0]
print(f"Audio stream URL: {audio_stream.url} ({audio_stream.mime_type})")
if __name__ == '__main__':
test_video()
test_trending()
test_get_audio()