zhaarey/AppleMusicDecrypt
1
0
Fork 0
mirror of https://github.com/zhaarey/AppleMusicDecrypt.git synced 2025-10-23 15:11:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: retry when raise DecryptException

Browse Source
This commit is contained in:
WorldObservationLog
2024-05-08 19:23:25 +08:00
parent 4586c7b4bb
commit 4af2b17a75
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/decrypt.py
View File

@@ -1,6 +1,8 @@
import asyncio import asyncio
import logging
from loguru import logger from loguru import logger
from tenacity import retry, retry_if_exception_type, stop_after_attempt, before_sleep_log
from src.adb import Device from src.adb import Device
from src.exceptions import DecryptException from src.exceptions import DecryptException
@@ -9,6 +11,8 @@ from src.mp4 import SongInfo, SampleInfo
from src.types import defaultId, prefetchKey from src.types import defaultId, prefetchKey
@retry(retry=retry_if_exception_type(DecryptException), stop=stop_after_attempt(3),
before_sleep=before_sleep_log(logger, logging.WARN))
async def decrypt(info: SongInfo, keys: list[str], manifest: Datum, device: Device) -> bytes: async def decrypt(info: SongInfo, keys: list[str], manifest: Datum, device: Device) -> bytes:
async with device.decryptLock: async with device.decryptLock:
logger.info(f"Decrypting song: {manifest.attributes.artistName} - {manifest.attributes.name}") logger.info(f"Decrypting song: {manifest.attributes.artistName} - {manifest.attributes.name}")
@@ -28,7 +32,12 @@ async def decrypt(info: SongInfo, keys: list[str], manifest: Datum, device: Devi
writer.write(bytes([len(key_uri)])) writer.write(bytes([len(key_uri)]))
writer.write(key_uri.encode("utf-8")) writer.write(key_uri.encode("utf-8"))
last_index = sample.descIndex last_index = sample.descIndex
try:
result = await decrypt_sample(writer, reader, sample) result = await decrypt_sample(writer, reader, sample)
except DecryptException as e:
writer.write(bytes([0, 0, 0, 0]))
writer.close()
raise e
decrypted += result decrypted += result
writer.write(bytes([0, 0, 0, 0])) writer.write(bytes([0, 0, 0, 0]))
writer.close() writer.close()