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: add debug for KeyError

Browse Source
This commit is contained in:
WorldObservationLog
2024-05-30 00:23:50 +08:00
parent f17b860826
commit 5602064681
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/mp4.py
View File

@@ -1,6 +1,7 @@
import subprocess import subprocess
import sys import sys
import uuid import uuid
import pickle
from io import BytesIO from io import BytesIO
from pathlib import Path from pathlib import Path
from tempfile import TemporaryDirectory from tempfile import TemporaryDirectory
@@ -123,7 +124,15 @@ async def extract_song(raw_song: bytes, codec: str) -> SongInfo:
for trun in truns: for trun in truns:
for sample_number in range(int(trun.get("SampleCount"))): for sample_number in range(int(trun.get("SampleCount"))):
nhnt_sample_number += 1 nhnt_sample_number += 1
try:
nhnt_sample = nhnt_samples[nhnt_sample_number] nhnt_sample = nhnt_samples[nhnt_sample_number]
except KeyError as e:
with open("FOR_DEBUG_RAW_SONG.mp4", "wb") as f:
f.write(raw_song)
with open("FOR_DEBUG_NHNT_DUMP.bin", "wb") as f:
pickle.dump(nhnt_samples, f)
logger.error("An error occurred! Please send FOR_DEBUG_RAW_SONG.mp4 and FOR_DEBUG_NHNT_DUMP.bin to the developer!")
raise e
sample_data = media.read(int(nhnt_sample.get("dataLength"))) sample_data = media.read(int(nhnt_sample.get("dataLength")))
duration = int(nhnt_sample.get("duration")) duration = int(nhnt_sample.get("duration"))
samples.append(SampleInfo(descIndex=index, data=sample_data, duration=int(duration))) samples.append(SampleInfo(descIndex=index, data=sample_data, duration=int(duration)))