Command module for Microsoft.PolicyInsights

doc/sphinx/azhelpgen/doc_source_map.json

@@ -36,6 +36,7 @@
     "monitor": "src/command_modules/azure-cli-monitor/azure/cli/command_modules/monitor/_help.py",
     "network": "src/command_modules/azure-cli-network/azure/cli/command_modules/network/_help.py",
     "policy": "src/command_modules/azure-cli-resource/azure/cli/command_modules/resource/_help.py",
+    "policyinsights": "src/command_modules/azure-cli-policyinsights/azure/cli/command_modules/policyinsights/_help.py",
     "provider": "src/command_modules/azure-cli-resource/azure/cli/command_modules/resource/_help.py",
     "rdbms": "src/command_modules/azure-cli-rdbms/azure/cli/command_modules/rdbms/_help.py",
     "redis": "src/command_modules/azure-cli-redis/azure/cli/command_modules/redis/_help.py",

src/command_modules/azure-cli-policyinsights/HISTORY.rst

@@ -0,0 +1,9 @@
+.. :changelog:
+
+Release History
+===============
+
+0.1.0
+++++++++++++++++++
+
+* Initial Release

src/command_modules/azure-cli-policyinsights/MANIFEST.in

@@ -0,0 +1 @@
+include *.rst

src/command_modules/azure-cli-policyinsights/README.rst

@@ -0,0 +1,7 @@
+Microsoft Azure CLI 'policyinsights' Command Module
+===================================================
+
+This package is for the 'policyinsights' module.
+i.e. 'az policy event' and 'az policy state'
+
+

src/command_modules/azure-cli-policyinsights/azure/__init__.py

@@ -0,0 +1,7 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------------------------
+
+import pkg_resources
+pkg_resources.declare_namespace(__name__)

src/command_modules/azure-cli-policyinsights/azure/cli/__init__.py

@@ -0,0 +1,7 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------------------------
+
+import pkg_resources
+pkg_resources.declare_namespace(__name__)

src/command_modules/azure-cli-policyinsights/azure/cli/command_modules/__init__.py

@@ -0,0 +1,7 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------------------------
+
+import pkg_resources
+pkg_resources.declare_namespace(__name__)

src/command_modules/azure-cli-policyinsights/azure/cli/command_modules/policyinsights/__init__.py

@@ -0,0 +1,39 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------------------------
+
+# pylint: disable=unused-import
+
+from azure.cli.command_modules.policyinsights._help import helps
+from azure.cli.core import AzCommandsLoader
+
+
+class PolicyInsightsCommandsLoader(AzCommandsLoader):
+
+    def __init__(self, cli_ctx=None):
+        from azure.cli.core.commands import CliCommandType
+        from ._exception_handler import policy_insights_exception_handler
+
+        policyinsights_custom = CliCommandType(
+            operations_tmpl='azure.cli.command_modules.policyinsights.custom#{}',
+            exception_handler=policy_insights_exception_handler)
+
+        super(PolicyInsightsCommandsLoader, self).__init__(
+            cli_ctx=cli_ctx,
+            min_profile='2017-03-10-profile',
+            custom_command_type=policyinsights_custom)
+
+    def load_command_table(self, args):
+        from .commands import load_command_table
+
+        load_command_table(self, args)
+        return self.command_table
+
+    def load_arguments(self, command):
+        from ._params import load_arguments
+
+        load_arguments(self, command)
+
+
+COMMAND_LOADER_CLS = PolicyInsightsCommandsLoader

src/command_modules/azure-cli-policyinsights/azure/cli/command_modules/policyinsights/_client_factory.py

@@ -0,0 +1,19 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------------------------
+
+
+def _cf_policy_insights(cli_ctx, **_):
+    from azure.cli.core.commands.client_factory import get_mgmt_service_client
+    from azure.mgmt.policyinsights import PolicyInsightsClient
+
+    return get_mgmt_service_client(cli_ctx, PolicyInsightsClient, subscription_bound=False)
+
+
+def policy_events_operations(cli_ctx, _):
+    return _cf_policy_insights(cli_ctx).policy_events
+
+
+def policy_states_operations(cli_ctx, _):
+    return _cf_policy_insights(cli_ctx).policy_states

src/command_modules/azure-cli-policyinsights/azure/cli/command_modules/policyinsights/_exception_handler.py

@@ -0,0 +1,19 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------------------------
+
+from azure.cli.core.util import CLIError
+
+
+def policy_insights_exception_handler(ex):
+    from azure.mgmt.policyinsights.models import QueryFailureException
+
+    if isinstance(ex, QueryFailureException):
+        message = '({}) {}'.format(ex.error.error.code, ex.error.error.message)
+        raise CLIError(message)
+    else:
+        import sys
+        from six import reraise
+
+        reraise(*sys.exc_info())