Mirrors/extract_otp_secrets
1
0
Fork 0
mirror of https://github.com/scito/extract_otp_secrets.git synced 2025-12-18 08:12:26 +01:00
Code Issues Packages Projects Releases 1 Wiki Activity

refactor code: extract method

Browse Source
This commit is contained in:
scito
2022-12-24 02:56:40 +01:00
parent af0d7ffd5d
commit 82da427d1a
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
extract_otp_secret_keys.py
View File

@@ -139,8 +139,14 @@ def extract_otps(args):
def get_lines_from_file(filename): def get_lines_from_file(filename):
global verbose lines = read_lines_from_text_file(filename)
if lines != None:
return lines
# could not process text file, try reading as image
return convert_img_to_line(filename)
def read_lines_from_text_file(filename):
if filename != '=': if filename != '=':
check_file_exists(filename) check_file_exists(filename)
finput = fileinput.input(filename) finput = fileinput.input(filename)
@@ -158,11 +164,12 @@ def get_lines_from_file(filename):
if filename == '-': if filename == '-':
abort('\nERROR: Unable to open text file form stdin. ' abort('\nERROR: Unable to open text file form stdin. '
'In case you want read an image file from stdin, you must use "=" instead of "-".') 'In case you want read an image file from stdin, you must use "=" instead of "-".')
# else: The file is probably an image, process below else: # The file is probably an image, process below
return None
finally: finally:
finput.close() finput.close()
# could not process text file, try reading as image def convert_img_to_line(filename):
if filename != '-': if filename != '-':
try: try:
if filename != '=': if filename != '=':