97 lines
3 KiB
Python
97 lines
3 KiB
Python
import math
|
|
import os
|
|
from urllib.parse import quote
|
|
from .Utils import calcDownloadPath, StreamingError
|
|
from .ffprobe import ffprobe
|
|
|
|
|
|
class M3u8:
|
|
duration: float = 6.000000
|
|
|
|
ok_vcodecs: tuple = (
|
|
"h264",
|
|
"x264",
|
|
"avc",
|
|
)
|
|
ok_acodecs: tuple = (
|
|
"aac",
|
|
"m4a",
|
|
"mp3",
|
|
)
|
|
|
|
def __init__(self, url: str, segment_duration: float = None):
|
|
self.url = url
|
|
self.duration = float(segment_duration) if segment_duration is not None else self.duration
|
|
|
|
async def make_stream(self, download_path: str, file: str) -> str:
|
|
realFile: str = calcDownloadPath(basePath=download_path, folder=file, createPath=False)
|
|
|
|
if not os.path.exists(realFile):
|
|
raise StreamingError(f"File '{realFile}' does not exist.")
|
|
|
|
try:
|
|
ff = await ffprobe(realFile)
|
|
except UnicodeDecodeError:
|
|
pass
|
|
|
|
if "duration" not in ff.metadata:
|
|
raise StreamingError(f"Unable to get '{realFile}' play duration.")
|
|
|
|
duration: float = float(ff.metadata.get("duration"))
|
|
|
|
m3u8 = []
|
|
|
|
m3u8.append("#EXTM3U")
|
|
m3u8.append("#EXT-X-VERSION:3")
|
|
m3u8.append(f"#EXT-X-TARGETDURATION:{int(self.duration)}")
|
|
m3u8.append("#EXT-X-MEDIA-SEQUENCE:0")
|
|
m3u8.append("#EXT-X-PLAYLIST-TYPE:VOD")
|
|
|
|
segmentSize: float = "{:.6f}".format(self.duration)
|
|
splits: int = math.ceil(duration / self.duration)
|
|
|
|
segmentParams: dict = {}
|
|
|
|
for stream in ff.streams():
|
|
if stream.is_video():
|
|
if stream.codec_name not in self.ok_vcodecs:
|
|
segmentParams["vc"] = 1
|
|
if stream.is_audio():
|
|
if stream.codec_name not in self.ok_acodecs:
|
|
segmentParams["ac"] = 1
|
|
|
|
for i in range(splits):
|
|
if (i + 1) == splits:
|
|
segmentSize = "{:.6f}".format(duration - (i * self.duration))
|
|
segmentParams.update({"sd": segmentSize})
|
|
|
|
m3u8.append(f"#EXTINF:{segmentSize},")
|
|
|
|
url = f"{self.url}api/player/segments/{i}/{quote(file)}.ts"
|
|
if len(segmentParams) > 0:
|
|
url += "?" + "&".join([f"{key}={value}" for key, value in segmentParams.items()])
|
|
|
|
m3u8.append(url)
|
|
|
|
m3u8.append("#EXT-X-ENDLIST")
|
|
|
|
return "\n".join(m3u8)
|
|
|
|
async def make_subtitle(self, download_path: str, file: str, duration: float) -> str:
|
|
realFile: str = calcDownloadPath(basePath=download_path, folder=file, createPath=False)
|
|
|
|
if not os.path.exists(realFile):
|
|
raise StreamingError(f"File '{realFile}' does not exist.")
|
|
|
|
m3u8 = []
|
|
|
|
m3u8.append("#EXTM3U")
|
|
m3u8.append("#EXT-X-VERSION:3")
|
|
m3u8.append(f"#EXT-X-TARGETDURATION:{int(self.duration)}")
|
|
m3u8.append("#EXT-X-MEDIA-SEQUENCE:0")
|
|
m3u8.append("#EXT-X-PLAYLIST-TYPE:VOD")
|
|
m3u8.append(f"#EXTINF:{duration},")
|
|
m3u8.append(f"{self.url}api/player/subtitle/{quote(file)}.vtt")
|
|
m3u8.append("#EXT-X-ENDLIST")
|
|
|
|
return "\n".join(m3u8)
|