[ie/youtube] Move registry from globals to pot module

This commit is contained in:
coletdjnz 2025-04-12 09:28:22 +12:00
parent 71de622d19
commit a0e036897b
No known key found for this signature in database
GPG Key ID: 91984263BB39894A
9 changed files with 25 additions and 24 deletions

View File

@ -6,7 +6,7 @@ from yt_dlp.extractor.youtube.pot._provider import IEContentProvider, BuiltInIEC
from yt_dlp.utils import bug_reports_message
from yt_dlp.extractor.youtube.pot._builtin.memory_cache import MemoryLRUPCP, memorylru_preference, initialize_global_cache
from yt_dlp.version import __version__
from yt_dlp.globals import _pot_memory_cache, _pot_cache_providers
from yt_dlp.extractor.youtube.pot._registry import _pot_cache_providers, _pot_memory_cache
class TestMemoryLRUPCS:

View File

@ -11,7 +11,7 @@ from yt_dlp.extractor.youtube.pot.provider import (
from yt_dlp.version import __version__
from yt_dlp.extractor.youtube.pot._builtin.webpo_cachespec import WebPoPCSP
from yt_dlp.globals import _pot_pcs_providers
from yt_dlp.extractor.youtube.pot._registry import _pot_pcs_providers
@pytest.fixture()

View File

@ -18,8 +18,6 @@ from yt_dlp.extractor.youtube.pot.cache import (
import yt_dlp.extractor.youtube.pot.cache as cache
from yt_dlp.globals import _pot_cache_providers, _pot_cache_provider_preferences, _pot_pcs_providers
from yt_dlp.networking import Request
from yt_dlp.extractor.youtube.pot.provider import (
PoTokenResponse,
@ -30,7 +28,7 @@ from yt_dlp.extractor.youtube.pot.provider import (
register_preference,
)
from yt_dlp.globals import _pot_providers, _ptp_preferences
from yt_dlp.extractor.youtube.pot._registry import _pot_providers, _ptp_preferences, _pot_pcs_providers, _pot_cache_providers, _pot_cache_provider_preferences
class ExamplePTP(PoTokenProvider):

View File

@ -6,12 +6,12 @@ from collections import OrderedDict
from threading import Lock
from yt_dlp.extractor.youtube.pot._provider import BuiltInIEContentProvider
from yt_dlp.extractor.youtube.pot._registry import _pot_memory_cache
from yt_dlp.extractor.youtube.pot.cache import (
PoTokenCacheProvider,
register_preference,
register_provider,
)
from yt_dlp.globals import _pot_memory_cache
def initialize_global_cache(max_size: int):

View File

@ -15,6 +15,13 @@ from yt_dlp.extractor.youtube.pot._provider import (
IEContentProvider,
IEContentProviderLogger,
)
from yt_dlp.extractor.youtube.pot._registry import (
_pot_cache_provider_preferences,
_pot_cache_providers,
_pot_pcs_providers,
_pot_providers,
_ptp_preferences,
)
from yt_dlp.extractor.youtube.pot.cache import (
CacheProviderWritePolicy,
PoTokenCacheProvider,
@ -30,13 +37,6 @@ from yt_dlp.extractor.youtube.pot.provider import (
PoTokenResponse,
provider_bug_report_message,
)
from yt_dlp.globals import (
_pot_cache_provider_preferences,
_pot_cache_providers,
_pot_pcs_providers,
_pot_providers,
_ptp_preferences,
)
from yt_dlp.utils import ExtractorError, bug_reports_message, format_field, join_nonempty, traverse_obj
if typing.TYPE_CHECKING:

View File

@ -0,0 +1,8 @@
from yt_dlp.globals import Indirect
_pot_providers = Indirect({})
_ptp_preferences = Indirect(set())
_pot_pcs_providers = Indirect({})
_pot_cache_providers = Indirect({})
_pot_cache_provider_preferences = Indirect(set())
_pot_memory_cache = Indirect({})

View File

@ -13,8 +13,12 @@ from yt_dlp.extractor.youtube.pot._provider import (
register_preference_generic,
register_provider_generic,
)
from yt_dlp.extractor.youtube.pot._registry import (
_pot_cache_provider_preferences,
_pot_cache_providers,
_pot_pcs_providers,
)
from yt_dlp.extractor.youtube.pot.provider import PoTokenRequest
from yt_dlp.globals import _pot_cache_provider_preferences, _pot_cache_providers, _pot_pcs_providers
class PoTokenCacheProviderError(IEContentProviderError):

View File

@ -16,7 +16,7 @@ from yt_dlp.extractor.youtube.pot._provider import (
register_preference_generic,
register_provider_generic,
)
from yt_dlp.globals import _pot_providers, _ptp_preferences
from yt_dlp.extractor.youtube.pot._registry import _pot_providers, _ptp_preferences
from yt_dlp.networking import Request
from yt_dlp.utils import traverse_obj
from yt_dlp.utils.networking import HTTPHeaderDict

View File

@ -28,12 +28,3 @@ plugin_ies_overrides = Indirect(defaultdict(list))
# Misc
IN_CLI = Indirect(False)
LAZY_EXTRACTORS = Indirect(None) # `False`=force, `None`=disabled, `True`=enabled
# YouTube extractor - Internal only
_pot_providers = Indirect({})
_ptp_preferences = Indirect(set())
_pot_pcs_providers = Indirect({})
_pot_cache_providers = Indirect({})
_pot_cache_provider_preferences = Indirect(set())
_pot_memory_cache = Indirect({})