fix: type hints (GH-2)
This commit is contained in:
@@ -3,6 +3,7 @@ from Crypto.Cipher import AES
|
|||||||
import logging
|
import logging
|
||||||
import os
|
import os
|
||||||
import subprocess
|
import subprocess
|
||||||
|
from typing import Dict, List, Tuple
|
||||||
|
|
||||||
from runtime import RuntimeInfo
|
from runtime import RuntimeInfo
|
||||||
|
|
||||||
@@ -15,7 +16,7 @@ def general_aes_ctr_decrypt(data: bytes, key: bytes, nonce: bytes) -> bytes:
|
|||||||
return cipher.decrypt(data)
|
return cipher.decrypt(data)
|
||||||
|
|
||||||
|
|
||||||
def decrypt_process(runtimes: dict[str, RuntimeInfo], sequences: list[tuple[str, bytes]], args):
|
def decrypt_process(runtimes: Dict[str, RuntimeInfo], sequences: List[Tuple[str, bytes]], args):
|
||||||
logger = logging.getLogger('shot')
|
logger = logging.getLogger('shot')
|
||||||
output_dir: str = args.output_dir or args.directory
|
output_dir: str = args.output_dir or args.directory
|
||||||
for path, data in sequences:
|
for path, data in sequences:
|
||||||
@@ -134,14 +135,14 @@ def main():
|
|||||||
specified_runtime = None
|
specified_runtime = None
|
||||||
runtimes = {}
|
runtimes = {}
|
||||||
|
|
||||||
sequences: list[tuple[str, bytes]] = []
|
sequences: List[Tuple[str, bytes]] = []
|
||||||
|
|
||||||
if args.output_dir and not os.path.exists(args.output_dir):
|
if args.output_dir and not os.path.exists(args.output_dir):
|
||||||
os.makedirs(args.output_dir)
|
os.makedirs(args.output_dir)
|
||||||
|
|
||||||
dir_path: str
|
dir_path: str
|
||||||
dirs: list[str]
|
dirs: List[str]
|
||||||
files: list[str]
|
files: List[str]
|
||||||
for dir_path, dirs, files in os.walk(args.directory, followlinks=False):
|
for dir_path, dirs, files in os.walk(args.directory, followlinks=False):
|
||||||
for d in ['__pycache__', 'site-packages']:
|
for d in ['__pycache__', 'site-packages']:
|
||||||
if d in dirs:
|
if d in dirs:
|
||||||
|
Reference in New Issue
Block a user