最近有这样一个需求,文件夹中有800个学生的照片,以准考证命名,现在想根据excel文件,把文件名改为以身份证命名。
具体代码如下:
import pandas as pd
import os
import re
import argparse
# 获取目录下的excel文件
def getExcel(dataPath, excelPath):
data = {}
df = pd.read_excel(excelPath, sheet_name='学生具体信息')
df = pd.DataFrame(df, columns=['考生号', '证件号码'])
for i in df.values:
data[i[0]] = i[1]
modifyName(data, dataPath)
# 修改文件名
def modifyName(data, dataPath):
try:
for maindir, subdir, file_name_list in os.walk(dataPath):
for filename in file_name_list:
try:
apath = os.path.join(maindir, filename)
file_path = apath.replace("\\", "/")
student_name = filename.strip(".jpg")
p = re.compile(r"(.*)/")
filepath = p.match(file_path).group(1)
new_file_name = filepath + "/" + \
data[int(student_name)] + ".jpg"
print("[+] 修改成功", file_path, new_file_name)
os.rename(file_path, new_file_name)
except Exception as e:
# print("error:", e)
# print("[-] 请检查:", student_name,
# "是否存在对应身份证号码。", "文件路径:", file_path)
continue
except Exception as e:
print("error:", e)
data = 'D:\PycharmProjects\爬取集思录数据\菏泽大专新生照片'
excel = 'D:\PycharmProjects\爬取集思录数据\菏泽校区802人.xlsx'
getExcel(data, excel)