-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtxt-rename.py
125 lines (109 loc) · 4.45 KB
/
txt-rename.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
# -*- encoding: utf-8 -*-
"""
@File : test.py
@Time : 2021/1/11 19:12
@Author : SJY
@Email : [email protected]
@Software: PyCharm
@function: 根据图片对应的txt文档，获取图片学号信息，进行重命名，参数为最外层文件夹
"""
import re
import os
# import pytesseract
# from PIL import Image
import tkinter as tk
from tkinter.filedialog import askdirectory
# from tqdm import tqdm
# pytesseract.pytesseract.tesseract_cmd = 'C:\\Program Files\\Tesseract-OCR'
class BatchRename:
def __init__(self):
pass
def rename(self, dirpath_, flag): # 表示需要命名处理的文件夹
print(dirpath_)
filelist = os.listdir(dirpath_) # 获取文件列表
for item in filelist:
imgpath = os.path.join(os.path.abspath(dirpath_), item)
if os.path.isdir(imgpath):
self.rename(imgpath, True)
continue
if item.endswith('.jpg'):
continue
# if flag:
# pbar.update(1)
# print(imgpath)
try:
txtf = open(imgpath,encoding='gbk')
except:
txtf = open(imgpath,encoding='utf-8')
# print(type(txtf))
lines = txtf.read(40)
txtf.close()
# print(lines)
# flines = len(lines)
#(?<=学号: )
reg = re.compile(r"\d+\.*?\d")
match = reg.search(lines)
if not match:
os.remove(imgpath)
# print("未识别出来学号，跳过")
continue
# print("识别出学号{}".format(match.group(0)))
#reg0 = re.compile(r"学籍卡")
reg0 = re.compile(r"成绩表")
match0 = reg0.search(lines)
if match0:
# print("其他: ")
os.remove(imgpath)
# print("删除txt")
portion = os.path.splitext(imgpath) # 分离文件名字和后缀
# print(portion)
newname = portion[0] + ".jpg" # 要改的新后缀
os.remove(newname)
# print("删除jpg")
continue
# print(match0.group(0))
print('学号 {0}'.format(match0.group(0)))
os.remove(imgpath)
# print("删除学籍卡txt")
dst = os.path.join(os.path.abspath(dirpath_),
'' + match.group(0) + '.jpg') # 处理后的格式也为jpg格式的，当然这里可以改成png格式
portion = os.path.splitext(imgpath) # 分离文件名字和后缀
# print(portion)
newname = portion[0] + ".jpg" # 要改的新后缀
try:
os.rename(newname, dst)
print('改名')
except:
# print("跳过 %s" % src)
continue
# os.remove(imgpath)
# image = im.crop((self.left, self.top, im.size[0]/3, im.size[1]/3)) # 对截图进行裁剪
# image.show()
# code_str = pytesseract.image_to_string(image, lang="eng",
# config="--psm 6") # lang的值可以根据安装的语言选择，也不是都可以用，一般看的是验证码类型就可以了。
# code_str = "".join(code) # 字符串
# print(code_str)
# code_str = re.findall(r"\d+\.*?\d", code_str) # 正则 遇到开始和结束就进行截取
# code_str = "".join(code_str)
# print("识别出", code_str)
# src = imgpath
# if code_str.strip() == '':
# os.remove(src)
# # print("删除 %s" % src)
# continue
# dst = os.path.join(os.path.abspath(dirpath_),
# '' + match.group(0) + '.jpg') # 处理后的格式也为jpg格式的，当然这里可以改成png格式
# try:
# os.rename(src, dst)
# print('converting %s to %s!' % (src, dst))
# except:
# # print("跳过 %s" % src)
# continue
# if flag:
# pbar.close()
if __name__ == '__main__':
root = tk.Tk()
root.withdraw()
dirpath = askdirectory()
demo = BatchRename()
demo.rename(dirpath, False)