sql: knack conversion (Azure#5027)

timwarner-msft · Dec 4, 2017 · baafddf · baafddf
1 parent 9656401
commit baafddf
Show file tree

Hide file tree

Showing 26 changed files with 11,833 additions and 9,301 deletions.
diff --git a/azure-cli.pyproj b/azure-cli.pyproj
@@ -524,6 +524,9 @@
     <Compile Include="command_modules\azure-cli-sql\azure\cli\command_modules\sql\_help.py" />
     <Compile Include="command_modules\azure-cli-sql\azure\cli\command_modules\sql\_params.py" />
     <Compile Include="command_modules\azure-cli-sql\azure\cli\command_modules\sql\_util.py" />
+    <Compile Include="command_modules\azure-cli-sql\azure\cli\command_modules\sql\_validators.py">
+      <SubType>Code</SubType>
+    </Compile>
     <Compile Include="command_modules\azure-cli-sql\azure\cli\command_modules\sql\__init__.py" />
     <Compile Include="command_modules\azure-cli-sql\azure_bdist_wheel.py" />
     <Compile Include="command_modules\azure-cli-sql\setup.py" />

diff --git a/src/command_modules/azure-cli-sql/azure/cli/command_modules/sql/__init__.py b/src/command_modules/azure-cli-sql/azure/cli/command_modules/sql/__init__.py
@@ -3,14 +3,27 @@
 # Licensed under the MIT License. See License.txt in the project root for license information.
 # --------------------------------------------------------------------------------------------
 
+from azure.cli.core import AzCommandsLoader
+
 import azure.cli.command_modules.sql._help  # pylint: disable=unused-import
 
-__all__ = ['load_params', 'load_commands']
 
+class SqlCommandsLoader(AzCommandsLoader):
+
+    def __init__(self, cli_ctx=None):
+        from azure.cli.core.commands import CliCommandType
+        sql_custom = CliCommandType(operations_tmpl='azure.cli.command_modules.sql.custom#{}')
+        super(SqlCommandsLoader, self).__init__(cli_ctx=cli_ctx,
+                                                custom_command_type=sql_custom)
+
+    def load_command_table(self, args):
+        from azure.cli.command_modules.sql.commands import load_command_table
+        load_command_table(self, args)
+        return self.command_table
 
-def load_params(_):
-    import azure.cli.command_modules.sql._params  # pylint: disable=redefined-outer-name, unused-variable
+    def load_arguments(self, command):
+        from azure.cli.command_modules.sql._params import load_arguments
+        load_arguments(self, command)
 
 
-def load_commands():
-    import azure.cli.command_modules.sql.commands  # pylint: disable=redefined-outer-name, unused-variable
+COMMAND_LOADER_CLS = SqlCommandsLoader