python封裝常用系統(tǒng)命令
python寫的系統(tǒng)常用命令，linux和windows通用倔撞，用的時(shí)候直接from util import *導(dǎo)入即可使用吕粹，很方便
#!/usr/bin/python
# -*- coding: utf-8 -*-
# 通用功能類封裝
import os,time,sys,string,urllib,httplib,shutil,platform,tarfile
from commands import getstatusoutput as getso
from ConfigParser import *
def hostname(host_name):
'''''
linux適用
hostname封裝庭敦，修改主機(jī)名历涝。
'''
str_cmd = "/bin/sed -i 's/HOSTNAME/#&/;$a HOSTNAME=%s' /etc/sysconfig/network;/bin/hostname %s" % (host_name,host_name)
status, result = getso(str_cmd)
wr_log(str_cmd, status, result)
def md5sum(file_name):
'''''
md5sum封裝扭倾，獲取文件的md5值
'''
if os.path.isfile(file_name):
f = open(file_name,'rb')
py_ver = sys.version[:3]
if py_ver == "2.4":
import md5 as hashlib
else:
import hashlib
md5 = hashlib.md5(f.read()).hexdigest()
f.close()
return md5
else:
return 0
def md5(file_name):
'''''
linux適用
md5sum -c 封裝淀零，校驗(yàn)md5文件，返回校驗(yàn)成功或失敗狀態(tài)
'''
str_cmd="/usr/bin/md5sum -c %s" % file_name
status,result=getso(str_cmd)
return status
def grep(s_str, file_name):
'''''
grep封裝膛壹，查找文件中關(guān)鍵字驾中，有則返回所在行，否則返回空字符串模聋。
'''
try:
fd = open(file_name)
content = fd.read()
result = ""
if content.find(s_str) != -1:
for line in content.split("\n"):
if line.find(s_str) != -1:
result = result + line + "\n"
return result.strip()
except Exception, e:
wr_log("grep %s %s" % (s_str, file_nsme), 1, e)
def rwconf(type, file_name, section, option, s_str=""):
'''''
讀取標(biāo)準(zhǔn)的ini格式配置文件肩民，type可為以下值：
get 獲取section下的option的值，值為字符串链方；
getint 獲取section下的option的值持痰，值為數(shù)字；
modi 修改section下的option的值祟蚀，并保存工窍；
del 刪除section下的option，并保存前酿。
注：option嚴(yán)格區(qū)分大小寫
'''
try:
if type == "get" or type == "getint":
cf = ConfigParser()
else:
cf = ConfParser()
cf.read(file_name)
if type == "get":
return cf.get(section, option)
elif type == "getint":
return cf.getint(section, option)
elif type == "modi":
try:
cf.set(section, option, s_str)
cf.write(open(file_name, "w"))
wr_log("modify %s for %s" % (option, file_name))
except Exception, e:
wr_log("modify %s for %s" % (option, file_name), 1, str(e))
elif type == "del":
try:
cf.remove_option(section, option)
cf.write(open(file_name, "w"))
wr_log("del %s for %s" % (option, file_name))
except Exception, e:
wr_log("del %s for %s" % (option, file_name), 1, str(e))
except Exception, e:
wr_log("read %s for %s" % (option, file_name), 1, str(e))
def chkconfig(type, svr_name, switch=""):
'''''
linux適用
chkconfig封裝患雏，根據(jù)傳入的type參數(shù)執(zhí)行相應(yīng)操作，type可以為以下幾種：
add 添加服務(wù)至啟動(dòng)項(xiàng)罢维；
del 從啟動(dòng)項(xiàng)刪除服務(wù)淹仑；
數(shù)字 指定運(yùn)行級(jí)別的服務(wù)開啟或關(guān)閉。
type及svr_name為必需的參數(shù)肺孵。
例子：
開啟運(yùn)行級(jí)別3的sshd服務(wù)：chkconfig("3", "sshd", "on")
'''
if type != "add" and type != "del":
type = "--level %s" % str(type)
str_cmd = "/sbin/chkconfig %s %s %s" % (type, svr_name, switch)
status, result = getso(str_cmd)
wr_log(str_cmd, status, result)
def passwd(user_name,newpass):
'''''
passwd封裝匀借，修改用戶密碼
'''
os_type = platform.system()
if os_type == "Linux":
str_cmd = "echo '%s' | passwd %s --stdin" % (newpass, user_name)
status, result = getso(str_cmd)
wr_log(str_cmd, status, result)
elif os_type == "Windows":
try:
if os.system('net user %s "%s" ' %(user_name,newpass)) == 0:
wr_log("modify passwd for %s " % user_name)
elif os.system('net user %s "%s" ' %(user_name,newpass)) == 2:
raise Exception, "user %s isnot exists" % user_name
except Exception,e:
wr_log("modify passwd for %s " % user_name, 1, e)
def echo(str, file_name):
'''''
linux適用
echo封裝，添加字符串到文件尾部
'''
str_cmd = "/bin/echo '%s' >> %s" % (str, file_name)
status, result = getso(str_cmd)
wr_log(str_cmd, status, result)
def upload(localfiles, remotepath, host="xxx", username="xxx", password="xxxx"):
'''''
上傳文件至ftp服務(wù)器平窘，默認(rèn)上傳至208FTP怀吻，如要上傳至其它FTP服務(wù)器，請(qǐng)指定host/user/pass
例：
upload("a.txt,b.txt", "/test/")
上傳a.txt初婆、b.txt文件到208的test目錄下
'''
import base64
from ftplib import FTP
try:
localfiles = localfiles.split(",")
f =FTP(host)
f.login(username,password)
f.cwd(remotepath)
for localfile in localfiles:
fd = open(localfile,'rb')
f.storbinary('STOR %s' % os.path.basename(localfile),fd)
fd.close()
f.quit()
wr_log("upload %s" % localfiles)
except Exception, e:
wr_log("upload %s" % localfiles, 1, e)
class ConfParser(RawConfigParser):
'''''
ConfigParser模塊有一個(gè)缺陷,改寫ini文件的某個(gè)section的某個(gè)option,寫入ini文件后
ini文件的注釋都丟掉了,并且option的大寫字母都轉(zhuǎn)換成了小寫
為了保存ini文件的注釋以及option的大小寫蓬坡，重寫了write、set磅叛、optionxform等方法屑咳，由rwconf函數(shù)調(diào)用
'''
def write(self, fp):
"""Write an .ini-format representation of the configuration state.
write ini by line no
"""
if self._defaults:
section = DEFAULTSECT
lineno = self._location[section]
self._data[lineno] = "[%s]\n" %section
for (key, value) in self._defaults.items():
if key != "__name__":
wholename = section + '_' + key #KVS
lineno = self._location[wholename]
self._data[lineno] = "%s = %s\n" %(key, str(value).replace('\n', '\n\t'))
for section in self._sections:
lineno = self._location[section]
self._data[lineno] = "[%s]\n" % section
for (key, value) in self._sections[section].items():
if key != "__name__":
wholename = section + '_' + key #KVS
lineno = self._location[wholename]
self._data[lineno] = "%s = %s\n" %(key, str(value).replace('\n', '\n\t'))
for line in self._data:
fp.write("%s"%line)
fp.close()
def _read(self, fp, fpname):
"""Parse a sectioned setup file.
When parsing ini file, store the line no in self._location
and store all lines in self._data
"""
self._location = {}
self._data = []
cursect = None # None, or a dictionary
optname = None
lineno = 0
e = None # None, or an exception
while True:
line = fp.readline()
self._data.append(line) #KVS
if not line:
break
lineno = lineno + 1
if line.strip() == '' or line[0] in '#;':
continue
if line.split(None, 1)[0].lower() == 'rem' and line[0] in "rR":
# no leading whitespace
continue
if line[0].isspace() and cursect is not None and optname:
value = line.strip()
if value:
cursect[optname] = "%s\n%s" % (cursect[optname], value)
else:
mo = self.SECTCRE.match(line)
if mo:
sectname = mo.group('header')
if sectname in self._sections:
cursect = self._sections[sectname]
elif sectname == DEFAULTSECT:
cursect = self._defaults
self._location[DEFAULTSECT] = lineno -1 #KVS
else:
cursect = {'__name__': sectname}
self._location[sectname] = lineno -1 #KVS
self._sections[sectname] = cursect
optname = None
elif cursect is None:
raise MissingSectionHeaderError(fpname, lineno, line)
else:
mo = self.OPTCRE.match(line)
if mo:
optname, vi, optval = mo.group('option', 'vi', 'value')
if vi in ('=', ':') and ';' in optval:
pos = optval.find(';')
if pos != -1 and optval[pos-1].isspace():
optval = optval[:pos]
optval = optval.strip()
if optval == '""':
optval = ''
optname = self.optionxform(optname.rstrip())
cursect[optname] = optval
if cursect == self._defaults:
wholename = DEFAULTSECT + '_' + optname #KVS
else:
wholename = cursect['__name__'] + '_' + optname #KVS
self._location[wholename] = lineno-1 #KVS
else:
if not e:
e = ParsingError(fpname)
e.append(lineno, repr(line))
if e:
raise e
def add_section(self, section):
"""Create a new section in the configuration.
Raise DuplicateSectionError if a section by the specified name
already exists.
"""
if section in self._sections:
raise DuplicateSectionError(section)
self._sections[section] = {}
linecount = len(self._data)
self._data.append('\n')
self._data.append('%s'%section)
self._location[section] = linecount + 1
def set(self, section, option, value):
"""Set an option."""
if not section or section == DEFAULTSECT:
sectdict = self._defaults
else:
try:
sectdict = self._sections[section]
except KeyError:
raise NoSectionError(section)
option = self.optionxform(option)
add = False
if not option in sectdict:
add = True
sectdict[self.optionxform(option)] = value
if add:
lineno = self._location[section]
self._data.append('')
idx = len(self._data)
while idx>lineno:
self._data[idx-1] = self._data[idx-2]
idx = idx-1
self._data[idx+1] = '%s = %s\n'%(option,value)
self._location[section+'_'+option]=idx+1
for key in self._location:
if self._location[key] > lineno:
self._location[key] = self._location[key] + 1
self._data[idx+1] = '%s = %s\n'%(option,value)
self._location[section+'_'+option]=idx+1
def remove_option(self, section, option):
"""Remove an option. """
if not section or section == DEFAULTSECT:
sectdict = self._defaults
else:
try:
sectdict = self._sections[section]
except KeyError:
raise NoSectionError(section)
option = self.optionxform(option)
existed = option in sectdict
if existed:
del sectdict[option]
wholename = section + '_' + option
lineno = self._location[wholename]
del self._location[wholename]
for key in self._location:
if self._location[key] > lineno:
self._location[key] = self._location[key] -1
del self._data[lineno]
return existed
def remove_section(self, section):
"""Remove a file section."""
existed = section in self._sections
if existed:
lstOpts = []
for option in self._sections[section]:
if option == '__name__':
continue
lstOpts.append(option)
for option in lstOpts:
self.remove_option(section,option)
del self._sections[section]
wholename = section
lineno = self._location[wholename]
del self._location[wholename]
for key in self._location:
if self._location[key] > lineno:
self._location[key] = self._location[key] -1
del self._data[lineno]
return existed
def optionxform(self, optionstr):
''''' 防止大小寫轉(zhuǎn)換'''
return optionstr
