Python處理命令行參數
1. 將命令行參數保存在列表中,註意argv[0]是程序本身的名字:
import sys print(sys.argv) print(sys.argv[1])
python argv.py localhost 3306
[‘argv.py‘, ‘localhost‘, ‘3306‘]
localhost
2. 使用sys.stdin和fileinput讀取標準輸入,並打印在終端類似shell中的管道
import sys for line in sys.stdin: print(line,end="")
可以像shell腳本一樣,同過標準輸入給程序輸入內容
python read_stdin.py </etc/passwd
python read_stdin.py -
cat /etc/passwd |python read_stdin.py
將標準輸入保存在一個列表中
import sys def get_content(): return sys.stdin.readlines() print(get_content())
python readlines_stdin.py <test/1.txt
[‘hello\n‘, ‘world\n‘]
3. 利用fileinput讀取標準輸入
#/usr/bin/env python #coding=utf-8 import fileinput for line in fileinput.input(): print(line,end="")
python file_input1.py /etc/passwd
python file_input1.py </etc/passwd
python file_input1.py /etc/passwd /etc/my.cnf
fileinput常用的方法:
#!/usr/bin/python from __future__ import print_function import fileinput for line in fileinput.input(): meta = [fileinput.filename(), fileinput.fileno(), fileinput.filelineno(), fileinput.isfirstline(), fileinput.isstdin()] print(*meta, end="")
print() print(line, end="")
4. 使用getpass讀取密碼:
import getpass user=getpass.getuser() passwd=getpass.getpass(‘you password: ‘) print(user,passwd)
可以避免輸入密碼被看見
5.使用argparse解析命令行參數
agrparse能夠根據從sys.arg中解析參數,並自動生成幫助信息
from __future__ import print_function import argparse def _argparse(): parser = argparse.ArgumentParser(description="This is description") parser.add_argument(‘--host‘, action=‘store‘, dest=‘server‘,default="localhost", help=‘connect to host‘) parser.add_argument(‘-t‘, action=‘store_true‘, default=False, dest=‘boolean_switch‘, help=‘Set a switch to true‘) return parser.parse_args() def main(): parser = _argparse() print(parser) print(‘host =‘, parser.server) print(‘boolean_switch=‘, parser.boolean_switch) if __name__ == ‘__main__‘: main()
格式:rgumentParser.add_argument(name or flags...[, action][, nargs][, const][, default][, type][, choices][, required][, help][, metavar][, dest])
name:參數的名字
action: 遇到參數時的動作
nargs:參數的個數
dest:解析後的參數的名字
type:參數的類型
6. 使用Click創建命令行解析
Click比較argparse更加的快速和簡單
pip inst click
import click @click.command() @click.option(‘--count‘, default=1, help=‘Number of greetings.‘) @click.option(‘--name‘, prompt=‘Your name‘, help=‘The person to greet.‘) def hello(count, name): """Simple program that greets NAME for a total of COUNT times.""" for x in range(count): click.echo(‘Hello %s!‘ % name) if __name__ == ‘__main__‘: hello()
commond讓函數成為命令行接口
option:增加命令行選項
echo:輸出結果
prompt:如果沒有指定name這個參數時,會進入交互模式下輸入
也可以像Linux中的fc一樣進入默認編輯器
import click message = click.edit() print(message,end="")
Python處理命令行參數