|
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"]) |
|
|
|
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 |
|
|
