python docx自动插入图片

        自动检索当前文件夹下的图片（jpg,jpeg,png可自行添加）自动创建一个docx文件并往docx里插入图片，效果如下：

        运行后：

         点开文件夹：

核心代码： 

1、获得当前文件夹位置

current_dir = os.path.dirname(os.path.abspath(__file__))

2、循环并对比后缀

for filename in os.listdir(current_dir):
            #对比文件名字后缀
            if filename.endswith('jpg') or filename.endswith('jpeg') or filename.endswith('png'):
                print('--------------------')
                print(current_dir+'\\'+filename)
                print('已添加')
                document.add_picture(current_dir+r'\\'+filename)
                #document.add_picture(address + '1.jpeg')
                #paragraph = document.add_paragraph('Lorem ipsum dolor sit amet.')

完整代码：

import re
import os
import sys
import docx
from docx import Document
from docx.shared import Inches



class findPicture(object):
    def __init__(self):

        #获得当前文件夹位置
        current_dir = os.path.dirname(os.path.abspath(__file__))
        print(current_dir)
        #创建doc
        
        document = Document()
        
        #document.add_heading('图片', 0)
        
        for filename in os.listdir(current_dir):
            #对比文件名字后缀
            if filename.endswith('jpg') or filename.endswith('jpeg') or filename.endswith('png'):
                print('--------------------')
                print(current_dir+'\\'+filename)
                print('已添加')
                document.add_picture(current_dir+r'\\'+filename)
                #document.add_picture(address + '1.jpeg')
                #paragraph = document.add_paragraph('Lorem ipsum dolor sit amet.')
        document.save('ceshi.docx')


if __name__ == '__main__':
    
    fp = findPicture()

        该代码仅为个人学习所作，非专业教程，仅供参考。