aboutsummaryrefslogtreecommitdiffstats
path: root/devscripts/generate_aes_testdata.py
diff options
context:
space:
mode:
Diffstat (limited to 'devscripts/generate_aes_testdata.py')
-rw-r--r--devscripts/generate_aes_testdata.py8
1 files changed, 6 insertions, 2 deletions
diff --git a/devscripts/generate_aes_testdata.py b/devscripts/generate_aes_testdata.py
index c7d83f1a7..7f3c88bcf 100644
--- a/devscripts/generate_aes_testdata.py
+++ b/devscripts/generate_aes_testdata.py
@@ -1,11 +1,15 @@
#!/usr/bin/env python3
-import codecs
+
+# Allow direct execution
import os
-import subprocess
import sys
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+import codecs
+import subprocess
+
from yt_dlp.aes import aes_encrypt, key_expansion
from yt_dlp.utils import intlist_to_bytes