Merge pull request #41 from Azure-Samples/t2-consumption-2020-10-28

[T2] consumption wave3

Author: 00Kai0

samples/consumption/README.md

@@ -0,0 +1,87 @@
+---
+page_type: sample
+languages:
+- python
+products:
+- azure
+description: "These code samples will show you how to manage consumption using Azure SDK for Python."
+urlFragment: consumption
+---
+
+# Getting started - Managing consumption using Azure Python SDK
+
+These code samples will show you how to manage consumption using Azure SDK for Python.
+
+## Features
+
+This project framework provides examples for the following services:
+
+### consumption
+* [] Using the Azure SDK for Python - consumption Management Library [azure-mgmt-consumption](https://pypi.org/project/azure-mgmt-consumption/) for the [consumption API](https://docs.microsoft.com/en-us/rest/api/consumption/)
+
+## Getting Started
+
+### Prerequisites
+
+1. Before we run the samples, we need to make sure we have setup the credentials. Follow the instructions in [register a new application using Azure portal](https://docs.microsoft.com/en-us/azure/active-directory/develop/howto-create-service-principal-portal) to obtain `subscription id`,`client id`,`client secret`, and `application id`
+
+2. Store your credentials an environment variables.
+For example, in Linux-based OS, you can do
+```bash
+export AZURE_TENANT_ID="xxx"
+export AZURE_CLIENT_ID="xxx"
+export AZURE_CLIENT_SECRET="xxx"
+export SUBSCRIPTION_ID="xxx"
+```
+
+### Installation
+
+1.  If you don't already have it, [install Python](https://www.python.org/downloads/).
+
+    This sample (and the SDK) is compatible with Python 2.7, 3.3, 3.4, 3.5 and 3.6.
+
+2.  General recommendation for Python development is to use a Virtual Environment.
+    For more information, see https://docs.python.org/3/tutorial/venv.html
+
+    Install and initialize the virtual environment with the "venv" module on Python 3 (you must install [virtualenv](https://pypi.python.org/pypi/virtualenv) for Python 2.7):
+
+    ```
+    python -m venv mytestenv # Might be "python3" or "py -3.6" depending on your Python installation
+    cd mytestenv
+    source bin/activate      # Linux shell (Bash, ZSH, etc.) only
+    ./scripts/activate       # PowerShell only
+    ./scripts/activate.bat   # Windows CMD only
+    ```
+
+### Quickstart
+
+1.  Clone the repository.
+
+    ```
+    git clone https://github.com/Azure-Samples/azure-samples-python-management.git
+    ```
+
+2.  Install the dependencies using pip.
+
+    ```
+    cd azure-samples-python-management/samples/consumption
+    pip install -r requirements.txt
+    ```
+
+## Demo
+
+A demo app is included to show how to use the project.
+
+To run the complete demo, execute `python example.py`
+
+To run each individual demo, point directly to the file. For example (i.e. not complete list):
+
+1. `python manage_consumption.py`
+
+If the script starts with `disable_***.py`, it means that it is unavailable now.
+
+The sample files do not have dependency each other and each file represents an individual end-to-end scenario. Please look at the sample that contains the scenario you are interested in
+
+## Resources
+
+- https://github.com/Azure/azure-sdk-for-python

samples/consumption/manage_consumption.py

@@ -0,0 +1,84 @@
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+# --------------------------------------------------------------------------
+
+import os
+import time
+from azure.identity import DefaultAzureCredential
+from azure.mgmt.consumption import ConsumptionManagementClient
+from azure.mgmt.resource import ResourceManagementClient
+
+# - other dependence -
+# - end -
+
+
+def main():
+
+    SUBSCRIPTION_ID = os.environ.get("SUBSCRIPTION_ID", None)
+    TIME = str(time.time()).replace('.','')
+    GROUP_NAME = "testconsumption" + TIME
+    CONSUMPTION = "consumption" + TIME
+
+    # Create client
+    # # For other authentication approaches, please see: https://pypi.org/project/azure-identity/
+    resource_client = ResourceManagementClient(
+        credential=DefaultAzureCredential(),
+        subscription_id=SUBSCRIPTION_ID
+    )
+    consumption_client = ConsumptionManagementClient(
+        credential=DefaultAzureCredential(),
+        subscription_id=SUBSCRIPTION_ID
+    )
+    # - init depended client -
+    # - end -
+
+    # Create resource group
+    resource_client.resource_groups.create_or_update(
+        GROUP_NAME,
+        {"location": "eastus"}
+    )
+
+    # - init depended resources -
+    # - end -
+
+    # Create consumption
+    SCOPE = '/subscriptions/{}/resourceGroups/{}'.format(SUBSCRIPTION_ID, GROUP_NAME)
+    consumption = consumption_client.budgets.create_or_update(
+        SCOPE,
+        CONSUMPTION,
+        {
+            "category": "Cost",
+            "amount": '100',
+            "timeGrain": "Monthly",
+            "timePeriod": {
+                "startDate": "2020-10-01T00:00:00Z",
+                "endDate": "2021-10-31T00:00:00Z"
+            }
+        }
+    )
+    print("Create consumption:\n{}\n".format(consumption))
+
+    # Get consumption
+    consumption = consumption_client.budgets.get(
+        SCOPE,
+        CONSUMPTION
+    )
+    print("Get consumption:\n{}\n".format(consumption))
+
+    # Delete consumption
+    consumption_client.budgets.delete(
+        SCOPE,
+        CONSUMPTION
+    )
+    print("Delete consumption.\n")
+
+    # Delete Group
+    resource_client.resource_groups.begin_delete(
+        GROUP_NAME
+    ).result()
+
+
+if __name__ == "__main__":
+    main()

samples/consumption/requirements.txt

@@ -0,0 +1,3 @@
+azure-identity
+azure-mgmt-resource==15.0.0
+azure-mgmt-consumption==8.0.0b1