{Notification-hub} Migrate notification-hub to AAZ

src/notification-hub/HISTORY.rst

@@ -3,7 +3,12 @@
 Release History
 ===============
 
+1.0.0a1
+++++++
+* Migrate command to AAZ
+
 0.2.0
+++++++
 * Ignore --sku in notification hub as sku can only be set at namespace level
 * Drop Python 3.5
 * Use Experimental tag

src/notification-hub/azext_notification_hub/__init__.py

@@ -16,11 +16,22 @@ def __init__(self, cli_ctx=None):
         notificationhubs_custom = CliCommandType(
             operations_tmpl='azext_notification_hub.custom#{}',
             client_factory=cf_notificationhubs)
-        super(NotificationHubsCommandsLoader, self).__init__(cli_ctx=cli_ctx,
-                                                             custom_command_type=notificationhubs_custom)
+        super().__init__(cli_ctx=cli_ctx,
+                         custom_command_type=notificationhubs_custom)
 
     def load_command_table(self, args):
         from azext_notification_hub.commands import load_command_table
+        from azure.cli.core.aaz import load_aaz_command_table
+        try:
+            from . import aaz
+        except ImportError:
+            aaz = None
+        if aaz:
+            load_aaz_command_table(
+                loader=self,
+                aaz_pkg_name=aaz.__name__,
+                args=args
+            )
         load_command_table(self, args)
         return self.command_table
 

src/notification-hub/azext_notification_hub/_client_factory.py

@@ -10,13 +10,5 @@ def cf_notificationhubs(cli_ctx, *_):
     return get_mgmt_service_client(cli_ctx, NotificationHubsManagementClient)
 
 
-def cf_operations(cli_ctx, *_):
-    return cf_notificationhubs(cli_ctx).operations
-
-
-def cf_namespaces(cli_ctx, *_):
-    return cf_notificationhubs(cli_ctx).namespaces
-
-
 def cf_notification_hubs(cli_ctx, *_):
     return cf_notificationhubs(cli_ctx).notification_hubs