Python3调用API打印服务器资产信息的脚本实现

# 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:])