python argparse模块使用

argparse 模块可以让人轻松编写用户友好的命令行接口。程序定义它需要的参数,然后 argparse 将弄清如何从 sys.argv 解析出那些参数。 argparse 模块还会自动生成帮助和使用手册,并在用户给程序传入无效参数时报出错误信息。

使用示例:

# coding: utf-8

import requests
import argparse

def op_interface(host, su, code):
    """调用控制机器流量接口"""
    url = "http://chegva.com/auth/v3/xxx"
    payload = "su={}&status_code={}&hostnames={}".format(su,code,host)
    print(payload)
    headers = { 'Cookie': "xxx" }
    response = requests.request("POST", url, data=payload, headers=headers)
    print(response.text)
    
def main():
    parser = argparse.ArgumentParser(description='op interface created by anzhihe')
    parser.add_argument('host', type=str, help='输入机器主机名')
    parser.add_argument('su', type=str, help='输入机器所在节点su')
    parser.add_argument('code', type=int, help='机器接流输入1,禁用输入0')
    args = parser.parse_args()
    host, su, code = args.host, args.su, args.code
    op_interface(host, su, code)
    
if __name__ == '__main__':
    main()
    
    
# 使用
$ opi -h    
                                                             
usage: opi [-h] host su code

op interface created by anzhihe

positional arguments:
  host        输入机器主机名
  su          输入机器所在节点su
  code        机器接流输入1,禁用输入0
  
optional arguments:
  -h, --help  show this help message and exit

 argparse模块使用:

anzhihe 安志合个人博客,版权所有 丨 如未注明,均为原创 丨 转载请注明转自:https://chegva.com/3667.html | ☆★★每天进步一点点,加油!★★☆ | 

您可能还感兴趣的文章!

发表评论

电子邮件地址不会被公开。 必填项已用*标注