mirror of
https://github.com/Arrowar/StreamingCommunity.git
synced 2025-06-06 19:45:24 +00:00
94 lines
3.4 KiB
Python
94 lines
3.4 KiB
Python
# 09.06.24
|
|
|
|
import os
|
|
import sys
|
|
import logging
|
|
|
|
|
|
# External libraries
|
|
import httpx
|
|
from tqdm import tqdm
|
|
|
|
|
|
# Internal utilities
|
|
from Src.Util.headers import get_headers
|
|
from Src.Util.color import Colors
|
|
from Src.Util.console import console, Panel
|
|
from Src.Util._jsonConfig import config_manager
|
|
from Src.Util.os import format_file_size
|
|
|
|
|
|
# Logic class
|
|
from ...FFmpeg import print_duration_table
|
|
|
|
|
|
# Config
|
|
TQDM_USE_LARGE_BAR = config_manager.get_int('M3U8_DOWNLOAD', 'tqdm_use_large_bar')
|
|
REQUEST_VERIFY = config_manager.get_float('REQUESTS', 'verify_ssl')
|
|
REQUEST_TIMEOUT = config_manager.get_float('REQUESTS', 'timeout')
|
|
|
|
|
|
|
|
def MP4_downloader(url: str, path: str, referer: str = None):
|
|
|
|
"""
|
|
Downloads an MP4 video from a given URL using the specified referer header.
|
|
|
|
Parameter:
|
|
- url (str): The URL of the MP4 video to download.
|
|
- path (str): The local path where the downloaded MP4 file will be saved.
|
|
- referer (str): The referer header value to include in the HTTP request headers.
|
|
"""
|
|
|
|
if "http" not in url or "https" not in url:
|
|
logging.error(f"Invalid url: {url}")
|
|
sys.exit(0)
|
|
|
|
# Make request to get content of video
|
|
logging.info(f"Make request to fetch mp4 from: {url}")
|
|
|
|
if referer != None:
|
|
headers = {'Referer': referer, 'user-agent': get_headers()}
|
|
else:
|
|
headers = {'user-agent': get_headers()}
|
|
|
|
with httpx.Client(verify=REQUEST_VERIFY, timeout=REQUEST_TIMEOUT) as client:
|
|
with client.stream("GET", url, headers=headers, timeout=10) as response:
|
|
total = int(response.headers.get('content-length', 0))
|
|
|
|
# Create bar format
|
|
if TQDM_USE_LARGE_BAR:
|
|
bar_format = (f"{Colors.YELLOW}[MP4] {Colors.WHITE}({Colors.CYAN}video{Colors.WHITE}): "
|
|
f"{Colors.RED}{{percentage:.2f}}% {Colors.MAGENTA}{{bar}} {Colors.WHITE}[ "
|
|
f"{Colors.YELLOW}{{n_fmt}}{Colors.WHITE} / {Colors.RED}{{total_fmt}} {Colors.WHITE}] "
|
|
f"{Colors.YELLOW}{{elapsed}} {Colors.WHITE}< {Colors.CYAN}{{remaining}} {Colors.WHITE}| "
|
|
f"{Colors.YELLOW}{{rate_fmt}}{{postfix}} {Colors.WHITE}]")
|
|
else:
|
|
bar_format = (f"{Colors.YELLOW}Proc{Colors.WHITE}: {Colors.RED}{{percentage:.2f}}% "
|
|
f"{Colors.WHITE}| {Colors.CYAN}{{remaining}}{{postfix}} {Colors.WHITE}]")
|
|
|
|
# Create progress bar
|
|
progress_bar = tqdm(
|
|
total=total,
|
|
ascii='░▒█',
|
|
bar_format=bar_format,
|
|
unit_scale=True,
|
|
unit_divisor=1024,
|
|
mininterval=0.05
|
|
)
|
|
|
|
# Download file
|
|
with open(path, 'wb') as file, progress_bar as bar:
|
|
for chunk in response.iter_bytes(chunk_size=1024):
|
|
if chunk:
|
|
size = file.write(chunk)
|
|
bar.update(size)
|
|
|
|
# Get summary
|
|
console.print(Panel(
|
|
f"[bold green]Download completed![/bold green]\n"
|
|
f"[cyan]File size: [bold red]{format_file_size(os.path.getsize(path))}[/bold red]\n"
|
|
f"[cyan]Duration: [bold]{print_duration_table(path, description=False, return_string=True)}[/bold]",
|
|
title=f"{os.path.basename(path.replace('.mp4', ''))}",
|
|
border_style="green"
|
|
)) |