2024-03-26 10:08:24 +01:00

135 lines
4.7 KiB
Python

# 3.12.23 -> 10.12.23
# General import
import binascii
import json
import os
import re
import requests
import sys
from bs4 import BeautifulSoup
# Class import
from Src.Lib.FFmpeg.my_m3u8 import download_m3u8
from Src.Lib.FFmpeg.util import audio_extractor_m3u8
from Src.Util.config import config
from Src.Util.console import console
from Src.Util.headers import get_headers
def get_iframe(id_title, domain):
req = requests.get(url=f"https://streamingcommunity.{domain}/iframe/{id_title}", headers={
"User-agent": get_headers()
})
if req.ok:
url_embed = BeautifulSoup(req.text, "lxml").find("iframe").get("src")
req_embed = requests.get(url_embed, headers={"User-agent": get_headers()}).text
try:
return BeautifulSoup(req_embed, "lxml").find("body").find("script").text
except Exception:
console.log("[red]Couldn't play this video file (video not available)")
sys.exit(0)
else:
console.log(f"[red]Error: {req.status_code}")
sys.exit(0)
def select_quality(json_win_param):
if json_win_param['token1080p']:
return "1080p"
elif json_win_param['token720p']:
return "720p"
elif json_win_param['token480p']:
return "480p"
else:
return "360p"
def parse_content(embed_content):
# Parse parameter from req embed content
win_video = re.search(r"window.video = {.*}", str(embed_content)).group()
win_param = re.search(r"params: {[\s\S]*}", str(embed_content)).group()
# Parse parameter to make read for json
json_win_video = "{" + win_video.split("{")[1].split("}")[0] + "}"
json_win_param = ("{" + win_param.split("{")[1]
.split("}")[0]
.replace("\n", "")
.replace(" ", "") + "}")
json_win_param = json_win_param.replace(",}", "}").replace("'", '"')
return json.loads(json_win_video), json.loads(json_win_param), select_quality(json.loads(json_win_param))
def get_m3u8_url(json_win_video, json_win_param, render_quality):
token_render = f"token{render_quality}"
return f"https://vixcloud.co/playlist/{json_win_video['id']}?type=video&rendition={render_quality}&token={json_win_param[token_render]}&expires={json_win_param['expires']}"
def get_playlist(json_win_video, json_win_param, render_quality):
token_render = f"token{render_quality}"
return f"https://vixcloud.co/playlist/{json_win_video['id']}?token={json_win_param['token']}&{token_render}={json_win_param[token_render]}&expires={json_win_param['expires']}"
def get_m3u8_key():
response = requests.get('https://vixcloud.co/storage/enc.key')
if response.ok:
return binascii.hexlify(response.content).decode('utf-8')
else:
console.log(f"[red]Error: {response.status_code}")
sys.exit(0)
def get_m3u8_audio(json_win_video, json_win_param, title_name, token_render):
req = requests.get(
f'https://vixcloud.co/playlist/{json_win_video["id"]}',
params={'token': json_win_param['token'],
'expires': json_win_param["expires"]},
headers={
'referer':
f'https://vixcloud.co/embed/{json_win_video["id"]}?token={json_win_param[token_render]}&title={title_name}&referer=1&expires={json_win_param["expires"]}'
})
if req.ok:
result = audio_extractor_m3u8(req)
return result
else:
console.log(f"[red]Error: {req.status_code}")
sys.exit(0)
# [func \ main]
def main_dw_film(id_film, title_name, domain):
embed_content = get_iframe(id_film, domain)
json_win_video, json_win_param, render_quality = parse_content(embed_content)
token_render = f"token{render_quality}"
console.print(f"[blue]Selected quality => [red]{render_quality}")
m3u8_url = get_m3u8_url(json_win_video, json_win_param, render_quality)
m3u8_key = get_m3u8_key()
m3u8_playlist = get_playlist(json_win_video, json_win_param, render_quality)
mp4_name = title_name.replace("+", " ").replace(",", "").replace("-", "_")
mp4_format = mp4_name + ".mp4"
mp4_path = os.path.join(config['root_path'], config['movies_folder_name'], mp4_name, mp4_format)
m3u8_url_audio = get_m3u8_audio(json_win_video, json_win_param, title_name, token_render)
if m3u8_url_audio is not None:
console.print("[blue]Using m3u8 audio => [red]True")
subtitle_path = os.path.join(config['root_path'], config['movies_folder_name'], mp4_name)
download_m3u8(
m3u8_index = m3u8_url,
m3u8_audio = m3u8_url_audio,
m3u8_subtitle = m3u8_playlist,
key = m3u8_key,
output_filename = mp4_path,
subtitle_folder = subtitle_path,
content_name = mp4_name
)