52 lines
2.1 KiB
Python
52 lines
2.1 KiB
Python
# import os
|
|
|
|
# def add_lines_to_files(directory, lines_to_add):
|
|
# for filename in os.listdir(directory):
|
|
# filepath = os.path.join(directory, filename)
|
|
# if os.path.isfile(filepath):
|
|
# with open(filepath, 'r+', encoding='utf-8') as file:
|
|
# content = file.read()
|
|
# file.seek(0, 0)
|
|
# file.write('\n'.join(lines_to_add) + '\n' + content)
|
|
|
|
# # 示例用法
|
|
# directory = r'C:\Users\anenvoy\Desktop\notes'
|
|
# lines_to_add = ['---', 'title:','date:','tags:','categories:','---']
|
|
|
|
# add_lines_to_files(directory, lines_to_add)
|
|
|
|
|
|
import os
|
|
|
|
def insert_lines_in_files(directory, lines_to_add):
|
|
for filename in os.listdir(directory):
|
|
filepath = os.path.join(directory, filename)
|
|
if os.path.isfile(filepath):
|
|
with open(filepath, 'r+', encoding='utf-8') as file:
|
|
content = file.readlines()
|
|
first_line = content[0].strip() if content else '' # 获取原文的第一行，并去掉空白字符
|
|
|
|
# 去掉文件名的后缀名
|
|
file_name_without_ext = os.path.splitext(filename)[0]
|
|
|
|
# 生成新的内容
|
|
if len(lines_to_add) >= 2:
|
|
# new_lines = lines_to_add[:1] + [filename] + [f"{lines_to_add[2]} {first_line}"] + lines_to_add[3:]
|
|
new_lines = lines_to_add[:1] + [f"{lines_to_add[1]} {file_name_without_ext}"] + lines_to_add[2:]
|
|
else:
|
|
new_lines = [filename, f"{first_line}"] # 如果 lines_to_add 为空或长度不足，处理情况
|
|
|
|
# 将文件内容与新行结合
|
|
new_content = '\n'.join(new_lines) + '\n' + ''.join(content)
|
|
|
|
# 回到文件开头，写入新内容
|
|
file.seek(0)
|
|
file.write(new_content)
|
|
# 截断文件以删除多余的内容
|
|
file.truncate()
|
|
|
|
# 示例用法
|
|
directory = r'C:\Users\anenvoy\Desktop\notes' # 修改为你的实际文件夹路径
|
|
lines_to_add = ['---', 'title:','date:','tags:','categories:','---']
|
|
insert_lines_in_files(directory, lines_to_add)
|