[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.utils import bug_reports_message
from yt_dlp.extractor.youtube.pot._builtin.memory_cache import MemoryLRUPCP, memorylru_preference, initialize_global_cache 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.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: 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.version import __version__
from yt_dlp.extractor.youtube.pot._builtin.webpo_cachespec import WebPoPCSP 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() @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 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.networking import Request
from yt_dlp.extractor.youtube.pot.provider import ( from yt_dlp.extractor.youtube.pot.provider import (
PoTokenResponse, PoTokenResponse,
@ -30,7 +28,7 @@ from yt_dlp.extractor.youtube.pot.provider import (
register_preference, 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): class ExamplePTP(PoTokenProvider):

View File

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

View File

@ -15,6 +15,13 @@ from yt_dlp.extractor.youtube.pot._provider import (
IEContentProvider, IEContentProvider,
IEContentProviderLogger, 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 ( from yt_dlp.extractor.youtube.pot.cache import (
CacheProviderWritePolicy, CacheProviderWritePolicy,
PoTokenCacheProvider, PoTokenCacheProvider,
@ -30,13 +37,6 @@ from yt_dlp.extractor.youtube.pot.provider import (
PoTokenResponse, PoTokenResponse,
provider_bug_report_message, 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 from yt_dlp.utils import ExtractorError, bug_reports_message, format_field, join_nonempty, traverse_obj
if typing.TYPE_CHECKING: 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_preference_generic,
register_provider_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.extractor.youtube.pot.provider import PoTokenRequest
from yt_dlp.globals import _pot_cache_provider_preferences, _pot_cache_providers, _pot_pcs_providers
class PoTokenCacheProviderError(IEContentProviderError): class PoTokenCacheProviderError(IEContentProviderError):

View File

@ -16,7 +16,7 @@ from yt_dlp.extractor.youtube.pot._provider import (
register_preference_generic, register_preference_generic,
register_provider_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.networking import Request
from yt_dlp.utils import traverse_obj from yt_dlp.utils import traverse_obj
from yt_dlp.utils.networking import HTTPHeaderDict from yt_dlp.utils.networking import HTTPHeaderDict

View File

@ -28,12 +28,3 @@ plugin_ies_overrides = Indirect(defaultdict(list))
# Misc # Misc
IN_CLI = Indirect(False) IN_CLI = Indirect(False)
LAZY_EXTRACTORS = Indirect(None) # `False`=force, `None`=disabled, `True`=enabled 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({})