Saverr/svtplay-dl
1
0
Fork 0
mirror of https://github.com/spaam/svtplay-dl.git synced 2024-11-24 04:05:39 +01:00
Code Issues Projects Releases Wiki Activity

Move filter_files to its own file

Browse Source
This commit is contained in:
Johan Andersson 2022-06-04 00:45:55 +02:00
parent 911d222159
commit 5cb26d20a8
3 changed files with 15 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
lib/svtplay_dl/fetcher/hls.py
View File

@ -7,7 +7,6 @@ import re
import time import time
from datetime import datetime from datetime import datetime
from datetime import timedelta from datetime import timedelta
from difflib import SequenceMatcher
from cryptography.hazmat.backends import default_backend from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives.ciphers import algorithms from cryptography.hazmat.primitives.ciphers import algorithms
@ -17,6 +16,7 @@ from svtplay_dl.error import ServiceError
from svtplay_dl.error import UIException from svtplay_dl.error import UIException
from svtplay_dl.fetcher import VideoRetriever from svtplay_dl.fetcher import VideoRetriever
from svtplay_dl.subtitle import subtitle from svtplay_dl.subtitle import subtitle
from svtplay_dl.utils.fetcher import filter_files
from svtplay_dl.utils.http import get_full_url from svtplay_dl.utils.http import get_full_url
from svtplay_dl.utils.http import HTTP from svtplay_dl.utils.http import HTTP
from svtplay_dl.utils.output import ETA from svtplay_dl.utils.output import ETA
@ -240,7 +240,7 @@ class HLS(VideoRetriever):
hls_time_stamp = self.kwargs.pop("hls_time_stamp", False) hls_time_stamp = self.kwargs.pop("hls_time_stamp", False)
if self.kwargs.get("filter", False): if self.kwargs.get("filter", False):
m3u8 = _filter_files(m3u8) m3u8 = filter_files(m3u8)
decryptor = None decryptor = None
size_media = len(m3u8.media_segment) size_media = len(m3u8.media_segment)
eta = ETA(size_media) eta = ETA(size_media)
@ -559,13 +559,3 @@ def _get_tuple_attribute(attribute):
attr_tuple[name] = value attr_tuple[name] = value
return attr_tuple return attr_tuple
def _filter_files(m3u8):
files = []
good = m3u8.media_segment[1]["URI"]
for segment in m3u8.media_segment:
if SequenceMatcher(None, good, segment["URI"]).ratio() > 0.6:
files.append(segment)
m3u8.media_segment = files
return m3u8

4
lib/svtplay_dl/subtitle/__init__.py
View File

@ -5,7 +5,7 @@ import xml.etree.ElementTree as ET
from io import StringIO from io import StringIO
from requests import __build__ as requests_version from requests import __build__ as requests_version
from svtplay_dl.fetcher.hls import _filter_files from svtplay_dl.utils.fetcher import filter_files
from svtplay_dl.utils.http import get_full_url from svtplay_dl.utils.http import get_full_url
from svtplay_dl.utils.http import HTTP from svtplay_dl.utils.http import HTTP
from svtplay_dl.utils.output import find_dupes from svtplay_dl.utils.output import find_dupes
@ -288,7 +288,7 @@ class subtitle:
time = 0 time = 0
subs = [] subs = []
if self.kwargs.get("filter", False): if self.kwargs.get("filter", False):
self.kwargs["m3u8"] = _filter_files(self.kwargs["m3u8"]) self.kwargs["m3u8"] = filter_files(self.kwargs["m3u8"])
for _, i in enumerate(self.kwargs["m3u8"].media_segment): for _, i in enumerate(self.kwargs["m3u8"].media_segment):
itemurl = get_full_url(i["URI"], self.url) itemurl = get_full_url(i["URI"], self.url)

11
lib/svtplay_dl/utils/fetcher.py Normal file
View File

@ -0,0 +1,11 @@
from difflib import SequenceMatcher
def filter_files(m3u8):
files = []
good = m3u8.media_segment[1]["URI"]
for segment in m3u8.media_segment:
if SequenceMatcher(None, good, segment["URI"]).ratio() > 0.6:
files.append(segment)
m3u8.media_segment = files
return m3u8