hanxia 2020-08-17
pip3 install prettytable
url 为 资产信息接口地址,返回为json信息。
# encoding=utf-8 import getopt import sys import prettytable as pt import requests import json def main(argv): try: options, args = getopt.getopt(argv, "n:", ["name=", ]) except getopt.GetoptError: sys.exit() for option, value in options: if option in ("-n", "--name"): url = ‘http://xxxxxxxx/list‘ try: headers = {‘Content-Type‘: ‘application/json‘} r = requests.post(url, data=json.dumps({"name": value}), headers=headers) if r.status_code == 200: data = r.json() tb = pt.PrettyTable() tb.field_names = ["主机名", "外网IP"] tb.align["主机名"] = "l" tb.align["外网IP"] = "l" for i in data: tb.add_row([i["_id"], i["out_ip"]]) print(tb) else: print("获取信息错误") except Exception as e: print(e) if __name__ == ‘__main__‘: main(sys.argv[1:])
执行: /usr/bin/python3.6 test.py -n test