Public/extract_otp_secrets
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

add argument for initial setting of qr_mode

Browse Source
This commit is contained in:
scito
2022-12-31 12:43:17 +01:00
parent f731530f57
commit 2dea161cdc
3 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.vscode/launch.json vendored
View File

@@ -32,6 +32,7 @@
"request": "launch", "request": "launch",
"program": "src/extract_otp_secrets.py", "program": "src/extract_otp_secrets.py",
"args": [ "args": [
"--qr", "CV2"
], ],
"console": "integratedTerminal" "console": "integratedTerminal"
}, },

6
src/extract_otp_secrets.py
View File

@@ -113,6 +113,8 @@ Otps = List[Otp]
# PYTHON > 3.9: OtpUrls = list[OtpUrl] # PYTHON > 3.9: OtpUrls = list[OtpUrl]
OtpUrls = List[OtpUrl] OtpUrls = List[OtpUrl]
QRMode = Enum('QRMode', ['QREADER', 'DEEP_QREADER', 'CV2'], start=0)
# Constants # Constants
CAMERA: Final[str] = 'camera' CAMERA: Final[str] = 'camera'
@@ -157,6 +159,7 @@ python extract_otp_secrets.py = < example_export.png"""
b) image file containing a QR code or = for stdin for an image containing a QR code""", nargs='*' if qreader_available else '+') b) image file containing a QR code or = for stdin for an image containing a QR code""", nargs='*' if qreader_available else '+')
if qreader_available: if qreader_available:
arg_parser.add_argument('--camera', '-C', help='camera number of system (default camera: 0)', default=0, nargs=1, metavar=('NUMBER')) arg_parser.add_argument('--camera', '-C', help='camera number of system (default camera: 0)', default=0, nargs=1, metavar=('NUMBER'))
arg_parser.add_argument('--qr', '-Q', help=f'initial QR reader for camera (default: {QRMode.QREADER.name})', type=str, choices=[mode.name for mode in QRMode], default=QRMode.QREADER.name)
arg_parser.add_argument('--json', '-j', help='export json file or - for stdout', metavar=('FILE')) arg_parser.add_argument('--json', '-j', help='export json file or - for stdout', metavar=('FILE'))
arg_parser.add_argument('--csv', '-c', help='export csv file or - for stdout', metavar=('FILE')) arg_parser.add_argument('--csv', '-c', help='export csv file or - for stdout', metavar=('FILE'))
arg_parser.add_argument('--keepass', '-k', help='export totp/hotp csv file(s) for KeePass, - for stdout', metavar=('FILE')) arg_parser.add_argument('--keepass', '-k', help='export totp/hotp csv file(s) for KeePass, - for stdout', metavar=('FILE'))
@@ -198,8 +201,7 @@ def extract_otps_from_camera(args: Args) -> Otps:
otp_urls: OtpUrls = [] otp_urls: OtpUrls = []
otps: Otps = [] otps: Otps = []
QRMode = Enum('QRMode', ['QREADER', 'DEEP_QREADER', 'CV2'], start=0) qr_mode = QRMode[args.qr]
qr_mode = QRMode.QREADER
if verbose: print(f"QR reading mode: {qr_mode}") if verbose: print(f"QR reading mode: {qr_mode}")
cam = cv2.VideoCapture(args.camera) cam = cv2.VideoCapture(args.camera)

2
upgrade_deps.sh
View File

@@ -293,7 +293,7 @@ eval "$cmd"
# https://github.com/marketplace/actions/pytest-coverage-comment # https://github.com/marketplace/actions/pytest-coverage-comment
# Coverage-95%25-yellowgreen # Coverage-95%25-yellowgreen
echo -e "Upgrade code coverage in README.md" echo -e "Update code coverage in README.md"
TOTAL_COVERAGE=$(cat $COVERAGE_OUT | grep 'TOTAL' | perl -ne 'print "$&" if /\b(\d{1,3})%/') && perl -i -pe "s/coverage-(\d{1,3}%)25-/coverage-${TOTAL_COVERAGE}25-/" README.md TOTAL_COVERAGE=$(cat $COVERAGE_OUT | grep 'TOTAL' | perl -ne 'print "$&" if /\b(\d{1,3})%/') && perl -i -pe "s/coverage-(\d{1,3}%)25-/coverage-${TOTAL_COVERAGE}25-/" README.md
if $build_docker; then if $build_docker; then