用发票金额+发票日期+销售方名称+发票号码重命名发票文件增强版
之前写了一个用发票金额重命名pdf发票文件https://www.52pojie.cn/thread-1952842-1-1.html
有些朋友留言要把发票日期,销售方名称,发票号码也加进去,所以改了一下
用发票金额重命名pdf发票文件增强版
用 发票金额+发票日期+销售方名称+发票号码 重命名发票文件,在发票文件的目录下会新建一个‘重命名文件’的文件夹,存放修改后的文件。
代码依然简陋不规范,但好在能运行
步骤:运行程序,选择PDF文档,选择发票。
代码如下:
import pdfplumber
import re
import tkinter as tk
from tkinter import filedialog
import threading
import shutil
import os
import configparser
# 新线程的工作函数
def worker(arg):
# 创建ConfigParser对象
config = configparser.ConfigParser()
# 读取INI文件
config.read('config.ini')
xuhao = config.get('RunCount1', 'count1')
oldfile = config.get('RunCount2', 'count2')
oldfile_data = os.path.dirname(oldfile)
money = config.get('RunCount2', '金额')
fapiaoriqi = config.get('RunCount2', '发票日期')
xiaomingcheng_string = config.get('RunCount2', '销名称')
fapiaohaoma = config.get('RunCount2', '发票号码')
newfile = oldfile_data + '/重命名文件/' + money + '_' + fapiaohaoma + '_' + fapiaoriqi + '_' + xiaomingcheng_string + '(' + xuhao + ').pdf'
# 检查重命名文件文件夹是否存在
if not os.path.exists(os.path.join(oldfile_data, '重命名文件')):
# 如果不存在,就创建重命名文件文件夹
os.makedirs(os.path.join(oldfile_data, '重命名文件'))
print('重命名文件夹创建成功')
shutil.move(oldfile, newfile) # move是移动,如果只想复制可以更换为copy
def main():
filetypes = (
('pdf文件', '*.pdf'),
('所有文件', '*.*')
)
filename = filedialog.askopenfilename(
title='选择要打开的文件',
initialdir='/',
filetypes=filetypes)
if filename:
# 读取PDF文档
with pdfplumber.open(filename) as pdf:
# 获取文档的总页数
total_pages = len(pdf.pages)
# 遍历每一页
for page_number in range(total_pages):
# 获取当前页
page = pdf.pages
# 提取文本内容
text = page.extract_text()
pattern_jine = r'(小写)¥\d+\.\d+'
# 使用re.search 方法提取匹配的内容
match = re.search(pattern_jine, text)
money = match.group()
text_money = money
pattern_jine = r"(\d+(\.\d+)?)"
result = re.search(pattern_jine, text_money)
extracted_amount = result.group()
money = extracted_amount
pattern_fapiaohaoma = r'\b\d{20}\b'
matches_fapiaohaoma = re.findall(pattern_fapiaohaoma, text)
result_fapiaohaoma = ''.join(matches_fapiaohaoma)
pattern_fapiaoriqi = r'(\d{4})年(\d{2})月(\d{2})日'
matches_fapiaoriqi = re.search(pattern_fapiaoriqi,text)
fapiaoriqi = matches_fapiaoriqi.group()
pattern_xiaomingcheng = r'销 名称:(.+)'
match_xiaomingcheng = re.search(pattern_xiaomingcheng, text)
xiaomingcheng = match_xiaomingcheng.group()
xiaomingcheng_string = re.sub(r'\s+', '', xiaomingcheng)
# 检查是否找到匹配项并打印结果
if match:
# 要写入的新内容
new_lines =
# 创建配置文件对象
config = configparser.ConfigParser()
# 设置运行次数的节和键
run_count_section = 'RunCount1'
run_count_key = 'count1'
run_count_section_1 = 'RunCount2'
run_count_key_1 = 'count2'
# 检查配置文件是否存在,如果不存在则创建
config_file = 'config.ini'
if not config.read(config_file, encoding='ANSI'):
config = {run_count_key: '0'}
with open(config_file, 'w', encoding='ANSI') as configfile:
config.write(configfile)
config = {run_count_key_1: filename,
'金额': money,
'发票号码': result_fapiaohaoma,
'发票日期': fapiaoriqi,
'销名称': xiaomingcheng_string
}
# 读取运行次数
current_count = config.getint(run_count_section, run_count_key)
# 递增运行次数
new_count = current_count + 1
config.set(run_count_section, run_count_key, str(new_count))
# 保存配置文件
with open(config_file, 'w', encoding='ANSI') as configfile:
config.write(configfile)
print(f"重命名的第: {new_count}张发票", '金额:', money)
section1_values_1 = config['RunCount1']
section1_values_2 = config['RunCount2']
list_data = ['发票信息',
section1_values_2['count2'],
os.path.dirname(section1_values_2['count2']),
'/',
section1_values_2['金额'],
'(',
section1_values_1['count1'],
').pdf']
# 主线程中创建新线程,并传递参数
t = threading.Thread(target=worker, args=(list_data,))
t.start()
else:
print("未找到匹配项")
# 程序入口
if __name__ == "__main__":
root = tk.Tk()
root.title("文件上传")
root.geometry('300x150')
upload_button = tk.Button(root, text='选择PDF文档', command=main)
upload_button.pack(expand=True)
root.mainloop() Yathon 发表于 2024-8-28 11:42
很实用,赞一个!
希望加上按照路径遍历重命名,如果能识别是否是发票文件就更好了。
满足你,https://www.52pojie.cn/thread-1959322-1-1.html 很实用,赞一个!
希望加上按照路径遍历重命名,如果能识别是否是发票文件就更好了。 很实用 感谢分享 楼主大大,能出一个成品嘛,感谢感谢!!!!! 好东西,西安收藏了
报错啦,是不支持这种发票吗?能修复一下吗? 很实用的工具,希望能发一个成品 菜鸟期待成品 Yathon 发表于 2024-8-28 11:42
很实用,赞一个!
希望加上按照路径遍历重命名,如果能识别是否是发票文件就更好了。
好建议,希望楼主更新一下 这个确实很实用,望楼主能分享成品。另外,请问有能识别PDF某些内容自动重命名的工具吗?