github/StreamingCommunity
1
1
Fork 0
mirror of https://github.com/Arrowar/StreamingCommunity.git synced 2025-06-07 12:05:35 +00:00
Code Issues 2 Packages Projects Releases Wiki Activity

feat(config): add config feature to variables

Browse Source
This commit is contained in:
Cohvir 2024-03-04 16:10:56 +01:00
parent 161794ac43
commit a0eacfd478
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
Src/Lib/FFmpeg/my_m3u8.py
View File

@ -3,6 +3,7 @@
# Class import # Class import
from Src.Util.console import console from Src.Util.console import console
from Src.Util.headers import get_headers from Src.Util.headers import get_headers
from Src.Util.config import config
from Src.Lib.FFmpeg.util import print_duration_table from Src.Lib.FFmpeg.util import print_duration_table
# Import # Import
@ -20,8 +21,8 @@ warnings.filterwarnings("ignore", category=UserWarning, module="cryptography")
# Variable # Variable
MAX_WORKER = 20 MAX_WORKER = 20
DOWNLOAD_SUB = True DOWNLOAD_SUB = config['download_subtitles']
DOWNLOAD_DEFAULT_LANGUAGE = False DOWNLOAD_DEFAULT_LANGUAGE = config['download_default_language']
failed_segments = [] failed_segments = []
@ -113,7 +114,7 @@ class M3U8_Parser:
def download_subtitle(self): def download_subtitle(self):
"""Download all subtitle if present""" """Download all subtitle if present"""
path = os.path.join("videos", "subtitle") path = os.path.join(config['root_path'], "Film o Serie", "nome film o stagione serie")
if self.subtitle_playlist: if self.subtitle_playlist:
for sub_info in self.subtitle_playlist: for sub_info in self.subtitle_playlist:
@ -128,9 +129,10 @@ class M3U8_Parser:
sub_parse = M3U8_Parser() sub_parse = M3U8_Parser()
sub_parse.parse_data(req_sub_content.text) sub_parse.parse_data(req_sub_content.text)
url_subititle = sub_parse.subtitle[0] url_subtitle = sub_parse.subtitle[0]
# Movie_Name.[Language_Code].vtt
open(os.path.join(path, name_language + ".vtt"), "wb").write(requests.get(url_subititle).content) # Movie_Name.[Language_Code].forced.vtt
open(os.path.join(path, "nome film o serie" ,f".{name_language}" + "se conteneva forced: .forced" + ".vtt"), "wb").write(requests.get(url_subtitle).content)
else: else:
console.log("[red]No subtitle found") console.log("[red]No subtitle found")
@ -346,7 +348,7 @@ class M3U8_Downloader:
def start(self): def start(self):
video_m3u8 = M3U8_Segments(self.m3u8_url, self.key) video_m3u8 = M3U8_Segments(self.m3u8_url, self.key)
console.log("[green]Download video ts") console.log("[green]Downloading video ts")
video_m3u8.get_info() video_m3u8.get_info()
video_m3u8.download_ts() video_m3u8.download_ts()
video_m3u8.join(self.video_path) video_m3u8.join(self.video_path)
@ -355,7 +357,7 @@ class M3U8_Downloader:
if self.m3u8_audio != None: if self.m3u8_audio != None:
audio_m3u8 = M3U8_Segments(self.m3u8_audio, self.key) audio_m3u8 = M3U8_Segments(self.m3u8_audio, self.key)
console.log("[green]Download audio ts") console.log("[green]Downloading audio ts")
audio_m3u8.get_info() audio_m3u8.get_info()
audio_m3u8.download_ts() audio_m3u8.download_ts()
audio_m3u8.join(self.audio_path) audio_m3u8.join(self.audio_path)
@ -462,7 +464,7 @@ def download_m3u8(m3u8_playlist=None, m3u8_index = None, m3u8_audio=None, m3u8_s
else: else:
parse_class_m3u8_sub.parse_data(m3u8_subtitle) parse_class_m3u8_sub.parse_data(m3u8_subtitle)
# Download subtitle if present ( normaly in m3u8 playlist ) # Download subtitle if present ( normally in m3u8 playlist )
if DOWNLOAD_SUB: if DOWNLOAD_SUB:
parse_class_m3u8_sub.download_subtitle() parse_class_m3u8_sub.download_subtitle()