Moving rstgen here since it is quite specific to the CLI.

garnaat · garnaat · commit 71be459d3ddd · 2013-05-02T15:44:49.000-07:00
diff --git a/awscli/rstgen.py b/awscli/rstgen.py
@@ -0,0 +1,53 @@
+import os
+import sys
+import argparse
+from .clidriver import CLIDriver
+from bcdoc.mangen import gen_man
+
+
+def do_operation(session, args):
+    service = session.get_service(args.service)
+    operation = service.get_operation(args.operation)
+    gen_man(session, operation=operation)
+
+
+def do_service(session, args):
+    service = session.get_service(args.service)
+    gen_man(session, service=service)
+
+
+def do_provider(session, args):
+    cli_data = get_cli_data(session, provider_name=args.provider)
+    gen_man(session, provider=args.provider, cli_data=cli_data)
+
+
+def get_cli_data(session, provider_name):
+    cli_data = session.get_data('cli')
+    for option in cli_data['options']:
+        if option.startswith('--'):
+            option_data = cli_data['options'][option]
+            if 'choices' in option_data:
+                choices = option_data['choices']
+                if not isinstance(choices, list):
+                    choices_path = choices.format(provider=provider_name)
+                    choices = session.get_data(choices_path)
+                    option_data['choices'] = choices
+    return cli_data
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--provider',
+                        help='Name of provider', required=True)
+    parser.add_argument('--service',
+                        help='Name of service')
+    parser.add_argument('--operation',
+                        help='Name of operation')
+    args = parser.parse_args()
+    driver = CLIDriver()
+    if args.provider and args.service and args.operation:
+        do_operation(driver.session, args)
+    elif args.provider and args.service:
+        do_service(driver.session, args)
+    else:
+        do_provider(driver.session, args)
diff --git a/bin/rstgen b/bin/rstgen
@@ -0,0 +1,18 @@
+#!/usr/bin/env python
+# Copyright 2012 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+
+# Licensed under the Apache License, Version 2.0 (the "License"). You
+# may not use this file except in compliance with the License. A copy of
+# the License is located at
+
+#     http://aws.amazon.com/apache2.0/
+
+# or in the "license" file accompanying this file. This file is
+# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
+# ANY KIND, either express or implied. See the License for the specific
+# language governing permissions and limitations under the License.
+import sys
+import awscli.rstgen
+
+if __name__ == '__main__':
+    sys.exit(awscli.rstgen.main())
diff --git a/bin/rstgen.cmd b/bin/rstgen.cmd
@@ -0,0 +1,52 @@
+@echo OFF
+REM="""
+setlocal
+set PythonExe=
+set PythonExeFlags=
+
+for %%i in (cmd bat exe) do (
+    for %%j in (python.%%i) do (
+        call :SetPythonExe "%%~$PATH:j"
+    )
+)
+for /f "tokens=2 delims==" %%i in ('assoc .py') do (
+    for /f "tokens=2 delims==" %%j in ('ftype %%i') do (
+        for /f "tokens=1" %%k in ("%%j") do (
+            call :SetPythonExe %%k
+        )
+    )
+)
+"%PythonExe%" -x %PythonExeFlags% "%~f0" %*
+goto :EOF
+
+:SetPythonExe
+if not [%1]==[""] (
+    if ["%PythonExe%"]==[""] (
+        set PythonExe=%~1
+    )
+)
+goto :EOF
+"""
+
+# ===================================================
+# Python script starts here
+# ===================================================
+#!/usr/bin/env python
+# Copyright 2012 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+
+# Licensed under the Apache License, Version 2.0 (the "License"). You
+# may not use this file except in compliance with the License. A copy of
+# the License is located at
+
+#     http://aws.amazon.com/apache2.0/
+
+# or in the "license" file accompanying this file. This file is
+# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
+# ANY KIND, either express or implied. See the License for the specific
+# language governing permissions and limitations under the License.
+
+import sys
+import awscli.rstgen
+
+if __name__ == '__main__':
+    sys.exit(awscli.rstgen.main())
diff --git a/setup.py b/setup.py
@@ -36,7 +36,8 @@
     author_email='garnaat@amazon.com',
     url='http://aws.amazon.com/cli/',
     scripts=['bin/aws', 'bin/aws.cmd',
-             'bin/aws_completer', 'bin/aws_zsh_completer.sh'],
+             'bin/aws_completer', 'bin/aws_zsh_completer.sh',
+             'bin/rstgen', 'bin/rstgen.cmd'],
     packages=packages,
     package_dir={'awscli': 'awscli'},
     install_requires=requires,
