使用python在文本文件中添加特定内容

昨天折腾博客折腾了一整天，什么事情都没干，好气啊，于是又水了一篇blog。
主要就是加入了版权信息，具体效果可以看结尾。操作可参考此教程。这教程写的很不错，欢迎大家一起折腾。
其实还有些改动，比如想加入导航页，最后也没成功，哎，不多说了。开始吧。

需要在页面尾部显示版权信息，需要在开头加入copyright: true，效果如下：

---
title: 使用python在文本文件中添加特定内容
copyright: true
date: 2018-07-13 09:08:39
tags:
---

于是想用python写个脚本，批量在md文件开头加入copyright，代码也不难，直接贴了。

#!/usr/bin/python
import os
def file_name(file_dir):
    # 获取所有md文件名
    filelist = []
    for root, dirs, files in os.walk(file_dir):  
        for fileName in files:
            if(fileName[-3:]=='.md'):
                filelist.append(fileName) #当前路径下所有非目录子文件
    return filelist

path = './'
file_list = file_name(path)
print(file_list)
content_add='copyright: true \n' # 需要加入的信息
for i in file_list:
    file = open(i,'r',encoding='utf-8')
    content = file.read() # 读取文件
    if(content.find('copyright: true')==-1):
    # 如果没有copyright信息，则找到tags位置，并将copyright信息插入
        pos=content.find('tags:')
        content = content[:pos] + content_add + content[pos:]
    file.close()
    file = open(i,'w',encoding='utf-8') # 保存文件
    file.write( content )

昨天瞎改完，发现文章的图片已经不能正常显示了，然后折腾了一下也没解决，以后有机会再找个图床吧。

又开心地水了篇博客。