使用Python脚本替换模板文件中的版本号

需求

后端模板文件引用前端资源，发布时，使用Python脚本自动替换前端资源的版本。

准备工作

前端使用Webpack构建，每次构建完成时，会生成一个buildfile.json

buildfile.json

{
    "about.css": "about-ca3f85cec35d3ab39ac0.css",
    "about.js": "about-613281148cb8885f2b3d.js",
    "home.css": "home-94093b44a5a85be01f38.css",
    "home.js": "home-6dd4e116818ee4945f65.js"
}

读取前端构建文件

import os
import json

BUILD_FILES = {}

def get_build_files():
    build_file_path = '/Users/pengjie/try/iseo2/dist/buildfile.json'
    if(os.path.exists(build_file_path)):
        with open(build_file_path,'r') as f:
            return json.load(f)
    else:
        print('build file not exists.')
        return ''

BUILD_FILES = get_build_files();

获取所有模板文件

def get_template_files():
    template_path = '/Users/pengjie/try/iseo2/tmpviews'
    template_files = []
    for dirpath, dirnames, filenames in os.walk(template_path):
        for filename in filenames:
            if(filename[0] != '.'):
                filepath = os.path.join(dirpath,filename)
                template_files.append(filepath)
    return template_files

获取所有要替换的script和link

import os
import re
from bs4 import BeautifulSoup

def get_replace_list(content):
    replace_list = []
    soup = BeautifulSoup(content,'html.parser')

    # 获取所有link
    links = soup.find_all('link')
    for item in links:
        name = os.path.basename(item.get('href'))
        matches = re.search(r'\-(.+)?(\.)',name,re.I)
        key = name.replace(matches[0],'.')
        replace_list.append({ 'old': name, 'new': BUILD_FILES.get(key,'')})

    # 获取所有script
    scripts = soup.find_all('script')
    for item in scripts:
        name = os.path.basename(item.get('src'))
        matches = re.search(r'\-(.+)?(\.)',name,re.I)
        key = name.replace(matches[0],'.')
        replace_list.append({ 'old': name, 'new': BUILD_FILES.get(key,'')})

    return replace_list

查找并替换

def search_and_replace(tplpath):
    content = ''
    with open(tplpath,'r',encoding="utf-8") as f:
        content = f.read()
        replace_list = get_replace_list(content)
        for item in replace_list:
            if(item['old'] and item['new']):
                content = content.replace(item['old'],item['new'])

    with open(tplpath,"w",encoding="utf-8") as f:
        f.write(content)

完整脚本内容

import os
import re
import json
from bs4 import BeautifulSoup

BUILD_FILES = {}

def run():
    view_path = '/Users/pengjie/try/iseo2/tmpviews'
    for dirpath, dirnames, filenames in os.walk(view_path):
        for filename in filenames:
            if(filename[0] != '.'):
                filepath = os.path.join(dirpath,filename)

                # 读取文件，获取替换列表
                content = ''
                with open(filepath,'r',encoding="utf-8") as f:
                    content = f.read()
                    replace_list = get_replace_list(content)
                    for item in replace_list:
                        if(item['old'] and item['new']):
                            content = content.replace(item['old'],item['new'])

                with open(filepath,"w",encoding="utf-8") as f:
                    f.write(content)
                print('----')

# 获取替换列表
def get_replace_list(content):
    replace_list = []
    soup = BeautifulSoup(content,'html.parser')

    # 获取所有link
    links = soup.find_all('link')
    for item in links:
        name = os.path.basename(item.get('href'))
        matches = re.search(r'\-(.+)?(\.)',name,re.I)
        key = name.replace(matches[0],'.')
        replace_list.append({ 'old': name, 'new': BUILD_FILES.get(key,'')})

    # 获取所有script
    scripts = soup.find_all('script')
    for item in scripts:
        name = os.path.basename(item.get('src'))
        matches = re.search(r'\-(.+)?(\.)',name,re.I)
        key = name.replace(matches[0],'.')
        replace_list.append({ 'old': name, 'new': BUILD_FILES.get(key,'')})

    return replace_list
                    
# 获取构建文件
def get_build_file():
    build_file_path = '/Users/pengjie/try/iseo2/dist/buildfile.json'
    if(os.path.exists(build_file_path)):
        with open(build_file_path,'r') as f:
            return json.load(f)
    else:
        print('build file not exists.')
        return ''

if __name__ == '__main__':
    BUILD_FILES = get_build_file()
    if(BUILD_FILES):
        run()