diff options
author | pukkandan <pukkandan.ytdlp@gmail.com> | 2023-02-07 03:22:29 +0530 |
---|---|---|
committer | pukkandan <pukkandan.ytdlp@gmail.com> | 2023-02-08 07:28:46 +0530 |
commit | f6a765ceb59c55aea06921880c1c87d1ff36e5de (patch) | |
tree | 4120a1128be7c3ec48f0d81f6832fe55e25cff5b /yt_dlp/extractor/ivi.py | |
parent | 754c84e2e416cf6609dd0e4632b4985a08d34043 (diff) | |
download | hypervideo-pre-f6a765ceb59c55aea06921880c1c87d1ff36e5de.tar.lz hypervideo-pre-f6a765ceb59c55aea06921880c1c87d1ff36e5de.tar.xz hypervideo-pre-f6a765ceb59c55aea06921880c1c87d1ff36e5de.zip |
[dependencies] Standardize `Cryptodome` imports
Diffstat (limited to 'yt_dlp/extractor/ivi.py')
-rw-r--r-- | yt_dlp/extractor/ivi.py | 26 |
1 files changed, 7 insertions, 19 deletions
diff --git a/yt_dlp/extractor/ivi.py b/yt_dlp/extractor/ivi.py index dc6a48196..96220bea9 100644 --- a/yt_dlp/extractor/ivi.py +++ b/yt_dlp/extractor/ivi.py @@ -2,11 +2,8 @@ import json import re from .common import InfoExtractor -from ..utils import ( - ExtractorError, - int_or_none, - qualities, -) +from ..dependencies import Cryptodome +from ..utils import ExtractorError, int_or_none, qualities class IviIE(InfoExtractor): @@ -94,18 +91,8 @@ class IviIE(InfoExtractor): for site in (353, 183): content_data = (data % site).encode() if site == 353: - try: - from Cryptodome.Cipher import Blowfish - from Cryptodome.Hash import CMAC - pycryptodome_found = True - except ImportError: - try: - from Crypto.Cipher import Blowfish - from Crypto.Hash import CMAC - pycryptodome_found = True - except ImportError: - pycryptodome_found = False - continue + if not Cryptodome: + continue timestamp = (self._download_json( self._LIGHT_URL, video_id, @@ -118,7 +105,8 @@ class IviIE(InfoExtractor): query = { 'ts': timestamp, - 'sign': CMAC.new(self._LIGHT_KEY, timestamp.encode() + content_data, Blowfish).hexdigest(), + 'sign': Cryptodome.Hash.CMAC.new(self._LIGHT_KEY, timestamp.encode() + content_data, + Cryptodome.Cipher.Blowfish).hexdigest(), } else: query = {} @@ -138,7 +126,7 @@ class IviIE(InfoExtractor): extractor_msg = 'Video %s does not exist' elif site == 353: continue - elif not pycryptodome_found: + elif not Cryptodome: raise ExtractorError('pycryptodomex not found. Please install', expected=True) elif message: extractor_msg += ': ' + message |