Kabooshki
/
Textbook-Arrange
1
0
Fork 0
Code Issues Pull Requests Packages Projects 1 Releases 1 Wiki Activity

added dash to filename for compatability

This commit is contained in:
Kabooshki 2024-09-18 12:34:02 -05:00
parent fe186d6c12
commit d2913ac173
Signed by: Kabooshki
GPG Key ID: 5E7B7F2F405CD62A
2 changed files with 18 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
fixer.py
View File

@ -22,9 +22,9 @@ for files in os.listdir(directory):
cv2.imshow('Review', ROI) cv2.imshow('Review', ROI)
cv2.waitKey(1) cv2.waitKey(1)
manpage = input("please input the page number") 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() rng = numpy.random.default_rng().random()
os.rename (f'images/{manpage}', f'images/{manpage}.review-{rng}') os.rename (f'images/{manpage}-', f'images/{manpage}-.review-{rng}')
os.rename(f'{f}', f'images/{manpage}') os.rename(f'{f}', f'images/{manpage}-')
elif os.path.isfile(f'images/{manpage}') == False: elif os.path.isfile(f'images/{manpage}-') == False:
os.rename(f'{f}', f'images/{manpage}') os.rename(f'{f}', f'images/{manpage}-')

23
main.py
View File

@ -18,23 +18,26 @@ for filename in os.listdir(directory):
ROI = cv2.hconcat([ROI1, ROI2]) ROI = cv2.hconcat([ROI1, ROI2])
data = pytesseract.image_to_string(ROI, lang='eng',config='--psm 7 -c tessedit_char_whitelist=0123456789i' ) data = pytesseract.image_to_string(ROI, lang='eng',config='--psm 7 -c tessedit_char_whitelist=0123456789i' )
print(data) print(data)
if os.path.isfile(f'images/{data}') == False: if os.path.isfile(f'images/{data}-') == False:
os.rename(f'{f}', f'images/{data}') if data == "":
elif os.path.isfile(f'images/{data}') == True: print("failure")
elif data != "":
os.rename(f'{f}', f'images/{data}-')
elif os.path.isfile(f'images/{data}-') == True:
cv2.imshow(data, ROI) cv2.imshow(data, ROI)
cv2.waitKey(1) cv2.waitKey(1)
manpage = input("please input the number on the page (if the title is correct, enter nothing)") manpage = input("please input the number on the page (if the title is correct, enter nothing)")
#check if the new name already exists #check if the new name already exists
if manpage == "\n": if manpage == "\n":
rng = numpy.random.default_rng().random() rng = numpy.random.default_rng().random()
os.rename (f'images/{data}', f'images/{data}.review-{rng}') os.rename (f'images/{data}-', f'images/{data}-.review-{rng}')
os.rename(f'{f}', f'images/{data}') os.rename(f'{f}', f'images/{data}-')
elif manpage != "\n": 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() rng = numpy.random.default_rng().random()
os.rename (f'images/{manpage}', f'images/{manpage}.review-{rng}') os.rename (f'images/{manpage}-', f'images/{manpage}-.review-{rng}')
os.rename(f'{f}', f'images/{manpage}') os.rename(f'{f}', f'images/{manpage}-')
elif os.path.isfile(f'images/{manpage}') == False: elif os.path.isfile(f'images/{manpage}-') == False:
os.rename(f'{f}', f'images/{manpage}') os.rename(f'{f}', f'images/{manpage}-')
cv2.destroyAllWindows() cv2.destroyAllWindows()
cv2.destroyAllWindows() cv2.destroyAllWindows()