利用Python如何批量更新服務(wù)器文件

更新時間：2018年07月29日 14:23:09 作者：YxYYxY

這篇文章主要給大家介紹了關(guān)于利用Python如何批量更新服務(wù)器文件的相關(guān)資料，文中通過示例代碼介紹的非常詳細，對大家學習或者使用python具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧

前言

買了個Linux服務(wù)器，Centos系統(tǒng)，裝了個寶塔搭建了10個網(wǎng)站，比如有時候要在某個文件上加點代碼，就要依次去10個文件改動,雖然寶塔是可視化頁面操作，不需要用命令，但是也麻煩，雖然還有g(shù)it的hook方法,但是操作也麻煩，新建個目錄的話還得操作一次，所以萌生了一個想法，用Python來批量更新服務(wù)器上的文件

序言

在網(wǎng)上搜索了一圈，發(fā)現(xiàn)Python有個庫叫paramiko可以專門拿來干這個事，具體資料和安裝就網(wǎng)上去搜索吧，我就直接上代碼了,不到100行,其實還可以精簡吧,后面再說了,先把功能實現(xiàn)了再說,Show Code

代碼

import paramiko
import os

# 連接信息
host = 'xxx.65.9.191'
port = 22
username = 'root'
password = 'root'

# 忽略的目錄
skipArry = ['kai.xxxx.com','demo.xxxx.com']

fullpathArry = []
currentIndex = ''


# 判斷文件是否存在
def judgeFileExist():
 global currentIndex;
 currentIndex = os.getcwd() + '/Index.php'
 if os.path.isfile(currentIndex) == False:
  print('Index文件不存在')
  exit()
 print('文件檢測成功,準備連接服務(wù)器...')



def creatConnect():
 try:
  print('開始連接服務(wù)器...')
  s = paramiko.Transport((host, port))
  s.connect(username=username, password=password)
  sftp = paramiko.SFTPClient.from_transport(s)
  print('連接:' + host + '成功')
  return sftp,s
 except Exception as e:
  print('連接服務(wù)器失敗:' + str(e))



#

# 獲取目錄保存為數(shù)組
def getDirectory(sftp):
 print('開始獲取目錄...')
 sftp.chdir('/www/wwwroot')
 pathlist = sftp.listdir(path='.')
 for path in pathlist:
  fullpath = '/www/wwwroot/' + path + '/application/index/controller'
  if path in skipArry:
   continue
  fullpathArry.append(fullpath)
 print('目錄獲取完畢')

# 上傳Index文件
def uploadIndex(sftp):
 for fullpathitem in fullpathArry:
   remoteIndex = fullpathitem + '/Index.php'
   print('開始上傳:' + remoteIndex)
   try:
    sftp.put(currentIndex, remoteIndex)
    try:
     sftp.file(remoteIndex)
     sftp.chmod(remoteIndex, int("775", 8))
     print('修改' + remoteIndex + '權(quán)限為755')
     print(fullpathitem + '上傳成功')
    except:
     print(fullpathitem + '上傳失敗')
     continue
   except Exception as e:
    print('錯誤信息:' + str(e))
    continue
  

if __name__ == "__main__":
 judgeFileExist()
 sftp,s = creatConnect()
 getDirectory(sftp)
 uploadIndex(sftp)
 s.close()

代碼Show完了,開始詳細解釋一波

這個方法是檢測我當前目錄下有沒有Index.php這個文件,如果沒有的話就直接退出不進行下一步了,這里有個小坑，就是你Index.php這個文件名,你寫小寫的index.php，也能為True,這里有個要注意的地方,就是要修改currentIndex的值,必須在前面加上global,否則還是為空

def judgeFileExist():
 global currentIndex;
 currentIndex = os.getcwd() + '/Index.php'
 if os.path.isfile(currentIndex) == False:
  print('Index文件不存在')
  exit()
 print('文件檢測成功,準備連接服務(wù)器...')

這是連接服務(wù)器并創(chuàng)建SFTP,使用了Try來捕獲異常錯誤

def creatConnect():
 try:
  print('開始連接服務(wù)器...')
  s = paramiko.Transport((host, port))
  s.connect(username=username, password=password)
  sftp = paramiko.SFTPClient.from_transport(s)
  print('連接:' + host + '成功')
  return sftp,s
 except Exception as e:
  print('連接服務(wù)器失敗:' + str(e))

這里就是執(zhí)行操作命令了,使用sftp對象來操作,sftp.chdir是用于切換目錄,相當于shell命令的cd /www/wwwroot
sftp.listdir(path='.')是返回當前目錄下的文件夾,且是以數(shù)組形式返回,然后將其拼接成完整路徑后再保存在本地數(shù)組里備用,這里有個if in是用來跳過一些網(wǎng)站目錄，比如我xxx.demo.com這個目錄不想更新,就在開頭的SkipArry里寫上,用來跳過

def getDirectory(sftp):
 print('開始獲取目錄...')
 sftp.chdir('/www/wwwroot')
 pathlist = sftp.listdir(path='.')
 for path in pathlist:
  fullpath = '/www/wwwroot/' + path + '/application/index/controller'
  if path in skipArry:
   continue
  fullpathArry.append(fullpath)
 print('目錄獲取完畢')

這里就是關(guān)鍵的上傳部分了,首先遍歷出我們需要修改的文件夾目錄，后面拼接上需要修改的文件Index.php形成遠程服務(wù)器的文件路徑,然后使用sftp.put函數(shù)來上傳我們的文件,第一個參數(shù)是本地文件的路徑，第二個參數(shù)是遠程服務(wù)器上的路徑,上傳成功后使用sftp.file來驗證該文件是否存在,其實這里我是做了個備份處理的(有點bug就暫時先注釋掉了),先將原本的Index.php改名為BackIndex.php在上傳新的Index.php,這個判斷函數(shù)才有用,不然我這樣寫沒啥用,因為上沒上傳成功肯定都會存在一個Index.php文件.上傳好了之后使用sftp.chmod方法來改變該文件的權(quán)限為755,這里有個坑，你直接在第二個參數(shù)寫755,會發(fā)現(xiàn)生成的文件權(quán)限為363,經(jīng)過多次試驗發(fā)現(xiàn),第二個參數(shù)要傳入8進制的755,也就是493,生成的權(quán)限就是755了,感覺有點坑爹。

def uploadIndex(sftp):
 for fullpathitem in fullpathArry:
   remoteIndex = fullpathitem + '/Index.php'
   print('開始上傳:' + remoteIndex)
   try:
    sftp.put(currentIndex, remoteIndex)
    try:
     sftp.file(remoteIndex)
     sftp.chmod(remoteIndex, int("775", 8))
     print('修改' + remoteIndex + '權(quán)限為755')
     print(fullpathitem + '上傳成功')
    except:
     print(fullpathitem + '上傳失敗')
     continue
   except Exception as e:
    print('錯誤信息:' + str(e))
    continue

然后在main里依次執(zhí)行,就能將服務(wù)器上對應(yīng)的目錄下的文件全部替換成我本地的文件了,代碼不多，但效果好使啊，果然是人生苦短，我用Python

總結(jié)

以上就是這篇文章的全部內(nèi)容了，希望本文的內(nèi)容對大家的學習或者工作具有一定的參考學習價值，如果有疑問大家可以留言交流，謝謝大家對腳本之家的支持。

您可能感興趣的文章: