diff options
Diffstat (limited to 'devscripts/generate_aes_testdata.py')
| -rw-r--r-- | devscripts/generate_aes_testdata.py | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/devscripts/generate_aes_testdata.py b/devscripts/generate_aes_testdata.py index 09feeaa..f131e47 100644 --- a/devscripts/generate_aes_testdata.py +++ b/devscripts/generate_aes_testdata.py @@ -1,15 +1,17 @@ #!/usr/bin/env python3 -from __future__ import unicode_literals - -import codecs -import subprocess +# Allow direct execution import os import sys + sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) -from hypervideo_dl.utils import intlist_to_bytes + +import codecs +import subprocess + from hypervideo_dl.aes import aes_encrypt, key_expansion +from hypervideo_dl.utils import intlist_to_bytes secret_msg = b'Secret message goes here' |