Update odltools

[netvirt.git] / resources / tools / odltools / odltools / mdsal / cli.py

diff --git a/resources/tools/odltools/odltools/mdsal/cli.py b/resources/tools/odltools/odltools/mdsal/cli.py
index d36267e749f84dd65dd6e66aec34e6e3600b0b13..67611c66c7389d5304af52529e42c7c9e4d7d1ed 100644 (file)
--- a/resources/tools/odltools/odltools/mdsal/cli.py
+++ b/resources/tools/odltools/odltools/mdsal/cli.py
@@ -1,27 +1,34 @@
-import cmd
+# Copyright 2018 Red Hat, Inc. and others. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
 
+from odltools.cli_utils import add_common_args
+from odltools.mdsal.models import models
 
-def add_dump_parser(parsers):
-    parser = parsers.add_parser("dump", description="Get and write all mdsal models")
+
+def add_get_parser(parsers):
+    parser = parsers.add_parser("get", help="Get and write all mdsal models")
+    add_common_args(parser)
     parser.add_argument("path",
                         help="the directory that the parsed data is written into")
-    parser.add_argument("--transport", default="http",
-                        choices=["http", "https"],
-                        help="transport for connections")
-    parser.add_argument("-i", "--ip", default="localhost",
-                        help="OpenDaylight ip address")
-    parser.add_argument("-t", "--port", default="8181",
-                        help="OpenDaylight restconf port, defaul: 8181")
-    parser.add_argument("-u", "--user", default="admin",
-                        help="OpenDaylight restconf username, default: admin")
-    parser.add_argument("-w", "--pw", default="admin",
-                        help="OpenDaylight restconf password, default: admin")
-    parser.add_argument("-p", "--pretty_print", action="store_true",
-                        help="json dump with pretty_print")
-    parser.set_defaults(func=cmd.run_dump)
+    # Get a list of modules that was csv. The lambda parses the input into a list
+    parser.add_argument("--modules", default="all",
+                        type=lambda s: [item for item in s.split(',')],
+                        help="all or a list of modules")
+    parser.set_defaults(func=models.get_models)
 
 
 def add_parser(parsers):
     parser = parsers.add_parser("model", description="Tools for MDSAL models")
     subparsers = parser.add_subparsers(dest="subcommand", description="Model tools")
-    add_dump_parser(subparsers)
+    add_get_parser(subparsers)