脚本编程与系统管理
通过重定向/管道/文件接受输入
- 使用fileinput接收python 代码的输入
- 将python 文件变为可执行文件,可以在f_input中遍历输出
import fileinput
with fileinput.input() as f_input:
for line in f_input:
print (line, end='')
use:
./filein.py /etc/passwd
./filein.py < /etc/passwd
终止程序并给出错误信息
- 标准错误打印一条消息并返回某个非零状态码来终止程序运行
raise SystemExit('It failed!')
解析命令行选项
- 首先要创建一个 ArgumentParser实例,并使用 add_argument() 方法声明你想要支持的选项
- dest 参数指定解析结果被指派给属性的名字
- metavar 参数被用来生成帮助信息
- action 参数指定跟属性对应的处理逻辑,通常的值为 store(结果存储为字符串)store_true(s设置一个默认的boolean标识)append(追加一个列表)
- required 标志表示该参数至少要有一个。-p 和 --pat 表示两个参数名形式都可使用。
- choices={'slow', 'fast'}, default='slow',下面的参数说明接受一个值
import argparse
parser = argparse.ArgumentParser(description='Search some files')
parser.add_argument(dest='filenames', metavar='filename', nargs='*')
parser.add_argument('-p', '--pat', metavar='pattern', required=True,
dest='patterns', action='append',
help='text pattern to search for')
parser.add_argument('-v', dest='verbose', action='store_true',
help='verbose mode'
)
parser.add_argument('-o', dest='outfile', action='store',
help='output file'
)
parser.add_argument('--speed', dest='speed', action='store',
choices={'slow', 'fast'}, default='slow',
help='search speed'
)
args = parser.parse_args()
print (args.filenames)
print (args.patterns)
print (args.verbose)
print (args.outfile)
print (args.speed)
命令行输入密码
import getpass
passwd = getpass.getpass()
print (passwd)
执行外部命令并获取输出
- 使用subprocess获取标准输出的值和错误信息及其返回码
import subprocess
try:
out_bytes = subprocess.check_output(['cd', 'arg2'])
except subprocess.CalledProcessError as e:
out_bytes = e.output
code = e.returncode
print (code)
- 通常情况下,命令不会直接执行shell,会执行shell底层的函数,传递shell=True显式的声明执行shell
out_bytes = subprocess.check_output('ls', shell=True)
print (out_bytes.decode('utf8'))
- 使用Popen做更加复杂的操作,使用communicate需要从定向标准输出的标准输入
text = b"""
hello world
"""
p = subprocess.Popen(
['wc'],
stdout = subprocess.PIPE,
stdin = subprocess.PIPE
)
stdout, stderr = p.communicate(text)
#转为unicode
print (stdout.decode('utf8'))
使用shutil 复制文件
- 使用shutil复制文,并处理软连接
- 复制过程中出现异常,回抛出到Error中
try:
shutil.copytree(src, dst)
except shutil.Error as e:
for src, dst, msg in e.args[0]:
print(dst, src, msg)
创建和解压归档文件
shutil.unpack_archive('Python-3.3.0.tgz')
#第一个参数为打包的文件名字,最有一个参数为需要打包的文件夹
shutil.make_archive('py33','zip','Python-3.3.0')
通过文件名查找文件
- 可使用 os.walk() 函数,传一个顶级目录名给它
- os.walk() 方法为我们遍历目录树, 每次进入一个目录,它会返回一个三元组,包含相对于查找目录的相对路径,一个该目录下的目录名列表, 以及那个目录下面的文件名列表。
import os
import sys
def findfile(start, name):
for relpath, dirs, files in os.walk(start):
print (relpath)
if name in files:
full_path = os.path.join(start, relpath, name)
print (os.path.abspath(full_path))
if __name__ == '__main__':
findfile(sys.argv[1], sys.argv[2])
读取类型ini的配置文件
>>> from configparser import ConfigParser
>>> cfg = ConfigParser()
>>> cfg.read('config.ini')
['config.ini']
>>> cfg.sections()
['installation', 'debug', 'server']
>>> cfg.get('installation', 'library')
'/usr/local/lib'
>>> cfg.get('debug', 'log_errors')
'true'
>>> cfg.getboolean('debug', 'log_errors')
True
>>> cfg.getint('server', 'port')
8080
>>> cfg.get('server', 'signature')
'\n=================================\nBrought to you by the Python Cookbook\n===========
======================'
et('server','port','9000')
>>> import sys
>>> cfg.write(sys.stdout)
简单脚本增加日志功能
- 使用logging模块
- level=logging.INFO只输出info或比其级别高的日志,filename日志会定向到文件中,默认为标准输出
- format可以给日志加头
- logging.getLogger().level = logging.DEBUG 可以动态的修改日志配置
import logging
def main():
logging.basicConfig(
filename="app.log",
level=logging.INFO,
format='%(levelname)s:%(asctime)s:%(message)s'
)
hostname = 'www.python.org'
item = 'spam'
filename = 'data.csv'
mode = 'r'
logging.critical('Host %s connection', hostname)
logging.error("Couldn't find %r", item)
logging.warning('Feature is deprecated')
logging.info('Opening file %r, mode=%r', filename, mode)
logging.debug('Got here')
if __name__ == '__main__':
main()
打开浏览器
>>> import webbrowser
>>> c = webbrowser.get('safari')
>>> c.open('http://www.python.org')
True