脚本编程与系统管理

脚本编程与系统管理

通过重定向/管道/文件接受输入

  • 使用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
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,214评论 6 481
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,307评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 152,543评论 0 341
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,221评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,224评论 5 371
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,007评论 1 284
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,313评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,956评论 0 259
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,441评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,925评论 2 323
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,018评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,685评论 4 322
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,234评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,240评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,464评论 1 261
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,467评论 2 352
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,762评论 2 345

推荐阅读更多精彩内容

  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,599评论 18 139
  • Ubuntu的发音 Ubuntu,源于非洲祖鲁人和科萨人的语言,发作 oo-boon-too 的音。了解发音是有意...
    萤火虫de梦阅读 99,156评论 9 467
  • Spring Boot 参考指南 介绍 转载自:https://www.gitbook.com/book/qbgb...
    毛宇鹏阅读 46,748评论 6 342
  • linux资料总章2.1 1.0写的不好抱歉 但是2.0已经改了很多 但是错误还是无法避免 以后资料会慢慢更新 大...
    数据革命阅读 12,134评论 2 34
  • Android 自定义View的各种姿势1 Activity的显示之ViewRootImpl详解 Activity...
    passiontim阅读 171,504评论 25 707