From d2913ac173bca81ba9063ba1fd2f8943ab02c47b Mon Sep 17 00:00:00 2001 From: Kabooshki Date: Wed, 18 Sep 2024 12:34:02 -0500 Subject: [PATCH] added dash to filename for compatability --- fixer.py | 10 +++++----- main.py | 23 +++++++++++++---------- 2 files changed, 18 insertions(+), 15 deletions(-) diff --git a/fixer.py b/fixer.py index c968894..3f75c3d 100644 --- a/fixer.py +++ b/fixer.py @@ -22,9 +22,9 @@ for files in os.listdir(directory): cv2.imshow('Review', ROI) cv2.waitKey(1) manpage = input("please input the page number") - if os.path.isfile(f'images/{manpage}') == True: + if os.path.isfile(f'images/{manpage}-') == True: rng = numpy.random.default_rng().random() - os.rename (f'images/{manpage}', f'images/{manpage}.review-{rng}') - os.rename(f'{f}', f'images/{manpage}') - elif os.path.isfile(f'images/{manpage}') == False: - os.rename(f'{f}', f'images/{manpage}') \ No newline at end of file + os.rename (f'images/{manpage}-', f'images/{manpage}-.review-{rng}') + os.rename(f'{f}', f'images/{manpage}-') + elif os.path.isfile(f'images/{manpage}-') == False: + os.rename(f'{f}', f'images/{manpage}-') \ No newline at end of file diff --git a/main.py b/main.py index 6ef7e5b..152c0a0 100644 --- a/main.py +++ b/main.py @@ -18,23 +18,26 @@ for filename in os.listdir(directory): ROI = cv2.hconcat([ROI1, ROI2]) data = pytesseract.image_to_string(ROI, lang='eng',config='--psm 7 -c tessedit_char_whitelist=0123456789i' ) print(data) - if os.path.isfile(f'images/{data}') == False: - os.rename(f'{f}', f'images/{data}') - elif os.path.isfile(f'images/{data}') == True: + if os.path.isfile(f'images/{data}-') == False: + if data == "": + print("failure") + elif data != "": + os.rename(f'{f}', f'images/{data}-') + elif os.path.isfile(f'images/{data}-') == True: cv2.imshow(data, ROI) cv2.waitKey(1) manpage = input("please input the number on the page (if the title is correct, enter nothing)") #check if the new name already exists if manpage == "\n": rng = numpy.random.default_rng().random() - os.rename (f'images/{data}', f'images/{data}.review-{rng}') - os.rename(f'{f}', f'images/{data}') + os.rename (f'images/{data}-', f'images/{data}-.review-{rng}') + os.rename(f'{f}', f'images/{data}-') elif manpage != "\n": - if os.path.isfile(f'images/{manpage}') == True: + if os.path.isfile(f'images/{manpage}-') == True: rng = numpy.random.default_rng().random() - os.rename (f'images/{manpage}', f'images/{manpage}.review-{rng}') - os.rename(f'{f}', f'images/{manpage}') - elif os.path.isfile(f'images/{manpage}') == False: - os.rename(f'{f}', f'images/{manpage}') + os.rename (f'images/{manpage}-', f'images/{manpage}-.review-{rng}') + os.rename(f'{f}', f'images/{manpage}-') + elif os.path.isfile(f'images/{manpage}-') == False: + os.rename(f'{f}', f'images/{manpage}-') cv2.destroyAllWindows() cv2.destroyAllWindows()