Skip to content

Commit

Permalink
rename with AcatMgmtClient
Browse files Browse the repository at this point in the history
  • Loading branch information
HuiquanJiang-ms committed Jul 10, 2024
1 parent b602f6e commit e193022
Showing 1 changed file with 16 additions and 16 deletions.
32 changes: 16 additions & 16 deletions src/acat/azext_acat/custom.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,8 +52,8 @@
logger = get_logger(__name__)


@register_client("DupAadTokenClient")
class AAZDupAadTokenClient(AAZMgmtClient):
@register_client("AcatMgmtClient")
class AAZAcatMgmtClient(AAZMgmtClient):
@classmethod
def _build_configuration(cls, ctx, credential, **kwargs):
from azure.cli.core.auth.util import resource_to_scopes
Expand All @@ -75,7 +75,7 @@ def _build_configuration(cls, ctx, credential, **kwargs):

class OnboardAcat(_AcatOnboard):
class OnboardAcatWithDupAadToken(_AcatOnboard.ProviderActionsOnboard):
CLIENT_TYPE = "DupAadTokenClient"
CLIENT_TYPE = "AcatMgmtClient"

def _execute_operations(self):
self.pre_operations()
Expand All @@ -88,7 +88,7 @@ class TriggerEvaluation(_AcatTriggerEvaluation):
class TriggerEvaluationWithDupAadToken(
_AcatTriggerEvaluation.ProviderActionsTriggerEvaluation
):
CLIENT_TYPE = "DupAadTokenClient"
CLIENT_TYPE = "AcatMgmtClient"

def _execute_operations(self):
self.pre_operations()
Expand All @@ -98,7 +98,7 @@ def _execute_operations(self):

class ListAcatReport(_AcatListReport):
class ListAcatReportWithDupAadToken(_AcatListReport.ReportList):
CLIENT_TYPE = "DupAadTokenClient"
CLIENT_TYPE = "AcatMgmtClient"

def _execute_operations(self):
self.pre_operations()
Expand All @@ -108,7 +108,7 @@ def _execute_operations(self):

class ShowAcatReport(_AcatShowReport):
class ShowAcatReportWithDupAadToken(_AcatShowReport.ReportGet):
CLIENT_TYPE = "DupAadTokenClient"
CLIENT_TYPE = "AcatMgmtClient"

def _execute_operations(self):
self.pre_operations()
Expand All @@ -118,7 +118,7 @@ def _execute_operations(self):

class CreateAcatReport(_AcatCreateReport):
class CreateAcatReportWithDupAadToken(_AcatCreateReport.ReportCreateOrUpdate):
CLIENT_TYPE = "DupAadTokenClient"
CLIENT_TYPE = "AcatMgmtClient"

def _execute_operations(self):
self.pre_operations()
Expand Down Expand Up @@ -167,7 +167,7 @@ def pre_operations(self):

class UpdateAcatReport(_AcatUpdateReport):
class UpdateAcatReportWithDupAadToken(_AcatUpdateReport.ReportUpdate):
CLIENT_TYPE = "DupAadTokenClient"
CLIENT_TYPE = "AcatMgmtClient"

def _execute_operations(self):
self.pre_operations()
Expand All @@ -177,7 +177,7 @@ def _execute_operations(self):

class DeleteAcatReport(_AcatDeleteReport):
class DeleteAcatReportWithDupAadToken(_AcatDeleteReport.ReportDelete):
CLIENT_TYPE = "DupAadTokenClient"
CLIENT_TYPE = "AcatMgmtClient"

def _execute_operations(self):
self.pre_operations()
Expand All @@ -187,7 +187,7 @@ def _execute_operations(self):

class DownloadAcatReport(_AcatDownloadSnapshot):
class DownloadAcatReportWithDupAadToken(_AcatDownloadSnapshot.SnapshotDownload):
CLIENT_TYPE = "DupAadTokenClient"
CLIENT_TYPE = "AcatMgmtClient"

def _execute_operations(self):
self.pre_operations()
Expand Down Expand Up @@ -254,7 +254,7 @@ def pre_operations(self):

class GetControlAssessment(_AcatListSnapshot):
class GetControlAssessmentWithDupAadToken(_AcatListSnapshot.SnapshotList):
CLIENT_TYPE = "DupAadTokenClient"
CLIENT_TYPE = "AcatMgmtClient"

def _execute_operations(self):
self.pre_operations()
Expand Down Expand Up @@ -315,7 +315,7 @@ def _output(self, *args, **kwargs):

class ListAcatReportWebhook(_AcatListReportWebhook):
class ListAcatReportWebhookWithDupAadToken(_AcatListReportWebhook.WebhookList):
CLIENT_TYPE = "DupAadTokenClient"
CLIENT_TYPE = "AcatMgmtClient"

def _execute_operations(self):
self.pre_operations()
Expand All @@ -325,7 +325,7 @@ def _execute_operations(self):

class ShowAcatReportWebhook(_AcatShowReportWebhook):
class ShowAcatReportWebhookWithDupAadToken(_AcatShowReportWebhook.WebhookGet):
CLIENT_TYPE = "DupAadTokenClient"
CLIENT_TYPE = "AcatMgmtClient"

def _execute_operations(self):
self.pre_operations()
Expand All @@ -337,7 +337,7 @@ class CreateAcatReportWebhook(_AcatCreateReportWebhook):
class CreateAcatReportWebhookWithDupAadToken(
_AcatCreateReportWebhook.WebhookCreateOrUpdate
):
CLIENT_TYPE = "DupAadTokenClient"
CLIENT_TYPE = "AcatMgmtClient"

def _execute_operations(self):
self.pre_operations()
Expand Down Expand Up @@ -437,7 +437,7 @@ class UpdateAcatReportWebhook(_AcatUpdateReportWebhook):
class UpdateAcatReportWebhookWithDupAadToken(
_AcatUpdateReportWebhook.WebhookUpdate
):
CLIENT_TYPE = "DupAadTokenClient"
CLIENT_TYPE = "AcatMgmtClient"

def _execute_operations(self):
self.pre_operations()
Expand Down Expand Up @@ -530,7 +530,7 @@ class DeleteAcatReportWebhook(_AcatDeleteReportWebhook):
class DeleteAcatReportWebhookWithDupAadToken(
_AcatDeleteReportWebhook.WebhookDelete
):
CLIENT_TYPE = "DupAadTokenClient"
CLIENT_TYPE = "AcatMgmtClient"

def _execute_operations(self):
self.pre_operations()
Expand Down

0 comments on commit e193022

Please sign in to comment.