Python提高運(yùn)維效率的腳本分享

更新時(shí)間：2023年05月10日 08:27:43 作者：咸魚Linux運(yùn)維

這篇文章主要來和大家分享一個(gè)Python提高運(yùn)維效率的腳本，文中的示例代碼講解詳細(xì)，具有一定的學(xué)習(xí)價(jià)值，感興趣的小伙伴可以參考下

哈嘍大家好我是咸魚，今天給大家分享一個(gè)能夠提升運(yùn)維效率的 python 腳本

咸魚平常在工作當(dāng)中通常會(huì)接觸到下面類似的場景：

容災(zāi)切換的時(shí)候批量對機(jī)器上的配置文件內(nèi)容進(jìn)行修改替換
對機(jī)器批量替換某個(gè)文件中的字段

對于 Linux 機(jī)器，咸魚可以寫個(gè) shell 腳本或者直接批量使用 sed 命令就能很好的解決

但對于 Windows 機(jī)器，上面的方法就不管用了，我們就需要想其他的辦法

這里咸魚給大家分享一個(gè)由 python 編寫的腳本，這個(gè)腳本能夠去替換指定文本文件中的指定內(nèi)容，而且還能一次替換多個(gè)內(nèi)容

我們先看效果，目標(biāo)文件：name.txt

1.修改一個(gè)內(nèi)容

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

2.修改多個(gè)內(nèi)容

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

源碼在文末哦

首先我們導(dǎo)入模塊

import sys
import os
import time
from shutil import copy

然后我們來看一下腳本中的第一個(gè)函數(shù)——data_preprocess()

def data_preprocess():
    file_name, old_str, new_str = sys.argv[1:4] #接收輸入的參數(shù)：目標(biāo)文件名，舊內(nèi)容、新內(nèi)容、字符編碼
    try: # 傳入字符編碼參數(shù)
        encoding = sys.argv[4] 
    except IndexError: # 沒有傳入字符編碼參數(shù)的話就默認(rèn)使用 'utf-8' 字符編碼
        encoding = 'utf-8' 

    old_str_list = [i.encode(encoding) for i in old_str.split(',')] # 將舊內(nèi)容轉(zhuǎn)換成列表形式
    new_str_list = [i.encode(encoding) for i in new_str.split(',')] # 將新內(nèi)容轉(zhuǎn)換成列表形式

    assert len(old_str_list) == len(new_str_list) # 判斷用戶輸入的 old_str 和 new_str 是不是一一對應(yīng)
    trans_tabs = list(zip(old_str_list, new_str_list)) # 將要舊內(nèi)容列表(old_str_list)和新內(nèi)容列表( new_str_list)中的元素一一對應(yīng)

    return file_name, trans_tabs # 返回目標(biāo)文件名，以及一個(gè)舊內(nèi)容元素和新內(nèi)容元素一一對應(yīng)的列表

這個(gè)函數(shù)實(shí)現(xiàn)的功能是接收輸入的參數(shù)（目標(biāo)文件名、要替換的內(nèi)容，替換的內(nèi)容、字符編碼格式），然后將要替換的內(nèi)容與替換的內(nèi)容分別轉(zhuǎn)換成列表形式

注意：替換多個(gè)內(nèi)容時(shí)在多個(gè)內(nèi)容之間用逗號隔開

例如我們敲如下命令替換一個(gè)內(nèi)容：

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

這個(gè)函數(shù)就會(huì)返回下面內(nèi)容（由于命令里沒有傳入字符編碼參數(shù)，采取默認(rèn)的 UTF-8）

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

如果我們需要替換多個(gè)內(nèi)容（例如將首字母改成大寫）

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

這個(gè)函數(shù)就會(huì)返回下面內(nèi)容（命令傳入了字符編碼參數(shù)）

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

接下來我們來看第二個(gè)函數(shù)——backup()

def backup(file_name):
    time_mark = time.strftime('%Y%m%d_%H%M%S') #時(shí)間戳
    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)

這個(gè)函數(shù)的功能就是在修改文件之前先把文件備份，防止后期我們需要回滾復(fù)原

最后我們來看最后一個(gè)函數(shù)—— sed()

這個(gè)函數(shù)便是整個(gè)腳本的核心，它負(fù)責(zé)去執(zhí)行修改替換文件內(nèi)容的操作

def sed(file_name, trans_tabs):
    with open(file_name + '.swap', 'wb') as swap_fs, open(file_name, 'rb') as file_names: #打開一個(gè)臨時(shí)文件和目標(biāo)文件
        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) # 將修改后的內(nèi)容寫入到臨時(shí)文件當(dāng)中
    os.remove(file_name) # 刪除舊目標(biāo)文件
    os.rename(file_name + '.swap', file_name) # 將臨時(shí)文件重命名，就變成了新的目標(biāo)文件

首先先打開一個(gè)臨時(shí)文件（file_name.swap）和目標(biāo)文件（file_name）

然后對目標(biāo)文件（file_name）進(jìn)行逐行讀取到內(nèi)存上，再去對內(nèi)容進(jìn)行修改，最后將修改后的內(nèi)容寫入到這個(gè)臨時(shí)文件（file_name.swap）中

修改完畢之后，把臨時(shí)文件重命名一下、把舊目標(biāo)文件刪掉，這個(gè)臨時(shí)文件就變成了修改內(nèi)容后的目標(biāo)文件了

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)

到此這篇關(guān)于Python提高運(yùn)維效率的腳本分享的文章就介紹到這了,更多相關(guān)Python腳本內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: