#!/usr/bin/python
# -*- coding: utf-8 -*-
# 通用功能类封装
importos,time,sys,string,urllib,httplib,shutil,platform,tarfile
fromcommandsimportgetstatusoutput as getso
fromConfigParserimport*
defwr_log(str_cmd, status=0, result=""):
cur_time = time.strftime("%Y-%m-%d %H:%M:%S")
ifstatus ==0:
fp = open("../log/log.txt",'a+')
fp.write("[%s]\t%s\tdone!\n"% (cur_time, str_cmd))
fp.close()
else:
fp = open("../log/err_log.txt",'a+')
fp.write("[%s]\t%s\tfailed!\n%s\n\n"% (cur_time, str_cmd, result))
fp.close()
sys.exit(-1)
defwget(url, new_name=""):
'''''
wget封装,需提供下载地址,新文件名参数可省略。
例子:
wget("http://208.asktao.com/test.txt", "/tmp/test.txt")
'''
try:
file_name = url[url.rfind("/")+1:]
ifnew_name == "":
new_name = file_name
fp = urllib.urlopen(url)
py_ver = sys.version[:3]
ifpy_ver =="2.4":
domain = url.split("/")[2]
get_file ="/"+"/".join(url.split("/")[3:])
conn = httplib.HTTPConnection(domain)
conn.request('GET', get_file)
fp_code = conn.getresponse().status
else:
fp_code = fp.getcode()
iffp_code !=200:
raiseNameError,'%s not exist.'%file_name
buf_len =2048
f = open(new_name,'wb')
size =0
while1:
s = fp.read(buf_len)
ifnots:
break
f.write(s)
size += len(s)
fp.close()
f.close()
wr_log("wget %s"%url)
exceptException, e:
wr_log("wget %s"%url,1, e)
defsed(type, file_name, s_str="", d_str=""):
'''''
sed封装,根据传入的type参数执行相应操作,type可以为以下几种:
a 在s_str行后面追加d_str,若s_str为空,则默认在文件尾部追加;
i 在s_str行前面插入d_str,若s_str为空,则默认在文件头部插入;
d 删除s_str所在行;
s 将s_str替换为d_str。
type及file_name为必需的参数。
例子:
替换字符串:sed("s", "/etc/test.txt", "abc", "123")
'''
try:
fp = open(file_name)
cont = fp.read()
fp.close()
content = cont.split("\n")
content2 = cont.split("\n")
cnt =0
idx =0
iftype =="a":
str_cmd ="sed -i '/%s/a %s' %s"% (s_str, d_str, file_name)
ifnots_str:
content.append(d_str)
else:
foriincontent2:
ifi.find(s_str) != -1:
x = idx +1+ cnt
content.insert(x, d_str)
cnt +=1
idx +=1
eliftype =="i":
str_cmd ="sed -i '/%s/i %s' %s"% (s_str, d_str, file_name)
ifnots_str:
content.insert(0, d_str)
else:
foriincontent2:
ifi.find(s_str) != -1:
x = idx + cnt
content.insert(x, d_str)
cnt +=1
idx +=1
eliftype =="d":
str_cmd ="sed -i '/%s/d' %s"% (s_str, file_name)
foriincontent2:
ifi.find(s_str) != -1:
idx = content.remove(i)
eliftype =="s":
str_cmd ="sed -i 's/%s/%s/g' %s"% (s_str, d_str, file_name)
cont = string.replace(cont, s_str, d_str)
content = cont.split("\n")
fp = open(file_name,"w")
fp.write("\n".join(content))
fp.close()
wr_log(str_cmd)
exceptException, e:
wr_log("modify %s"% file_name,1, e)
defmd(dir_name):
'''''
mkdir封装,创建传入的目录名称。
'''
try:
ifnotos.path.exists(dir_name):
os.makedirs(dir_name)
wr_log("mkdir %s"%dir_name)
else:
wr_log("%s is exist."%dir_name)
exceptException, e:
wr_log("mkdir %s"%dir_name,1, e)
defmv(src, dst):
'''''
mv封装,移动文件或修改文件名。
'''
try:
try:
os.rename(src, dst)
exceptOSError:
ifos.path.dirname(src) == dstoros.path.dirname(src) == os.path.dirname(dst):
raiseException,"Cannot move self."
ifos.path.isdir(src):
ifos.path.abspath(dst).startswith(os.path.abspath(src)):
raiseException,"Cannot move a directory to itself."
_copy(src, dst)
shutil.rmtree(src)
else:
shutil.copy2(src,dst)
os.unlink(src)
wr_log("mv %s %s"%(src, dst))
exceptException, e:
wr_log("mv %s %s"%(src, dst),1, e)
defcp(src, dst):
'''''
cp封装,复制文件或目录。
'''
try:
_copy(src, dst)
wr_log("cp %s %s"%(src, dst))
exceptException, e:
wr_log("cp %s %s"%(src, dst),1, e)
def_copy(src, dst):
'''''
copy封装,供cp调用。
'''
ifos.path.isdir(src):
base = os.path.basename(src)
ifos.path.exists(dst):
dst = os.path.join(dst, base)
ifnotos.path.exists(dst):
os.makedirs(dst)
names = os.listdir(src)
fornameinnames:
srcname = os.path.join(src, name)
_copy(srcname, dst)
else:
shutil.copy2(src, dst)
deftouch(src):
'''''
linux适用
touch封装,新建空白文件。
'''
str_cmd ="/bin/touch %s"% src
status, result = getso(str_cmd)
wr_log(str_cmd, status, result)
defrm(src):
'''''
rm封装,删除文件或目录,非交互式操作,请谨慎操作。
'''
try:
ifos.path.exists(src):
ifos.path.isfile(src):
os.remove(src)
elifos.path.isdir(src):
shutil.rmtree(src)
wr_log("rm %s"% src)
exceptException, e:
wr_log("rm %s"% src,1, e)
defchmod(num, file_name):
'''''
linux适用
chmod封装,修改文件权限。
需要传入一个八进制数以及文件名。
例子:
chmod(644, "/tmp/test.txt")
'''
str_cmd ="/bin/chmod %s %s"% (num, file_name)
status, result = getso(str_cmd)
wr_log(str_cmd, status, result)
defchown(user, file_name, arg=""):
'''''
linux适用
chown封装,修改文件属主属组。
例子:
chown("nobody.nobody", "/tmp", "r")
'''
ifarg =="r":
str_cmd ="/bin/chown -R %s %s"% (user, file_name)
else:
str_cmd ="/bin/chown %s %s"% (user, file_name)
status, result = getso(str_cmd)
wr_log(str_cmd, status, result)
defrar(dst,src):
'''''
windows适用
winrar 压缩文件 如: rar(c:\dat.rar c:\dat)
'''
try:
ifnotos.path.exists(src)ornotos.path.exists(os.path.dirname(dst)):
raiseException,"%s or %s not exist!"% (src, os.path.dirname(dst))
os.system(r'C:\Progra~1\WinRAR\rar a %s %s'% (dst,src))
wr_log("rar %s to %s"% (src, dst))
exceptException,e:
wr_log("rar %s to %s"% (src, dst),1, e)
defunrar(src,dst):
'''''
windows适用
unrar 解压缩rar文件 到目标路径 如:unrar(c:\dat.rar c:\)
'''
try:
ifnotos.path.exists(dst)ornotos.path.exists(src):
raiseException,"%s or %s not exist!"% (src, dst)
os.system(r'C:\Progra~1\WinRAR\rar x %s %s'% (src, dst))
wr_log("unrar %s"% src)
exceptException,e:
wr_log("unrar %s"% src,1, e)
deftar(dst, src):
'''''
linux适用
tar封装,压缩文件。
例子:
tar("/tmp/test.tgz", "/tmp/test.txt")
'''
str_cmd ="/bin/tar zcf %s %s"% (dst, src)
status, result = getso(str_cmd)
wr_log(str_cmd, status, result)
defuntgz(tgz_file, dst="./"):
'''''
tar封装,解压缩文件。
例子:
untgz("/tmp/test.tgz", "/tmp")
'''
try:
tarobj = tarfile.open(tgz_file)
names = tarobj.getnames()
fornameinnames:
tarobj.extract(name,path=dst)
tarobj.close()
wr_log("untgz %s"% tgz_file)
exceptException, e:
wr_log("untgz %s"% tgz_file,1, e)
defkill(arg):
'''''
linux中,查找进程并杀死返回的pid。
windows中,查找进程或端口并杀死返回的pid。
'''
pid = get_pid(arg)
os_type = platform.system()
ifos_type =="Linux":
ifpid:
str_cmd ="/bin/kill -9 %s"% pid
status, result = getso(str_cmd)
wr_log("kill %s"% arg, status, result)
elifos_type =="Windows":
ifpid:
try:
importctypes
foriinpid:
handle = ctypes.windll.kernel32.OpenProcess(1,False, i)
ctypes.windll.kernel32.TerminateProcess(handle,0)
wr_log("kill %s"% arg)
exceptException, e:
wr_log("kill %s"% arg,1, e)
defget_pid(arg):
'''''
linux中,查找进程并返回pid。
windows中,查找进程或端口返回pid。
'''
os_type = platform.system()
ifos_type =="Linux":
str_cmd ="/bin/ps auxww | grep '%s' | grep -v grep | awk '{print $2}'"% arg
status, result = getso(str_cmd)
returnresult
elifos_type =="Windows":
iftype(arg) == int:
str_cmd ="netstat -ano|find \"%s\""%arg
try:
result = os.popen(str_cmd,"r").read()
result = result.split("\n")[0].strip()
ifresult.find("WAIT") != -1:
return0
pid = int(result[result.rfind(" "):].strip())
return[pid]
exceptException, e:
return0
else:
importwin32con,win32api,win32process
pids = []
forpidinwin32process.EnumProcesses():
try:
hProcess = win32api.OpenProcess(win32con.PROCESS_ALL_ACCESS,False, pid)
hProcessFirstModule = win32process.EnumProcessModules(hProcess)[0]
processName = os.path.splitext(os.path.split(win32process.GetModuleFileNameEx(hProcess, hProcessFirstModule))[1])[0]
ifprocessName == arg:
pids.append(pid)
exceptException, e:
pass
returnpids
defgrpadd(grp_name):
'''''
linux适用
groupadd封装,增加组。
'''
str_cmd ="/usr/sbin/groupadd %s"% grp_name
status, result = getso(str_cmd)
wr_log(str_cmd, status, result)
defuseradd(user_name, arg=""):
'''''
useradd封装,添加新用户;
linux和windows系统分别使用不同方法;
在linux中,arg填写新用户的gid;
windows中,arg填写新用户的密码。
'''
os_type = platform.system()
ifos_type =="Linux":
str_cmd ="/usr/bin/id %s"% user_name
status, result = getso(str_cmd)
ifstatus ==0:
return
ifnotarg:
str_cmd ="/usr/sbin/useradd %s"% user_name
else:
str_cmd ="/usr/sbin/useradd -g %s %s"% (arg, user_name)
status, result = getso(str_cmd)
wr_log(str_cmd, status, result)
elifos_type =="Windows":
try:
importwin32netcon,win32net,wmi
foruseinwmi.WMI().Win32_UserAccount():
ifuse.name == user_name:
raiseException,"user %s is already exists"% user_name
udata = {}
udata["name"] = user_name
udata["password"] = arg
udata["flags"] = win32netcon.UF_NORMAL_ACCOUNT | win32netcon.UF_SCRIPT
udata["priv"] = win32netcon.USER_PRIV_USER
win32net.NetUserAdd(None,1, udata)
wr_log("add user %s"% user_name)
exceptException,e:
wr_log("add user %s"% user_name,1, e)
defuserdel(user_name):
'''''
userdel封装,删除用户。
'''
os_type = platform.system()
ifos_type =="Linux":
str_cmd ="/usr/bin/id %s"% user_name
status, result = getso(str_cmd)
ifstatus ==0:
str_cmd ="/usr/sbin/userdel -r %s"% user_name
status, result = getso(str_cmd)
wr_log(str_cmd, status, result)
elifos_type =="Windows":
try:
importwin32net,wmi
foruseinwmi.WMI().Win32_UserAccount():
ifuse.name == user_name:
win32net.NetUserDel(None,user_name)
wr_log("del user %s"% user_name)
return
wr_log("user %s not exists"% user_name)
exceptException,e:
wr_log("del user %s"% user_name,1, e)