浏览代码

Clean up with black.

Joe Clarke 4 年之前
父节点
当前提交
c326f60f7e
共有 1 个文件被更改,包括 39 次插入20 次删除
  1. 39 20
      automation/cleu-ansible-n9k/add_aaa.py

+ 39 - 20
automation/cleu-ansible-n9k/add_aaa.py

@@ -9,31 +9,50 @@ import os
 
 def main():
     parser = argparse.ArgumentParser(
-        prog=sys.argv[0], description='Add a VLAN to the core')
-    parser.add_argument('--key', metavar='<TACACS_KEY>',
-                        help='Clear text TACACS+ key', required=True)
-    parser.add_argument('--username', '-u', metavar='<USERNAME>',
-                        help='Username to use to connect to the N9Ks', required=True)
-    parser.add_argument('--site', '-s', metavar='<SITE NAME>',
-                        help='Name of site to which to add N9Ks (default: all sites)')
+        prog=sys.argv[0], description="Add a VLAN to the core"
+    )
+    parser.add_argument(
+        "--key", metavar="<TACACS_KEY>", help="Clear text TACACS+ key", required=True
+    )
+    parser.add_argument(
+        "--username",
+        "-u",
+        metavar="<USERNAME>",
+        help="Username to use to connect to the N9Ks",
+        required=True,
+    )
+    parser.add_argument(
+        "--site",
+        "-s",
+        metavar="<SITE NAME>",
+        help="Name of site to which to add N9Ks (default: all sites)",
+    )
     args = parser.parse_args()
 
-    os.environ['ANSIBLE_FORCE_COLOR'] = 'True'
-    os.environ['ANSIBLE_HOST_KEY_CHECKING'] = 'False'
-    os.environ['ANSIBLE_PERSISTENT_COMMAND_TIMEOUT'] = '300'
+    os.environ["ANSIBLE_FORCE_COLOR"] = "True"
+    os.environ["ANSIBLE_HOST_KEY_CHECKING"] = "False"
+    os.environ["ANSIBLE_PERSISTENT_COMMAND_TIMEOUT"] = "300"
 
-    command = ['ansible-playbook', '-i', 'inventory/hosts',
-               '-u', args.username, '-k', '-e',
-               'tacacs_key={}'.format(args.key), '-e', 'ansible_python_interpreter={}'.format(sys.executable),
-               'add-aaa-playbook.yml']
+    command = [
+        "ansible-playbook",
+        "-i",
+        "inventory/hosts",
+        "-u",
+        args.username,
+        "-k",
+        "-e",
+        "tacacs_key={}".format(args.key),
+        "-e",
+        "ansible_python_interpreter={}".format(sys.executable),
+        "add-aaa-playbook.yml",
+    ]
     if args.site:
-        command += ['--limit', args.site]
-    p = subprocess.Popen(command, stdout=subprocess.PIPE,
-                         stderr=subprocess.STDOUT)
-    for c in iter(lambda: p.stdout.read(1), b''):
-        sys.stdout.write(c.decode('utf-8'))
+        command += ["--limit", args.site]
+    p = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
+    for c in iter(lambda: p.stdout.read(1), b""):
+        sys.stdout.write(c.decode("utf-8"))
         sys.stdout.flush()
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()