api.py

import os
import re

from yt_dlp import YoutubeDL

VIDEO_FORMATS = {
    'До 1080p': ['bestvideo[height<=1080][ext=mp4]+bestaudio[ext=m4a]/best[height<=1080][ext=mp4]/best', '_do1080p'],
    'Лучшее': ['bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best', '_luchsheye'],
}


# TODO: Удалять файлы через некоторое время
def delete_file(filename):
    if os.path.exists(filename):
        os.remove(filename)


def extract_video_id(url):
    video_id = re.search(r'(?<=v=)[^&#]+', url)
    if video_id:
        return video_id.group(0)
    else:
        video_id = re.search(r'(?<=v=)[^&]+', url)
        return video_id.group(0) if video_id else None


def download_video_api(url: str) -> dict:
    filename = extract_video_id(url)

    ydl_opts: dict = {
        'format': 'bestvideo[height<=1080][ext=mov]+bestaudio[ext=wav]/best[height<=1080][ext=mov]/best',
        'outtmpl': f'downloaded_videos/{filename}.mov',
        'noplaylist': True,
        'max_filesize': 52428800,
    }

    with YoutubeDL(ydl_opts) as ydl:
        info_dict = ydl.extract_info(url, download=False)

        ydl.download([url])

        video_info = {
            'title': info_dict.get('title', 'No title'),
            'filename': filename,
            'author': info_dict.get('uploader', 'No author'),
            'duration': info_dict.get('duration', 'No duration'),
        }

    return video_info