[AMS] Update AMS Module to 9.0 Python sdk (#22056)

* updated ams cli module to latest 9.0 python sdk

* updated tests

* updating test cases

Author: hivyas

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

@@ -85,3 +85,7 @@ def get_account_filters_client(cli_ctx, *_):
 
 def get_asset_filters_client(cli_ctx, *_):
     return cf_media(cli_ctx).asset_filters
+
+
+def get_asset_tracks_client(cli_ctx, *_):
+    return cf_media(cli_ctx).tracks

src/azure-cli/azure/cli/command_modules/ams/_help.py

@@ -241,6 +241,31 @@
     short-summary: Manage account filters for an Azure Media Services account.
 """
 
+helps['ams asset-track'] = """
+    type: group
+    short-summary: Manage asset tracks for an Azure Media Services account.
+"""
+
+helps['ams asset-track show'] = """
+    type: command
+    short-summary: Show the details of a track.
+"""
+
+helps['ams asset-track create'] = """
+    type: command
+    short-summary: Create a track for an Azure Media Services asset.
+"""
+
+helps['ams asset-track update'] = """
+    type: command
+    short-summary: Update the parameters of a track.
+"""
+
+helps['ams asset-track update-data'] = """
+    type: command
+    short-summary: Update a track if the file in the storage container was recently modified.
+"""
+
 helps['ams asset show'] = """
     type: command
     short-summary: Show the details of an asset.
@@ -473,6 +498,11 @@
     short-summary: Manage streaming endpoints for an Azure Media Service account.
 """
 
+helps['ams streaming-endpoint get-skus'] = """
+    type: command
+    short-summary: Get the sku details for a streaming endpoint.
+"""
+
 helps['ams streaming-endpoint start'] = """
     type: command
     short-summary: Start a streaming endpoint.

src/azure-cli/azure/cli/command_modules/ams/_params.py

@@ -46,7 +46,7 @@
 
 
 def load_arguments(self, _):  # pylint: disable=too-many-locals, too-many-statements
-    name_arg_type = CLIArgumentType(options_list=['--name', '-n'], id_part='name', help='The name of the Azure Media Services account.', metavar='NAME')
+    name_arg_type = CLIArgumentType(options_list=['--name', '-n'], id_part='name', help='The name of the resource.', metavar='NAME')
     account_name_arg_type = CLIArgumentType(options_list=['--account-name', '-a'], id_part='name', help='The name of the Azure Media Services account.', metavar='ACCOUNT_NAME')
     storage_account_arg_type = CLIArgumentType(options_list=['--storage-account'], validator=validate_storage_account_id, metavar='STORAGE_NAME')
     password_arg_type = CLIArgumentType(options_list=['--password', '-p'], metavar='PASSWORD_NAME')
@@ -203,6 +203,30 @@ def load_arguments(self, _):  # pylint: disable=too-many-locals, too-many-statem
     with self.argument_context('ams asset-filter list') as c:
         c.argument('account_name', id_part=None)
 
+    with self.argument_context('ams asset-track') as c:
+        c.argument('account_name', account_name_arg_type)
+        c.argument('asset_name', help='The asset name')
+        c.argument('track_name', help='The name of the track.')
+
+    with self.argument_context('ams asset-track create') as c:
+        c.argument('track_type', help='The type of track. Allowed values: Text.')
+        c.argument('file_name', help='The name of the file.'
+                                     ' Note: this file should already be uploaded to the storage container.', arg_group='Text Track')
+        c.argument('display_name', help='The display name of the text track on a video player.'
+                                        ' In HLS, this maps to the NAME attribute of EXT-X-MEDIA.', arg_group='Text Track')
+        c.argument('language_code', arg_group='Text Track', help='The RFC5646 language code for the text track.')
+        c.argument('player_visibility', arg_group='Text Track', help='When PlayerVisibility is set to "Visible", the text track will be present in the DASH manifest or HLS playlist when requested by a client. When the PlayerVisibility is set to "Hidden", the text will not be available to the client. The default value is "Visible". Possible values include: "Hidden", "Visible".')
+
+    with self.argument_context('ams asset-track list') as c:
+        c.argument('account_name', id_part=None)
+
+    with self.argument_context('ams asset-track update') as c:
+        c.argument('track_type', help='The type of track. Allowed values: Audio, Text, Video.')
+        c.argument('display_name', arg_group='Text Track', help='The display name of the text track on a video player.'
+                   ' In HLS, this maps to the NAME attribute of EXT-X-MEDIA.')
+        c.argument('language_code', arg_group='Text Track', help='The RFC5646 language code for the text track.')
+        c.argument('player_visibility', arg_group='Text Track', help='When PlayerVisibility is set to "Visible", the text track will be present in the DASH manifest or HLS playlist when requested by a client. When the PlayerVisibility is set to "Hidden", the text will not be available to the client. The default value is "Visible". Possible values include: "Hidden", "Visible".')
+
     with self.argument_context('ams job') as c:
         c.argument('account_name', account_name_arg_type)
         c.argument('transform_name', transform_name_arg_type, id_part='child_name_1',

src/azure-cli/azure/cli/command_modules/ams/commands.py

@@ -9,7 +9,7 @@
                               get_streaming_policies_client, get_streaming_endpoints_client,
                               get_locations_client, get_live_events_client, get_live_outputs_client,
                               get_content_key_policies_client, get_asset_filters_client,
-                              get_account_filters_client)
+                              get_account_filters_client, get_asset_tracks_client)
 from ._exception_handler import ams_exception_handler
 
 
@@ -127,6 +127,21 @@ def get_custom_sdk(custom_module, client_factory):
                                  custom_func_name='update_asset_filter',
                                  custom_func_type=get_custom_sdk('asset_filter', get_mediaservices_client))
 
+    with self.command_group('ams asset-track', get_sdk('Tracks', get_asset_tracks_client)) as g:
+        g.command('list', 'list')
+        g.custom_show_command('show', 'get_asset_track',
+                              custom_command_type=get_custom_sdk('tracks', get_asset_tracks_client))
+        g.command('delete', 'begin_delete')
+        g.custom_command('create', 'create_asset_track',
+                         custom_command_type=get_custom_sdk('tracks', get_asset_tracks_client),
+                         supports_no_wait=True)
+        g.custom_command('update', 'update_track',
+                         custom_command_type=get_custom_sdk('tracks', get_asset_tracks_client),
+                         supports_no_wait=True)
+        g.custom_command('update-data', 'update_track_data',
+                         custom_command_type=get_custom_sdk('tracks', get_asset_tracks_client),
+                         supports_no_wait=True)
+
     with self.command_group('ams job', get_sdk('Jobs', get_jobs_client)) as g:
         g.custom_show_command('show', 'get_job',
                               custom_command_type=get_custom_sdk('job', get_jobs_client))
@@ -205,6 +220,8 @@ def get_custom_sdk(custom_module, client_factory):
                               custom_command_type=get_custom_sdk('streaming_endpoint', get_streaming_endpoints_client))
         g.command('delete', 'begin_delete')
         g.wait_command('wait')
+        g.custom_command('get-skus', 'get_skus',
+                         custom_command_type=get_custom_sdk('streaming_endpoint', get_streaming_endpoints_client))
 
     with self.command_group('ams streaming-endpoint akamai', get_sdk('StreamingEndpoints', get_streaming_endpoints_client)) as g:
         g.custom_command('add', 'add_akamai_access_control',

src/azure-cli/azure/cli/command_modules/ams/operations/streaming_endpoint.py

@@ -213,3 +213,7 @@ def get_streaming_endpoint(client, resource_group_name, account_name,
             streaming_endpoint_name)
 
     return streaming_endpoint
+
+
+def get_skus(client, resource_group_name, account_name, streaming_endpoint_name):
+    return client.skus(resource_group_name, account_name, streaming_endpoint_name)

src/azure-cli/azure/cli/command_modules/ams/operations/tracks.py

@@ -0,0 +1,47 @@
+# --------------------------------------------------------------------------------------------
+# 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, sdk_no_wait
+from azure.mgmt.media.models import (AssetTrack, TextTrack)
+
+
+def get_asset_track(client, account_name, resource_group_name, asset_name, track_name):
+    return client.get(resource_group_name, account_name, asset_name, track_name)
+
+
+def create_asset_track(client, account_name, resource_group_name, asset_name, track_name, track_type, file_name=None,
+                       display_name=None, player_visibility=None, language_code=None, no_wait=False):
+    track = None
+    if track_type == 'Text':
+        track = TextTrack(file_name=file_name, display_name=display_name, player_visibility=player_visibility,
+                          language_code=language_code)
+    else:
+        raise CLIError("'{}' is not a valid track type. Currently supported track type is Text.".format(track_type))
+
+    return sdk_no_wait(no_wait, client.begin_create_or_update, resource_group_name=resource_group_name,
+                       account_name=account_name, asset_name=asset_name, track_name=track_name,
+                       parameters=AssetTrack(track=track))
+
+
+def update_track_data(client, account_name, resource_group_name, asset_name, track_name, no_wait=False):
+    return sdk_no_wait(no_wait, client.begin_update_track_data, resource_group_name=resource_group_name,
+                       account_name=account_name, asset_name=asset_name, track_name=track_name)
+
+
+def update_track(client, account_name, resource_group_name, asset_name, track_name,
+                 display_name=None, player_visibility=None, language_code=None, no_wait=False):
+    current_track = client.get(resource_group_name, account_name, asset_name, track_name)
+
+    if display_name:
+        current_track.track.display_name = display_name
+    if player_visibility:
+        current_track.track.player_visibility = player_visibility
+    if language_code:
+        current_track.track.language_code = language_code
+
+    return sdk_no_wait(no_wait, client.begin_update, resource_group_name=resource_group_name,
+                       account_name=account_name, asset_name=asset_name, track_name=track_name,
+                       parameters=current_track)

src/azure-cli/azure/cli/command_modules/ams/tests/latest/data/assetTrack.ttml

@@ -0,0 +1,18 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!--
+Generated by LyricPotato
+http://lyricpotato.com
+DKID: 3475014.2FDD1D176991032F79DC2DB683187859
+-->
+<tt xmlns="http://www.w3.org/ns/ttml"
+xmlns:tts="http://www.w3.org/ns/ttml#styling"
+xmlns:ttm="http://www.w3.org/ns/ttml#metadata"
+xml:lang="en">
+<head>
+</head>
+<body>
+<div>
+<p begin="00:00:00.000" end="00:00:04.838">hi this is a song</p>
+</div>
+</body>
+</tt>

src/azure-cli/azure/cli/command_modules/ams/tests/latest/data/sampleIsmFile.ism

@@ -0,0 +1,59 @@
+<?xml version="1.0" encoding="utf-8"?>
+<smil xmlns="http://www.w3.org/2001/SMIL20/Language">
+  <head>
+    <meta name="clientManifestRelativePath" content="Web capture_12-4-2022_141014_ms..ismc" />
+    <meta name="formats" content="mp4-v3" />
+  </head>
+  <body>
+    <switch>
+      <video src="Web capture_12-4-2022_141014_ms._1920x892_6000.mp4" systemBitrate="12520000">
+        <param name="systemBitrate" value="12520000" valuetype="data" />
+        <param name="trackID" value="1" valuetype="data" />
+        <param name="trackName" value="video" valuetype="data" />
+        <param name="trackIndex" value="Web capture_12-4-2022_141014_ms._1920x892_6000_1.mpi" valuetype="data" />
+      </video>
+      <video src="Web capture_12-4-2022_141014_ms._1920x892_4700.mp4" systemBitrate="10658000">
+        <param name="systemBitrate" value="10658000" valuetype="data" />
+        <param name="trackID" value="1" valuetype="data" />
+        <param name="trackName" value="video" valuetype="data" />
+        <param name="trackIndex" value="Web capture_12-4-2022_141014_ms._1920x892_4700_1.mpi" valuetype="data" />
+      </video>
+      <video src="Web capture_12-4-2022_141014_ms._1280x596_3400.mp4" systemBitrate="6626600">
+        <param name="systemBitrate" value="6626600" valuetype="data" />
+        <param name="trackID" value="1" valuetype="data" />
+        <param name="trackName" value="video" valuetype="data" />
+        <param name="trackIndex" value="Web capture_12-4-2022_141014_ms._1280x596_3400_1.mpi" valuetype="data" />
+      </video>
+      <video src="Web capture_12-4-2022_141014_ms._960x448_2250.mp4" systemBitrate="4001800">
+        <param name="systemBitrate" value="4001800" valuetype="data" />
+        <param name="trackID" value="1" valuetype="data" />
+        <param name="trackName" value="video" valuetype="data" />
+        <param name="trackIndex" value="Web capture_12-4-2022_141014_ms._960x448_2250_1.mpi" valuetype="data" />
+      </video>
+      <video src="Web capture_12-4-2022_141014_ms._960x448_1500.mp4" systemBitrate="3150200">
+        <param name="systemBitrate" value="3150200" valuetype="data" />
+        <param name="trackID" value="1" valuetype="data" />
+        <param name="trackName" value="video" valuetype="data" />
+        <param name="trackIndex" value="Web capture_12-4-2022_141014_ms._960x448_1500_1.mpi" valuetype="data" />
+      </video>
+      <video src="Web capture_12-4-2022_141014_ms._640x300_1000.mp4" systemBitrate="1580800">
+        <param name="systemBitrate" value="1580800" valuetype="data" />
+        <param name="trackID" value="1" valuetype="data" />
+        <param name="trackName" value="video" valuetype="data" />
+        <param name="trackIndex" value="Web capture_12-4-2022_141014_ms._640x300_1000_1.mpi" valuetype="data" />
+      </video>
+      <video src="Web capture_12-4-2022_141014_ms._640x300_650.mp4" systemBitrate="1139400">
+        <param name="systemBitrate" value="1139400" valuetype="data" />
+        <param name="trackID" value="1" valuetype="data" />
+        <param name="trackName" value="video" valuetype="data" />
+        <param name="trackIndex" value="Web capture_12-4-2022_141014_ms._640x300_650_1.mpi" valuetype="data" />
+      </video>
+      <video src="Web capture_12-4-2022_141014_ms._320x152_400.mp4" systemBitrate="675800">
+        <param name="systemBitrate" value="675800" valuetype="data" />
+        <param name="trackID" value="1" valuetype="data" />
+        <param name="trackName" value="video" valuetype="data" />
+        <param name="trackIndex" value="Web capture_12-4-2022_141014_ms._320x152_400_1.mpi" valuetype="data" />
+      </video>
+    </switch>
+  </body>
+</smil>