Add several composite CLI commands for common multi-step workflows

awscli/customizations/athena/__init__.py

@@ -0,0 +1,21 @@
+# Copyright 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.
+from awscli.customizations.athena.run_query import RunQueryCommand
+
+
+def register(event_handlers):
+    event_handlers.register('building-command-table.athena', _inject_commands)
+
+
+def _inject_commands(command_table, session, **kwargs):
+    command_table['run-query'] = RunQueryCommand(session)