python docx 批量修改docx文件内容和选择位置添加内容

#-*- encoding:utf_8 -*-
from docx import Document
from docx.shared import Inches
from log import logger
import os

'''
python docx 批量修改docx文件内容和选择位置添加内容的初步尝试
不喜勿喷♪(^∇^*)
'''

def folder_filelist(pathname):
    '''
    读取所有文件名，为了实现对文件夹的批处理
    '''
    filename_list = list()
    full_pathname = os.path.abspath(pathname)
    if os.path.isfile(full_pathname):
        if _is_legal_postfix(full_pathname):
            filename_list.append(full_pathname)
        else:
            raise TypeError('文件 {} 后缀名不合法！仅支持如下文件类型：{}。'.format(pathname, '、'.join(self._handle_postfix)))
    elif os.path.isdir(full_pathname):
        for relpath, _, files in os.walk(full_pathname):
            for name in files:
                filename = os.path.join(full_pathname, relpath, name)
                if _is_legal_postfix(filename):
                    filename_list.append(os.path.join(filename))
        return filename_list
    else:
        raise TypeError('文件/文件夹 {} 不存在或不合法！'.format(pathname))

def _is_legal_postfix(filename):
    _handle_postfix = ['doc', 'docx', 'ppt', 'pptx', 'xls', 'xlsx', 'pdf']
    return filename.split('.')[-1].lower() in _handle_postfix and not os.path.basename(filename).startswith(
        '~')


def insert_dash(string, index, insert_content):
    '''
    str中插入字符
    '''
    return string[:index] + insert_content + string[index:]

def find_replace(docxname, findname, replacename):
    '''
    docx中替换内容
    '''
    document = Document(docxname)

    for paragraph in document.paragraphs:
        line = paragraph.text
        logger.info(line)
        paragraph.clear()
        if line.find(findname) != -1:
            line = insert_dash(line, line.find(findname), replacename)
            line = line.replace(findname,"")
        paragraph.add_run(line)
        logger.info(line)

    document.save("1" + docxname)

def find_add_paragraph(docxname, findname, insert_paragraph):
    '''
    docx中想要的位置，插入段落
    '''
    document = Document(docxname)

    for paragraph in document.paragraphs:
        line = paragraph.text
        logger.info(line)
        if line.find(findname) != -1:
            for content in insert_paragraph:
                paragraph.add_run('\n'+content)
                logger.info(content)

    document.save("1" + docxname)

if __name__ == '__main__':

    for filename in folder_filelist('pdfconver'):
        logger.info(filename)
    find_replace('1.docx', '666', '6666')

    find_add_paragraph('1.docx', '6', ['你好', 'spa'])