diff --git a/extract_otp_secret_keys.py b/extract_otp_secret_keys.py index 2c16f1e..7aafcdc 100644 --- a/extract_otp_secret_keys.py +++ b/extract_otp_secret_keys.py @@ -52,13 +52,12 @@ import sys import urllib.parse as urlparse from enum import Enum from operator import add -from typing import TextIO, Any, TypedDict +from typing import Any, TextIO, TypedDict from qrcode import QRCode # type: ignore import protobuf_generated_python.google_auth_pb2 as pb - try: import cv2 # type: ignore import numpy diff --git a/test_extract_otp_secret_keys_pytest.py b/test_extract_otp_secret_keys_pytest.py index 7348978..0e81543 100644 --- a/test_extract_otp_secret_keys_pytest.py +++ b/test_extract_otp_secret_keys_pytest.py @@ -20,8 +20,8 @@ import io import os -import sys import pathlib +import sys import pytest from pytest_mock import MockerFixture diff --git a/test_extract_otp_secret_keys_unittest.py b/test_extract_otp_secret_keys_unittest.py index 25cfd5a..6a2d836 100644 --- a/test_extract_otp_secret_keys_unittest.py +++ b/test_extract_otp_secret_keys_unittest.py @@ -18,14 +18,15 @@ # You should have received a copy of the GNU General Public License # along with this program. If not, see . -import unittest import io -from contextlib import redirect_stdout -from utils import read_csv, read_json, remove_file, remove_dir_with_files, Capturing, read_file_to_str import os import sys +import unittest +from contextlib import redirect_stdout import extract_otp_secret_keys +from utils import (Capturing, read_csv, read_file_to_str, read_json, + remove_dir_with_files, remove_file) class TestExtract(unittest.TestCase): diff --git a/test_extract_qrcode_unittest.py b/test_extract_qrcode_unittest.py index aacc976..d7298ee 100644 --- a/test_extract_qrcode_unittest.py +++ b/test_extract_qrcode_unittest.py @@ -19,9 +19,9 @@ # along with this program. If not, see . import unittest -from utils import Capturing import extract_otp_secret_keys +from utils import Capturing class TestQRImageExtract(unittest.TestCase):