File size: 1,401 Bytes
e7ef6c8 |
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 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 |
import os
import logging
from urllib.parse import urlparse
logging.basicConfig(
level=logging.DEBUG, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
)
logger = logging.getLogger(__name__)
def get_file_extension_from_url(url):
"""
Get the file extension from a URL.
Parameters:
- url (str): URL of the file.
Returns:
str: File extension.
"""
url_path = urlparse(url).path
basename = os.path.basename(url_path)
return basename.split(".")[-1]
def get_resolution(info_dict):
"""
Get the width and height of a video from its info dictionary.
Parameters:
- info_dict (dict): Dictionary containing information about the video.
Returns:
tuple: Width and height of the video.
"""
width = 0
height = 0
if {"width", "height"} <= info_dict.keys():
width = int(info_dict["width"])
height = int(info_dict["height"])
# https://support.google.com/youtube/answer/6375112
elif info_dict["height"] == 1080:
width = 1920
height = 1080
elif info_dict["height"] == 720:
width = 1280
height = 720
elif info_dict["height"] == 480:
width = 854
height = 480
elif info_dict["height"] == 360:
width = 640
height = 360
elif info_dict["height"] == 240:
width = 426
height = 240
return width, height
|