Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
The table of contents is too big for display.
Diff view
Diff view
  •  
  •  
  •  
76 changes: 1 addition & 75 deletions src/azure-cli/azure/cli/command_modules/network/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,6 @@
# --------------------------------------------------------------------------------------------

from azure.cli.core import AzCommandsLoader
from azure.cli.core.profiles import ResourceType

import azure.cli.command_modules.network._help # pylint: disable=unused-import

Expand Down Expand Up @@ -51,77 +50,4 @@ def load_arguments(self, command):
load_arguments(self, command)


class AzureStackNetworkCommandsLoader(AzCommandsLoader):

def __init__(self, cli_ctx=None):
from azure.cli.core import ModExtensionSuppress
from azure.cli.core.commands import CliCommandType
network_custom = CliCommandType(operations_tmpl='azure.cli.command_modules.network.azure_stack.custom#{}')
super().__init__(cli_ctx=cli_ctx,
resource_type=ResourceType.MGMT_NETWORK,
custom_command_type=network_custom,
suppress_extension=[
ModExtensionSuppress(__name__, 'dns', '0.0.2',
reason='These commands are now in the CLI.',
recommend_remove=True),
ModExtensionSuppress(__name__, 'express-route', '0.1.3',
reason='These commands are now in the CLI.',
recommend_remove=True)
])

def load_command_table(self, args):
from azure.cli.core.aaz import load_aaz_command_table
from azure.cli.command_modules.network.azure_stack.commands import load_command_table
try:
from . import aaz
except ImportError:
aaz = None

if aaz:
profile = self.cli_ctx.cloud.profile
try:
if profile.lower() == "2019-03-01-hybrid":
# use aaz in 2018-03-01-hybrid profile, because apis are the some.
self.cli_ctx.cloud.profile = "2018-03-01-hybrid"
load_aaz_command_table(
loader=self,
aaz_pkg_name=aaz.__name__,
args=args
)
finally:
self.cli_ctx.cloud.profile = profile

load_command_table(self, args)

profile = self.get_module_by_profile("commands")
if profile and hasattr(profile, 'load_command_table'):
profile.load_command_table(self, args)

return self.command_table

def load_arguments(self, command):
from azure.cli.command_modules.network.azure_stack._params import load_arguments
load_arguments(self, command)

profile = self.get_module_by_profile("_params")
if profile and hasattr(profile, 'load_arguments'):
profile.load_arguments(self, command)

def get_module_name_by_profile(self, module_name):
from azure.cli.core.aaz.utils import get_aaz_profile_module_name
profile_module_name = get_aaz_profile_module_name(profile_name=self.cli_ctx.cloud.profile)
if module_name:
return f'azure.cli.command_modules.network.azure_stack.{profile_module_name}.{module_name}'
return f'azure.cli.command_modules.network.azure_stack.{profile_module_name}'

def get_module_by_profile(self, name):
import importlib
module_name = self.get_module_name_by_profile(name)
return importlib.import_module(module_name)


def get_command_loader(cli_ctx):
if cli_ctx.cloud.profile.lower() != "latest":
return AzureStackNetworkCommandsLoader

return NetworkCommandsLoader
COMMAND_LOADER_CLS = NetworkCommandsLoader

This file was deleted.

This file was deleted.

This file was deleted.

This file was deleted.

This file was deleted.

This file was deleted.

Loading
Loading