mirror of
https://github.com/UrloMythus/UnHided.git
synced 2026-04-09 02:40:47 +00:00
33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
from typing import Dict, Type
|
|
|
|
from mediaflow_proxy.extractors.base import BaseExtractor, ExtractorError
|
|
from mediaflow_proxy.extractors.doodstream import DoodStreamExtractor
|
|
from mediaflow_proxy.extractors.livetv import LiveTVExtractor
|
|
from mediaflow_proxy.extractors.mixdrop import MixdropExtractor
|
|
from mediaflow_proxy.extractors.uqload import UqloadExtractor
|
|
from mediaflow_proxy.extractors.streamtape import StreamtapeExtractor
|
|
from mediaflow_proxy.extractors.supervideo import SupervideoExtractor
|
|
|
|
|
|
|
|
|
|
class ExtractorFactory:
|
|
"""Factory for creating URL extractors."""
|
|
|
|
_extractors: Dict[str, Type[BaseExtractor]] = {
|
|
"Doodstream": DoodStreamExtractor,
|
|
"Uqload": UqloadExtractor,
|
|
"Mixdrop": MixdropExtractor,
|
|
"Streamtape": StreamtapeExtractor,
|
|
"Supervideo": SupervideoExtractor,
|
|
"LiveTV": LiveTVExtractor,
|
|
}
|
|
|
|
@classmethod
|
|
def get_extractor(cls, host: str, request_headers: dict) -> BaseExtractor:
|
|
"""Get appropriate extractor instance for the given host."""
|
|
extractor_class = cls._extractors.get(host)
|
|
if not extractor_class:
|
|
raise ExtractorError(f"Unsupported host: {host}")
|
|
return extractor_class(request_headers)
|