Python提高运维效率的脚本分享

python sed.py c:\name.txt Kanye Edison 

python sed.py c:\name.txt Kanye,male  Edison,female 

import sys
import os
import time
from shutil import copy

def data_preprocess():
    file_name, old_str, new_str = sys.argv[1:4] #接收输入的参数：目标文件名，旧内容、新内容、字符编码
    try: # 传入字符编码参数
        encoding = sys.argv[4] 
    except IndexError: # 没有传入字符编码参数的话就默认使用 'utf-8' 字符编码
        encoding = 'utf-8' 

    old_str_list = [i.encode(encoding) for i in old_str.split(',')] # 将旧内容转换成列表形式
    new_str_list = [i.encode(encoding) for i in new_str.split(',')] # 将新内容转换成列表形式

    assert len(old_str_list) == len(new_str_list) # 判断用户输入的 old_str 和 new_str 是不是一一对应
    trans_tabs = list(zip(old_str_list, new_str_list)) # 将要旧内容列表(old_str_list)和新内容列表( new_str_list)中的元素一一对应

    return file_name, trans_tabs # 返回目标文件名，以及一个旧内容元素和新内容元素一一对应的列表

python sed.py c:\test.txt Edison Kanye

file_name = c:\test.txt
trans_tabs = [(Edsion, Kanye)]

python sed.py c:\test.txt edsion,kanye,fish Edsion,Kanye,Fish utf-8

file_name = c:\test.txt
trans_tabs = [(edsion, Edsion), (kanye,Kanye), (fish,Fish)]

def backup(file_name):
    time_mark = time.strftime('%Y%m%d_%H%M%S') #时间戳
    bak_dir = r'C:\Users\Administrator\Desktop' #备份路径
    basename = os.path.basename(file_name)
    os.makedirs(bak_dir) if not os.path.isdir(bak_dir) else True
    copy(file_name, os.path.join(bak_dir, basename + '_' + time_mark))
    print("备份 %s 成功" %file_name)

def sed(file_name, trans_tabs):
    with open(file_name + '.swap', 'wb') as swap_fs, open(file_name, 'rb') as file_names: #打开一个临时文件和目标文件
        for line in file_names: # 逐行读取
            for tab in trans_tabs:
                line = line.replace(tab[0], tab[1]) if tab[0] in line else line # 修改替换操作
            swap_fs.write(line) # 将修改后的内容写入到临时文件当中
    os.remove(file_name) # 删除旧目标文件
    os.rename(file_name + '.swap', file_name) # 将临时文件重命名，就变成了新的目标文件

import sys
import os
import time
from shutil import copy


def data_preprocess():
    file_name, old_str, new_str = sys.argv[1:4]
    try:
        encoding = sys.argv[4]
    except IndexError:
        encoding = 'utf-8'

    old_str_list = [i.encode(encoding) for i in old_str.split(',')]
    new_str_list = [i.encode(encoding) for i in new_str.split(',')]

    assert len(old_str_list) == len(new_str_list)
    trans_tabs = list(zip(old_str_list, new_str_list))

    return file_name, trans_tabs


def backup(file_name):
    time_mark = time.strftime('%Y%m%d_%H%M%S')
    bak_dir = r'C:\Users\Administrator\Desktop'
    basename = os.path.basename(file_name)
    os.makedirs(bak_dir) if not os.path.isdir(bak_dir) else True
    copy(file_name, os.path.join(bak_dir, basename + '_' + time_mark))
    print("备份 %s" %file_name)


def sed(file_name, trans_tabs):
    with open(file_name + '.swap', 'wb') as swap_fs, open(file_name, 'rb') as file_names:
        for line in file_names:
            for tab in trans_tabs:
                line = line.replace(tab[0], tab[1]) if tab[0] in line else line
            swap_fs.write(line)
    os.remove(file_name)
    os.rename(file_name + '.swap', file_name)


file_name, trans_tabs = data_preprocess()
backup(file_name)
sed(file_name, trans_tabs)