Remove module/origin parsing in update generation

Author: remingtonc

src/cisco_gnmi/builder.py

@@ -119,8 +119,9 @@ def set_os(self, name=None):
         name : str
             "IOS XR" maps to the XRClient class.
             "NX-OS" maps to the NXClient class.
+            "IOS XE" maps to the XEClient class.
             None maps to the base Client class which simply wraps the gNMI stub.
-            ["IOS XR", "NX-OS", None]
+            ["IOS XR", "NX-OS", "IOS XE", None]
         
         Returns
         -------
@@ -273,7 +274,7 @@ def construct(self):
 
         Returns
         -------
-        Client or XRClient
+        Client or NXClient or XEClient or XRClient
         """
         channel = None
         channel_ssl_creds = None

src/cisco_gnmi/xe.py

@@ -159,24 +159,9 @@ def create_updates(name, configs):
                 if len(config.keys()) > 1:
                     raise Exception("config should only target one YANG module!")
                 top_element = next(iter(config.keys()))
-                top_element_split = top_element.split(":")
-                if len(top_element_split) < 2:
-                    raise Exception(
-                        "Top level config element {} should be module prefixed!".format(
-                            top_element
-                        )
-                    )
-                if len(top_element_split) > 2:
-                    raise Exception(
-                        "Top level config element {} appears malformed!".format(
-                            top_element
-                        )
-                    )
-                origin = top_element_split[0]
-                element = top_element_split[1]
-                config = config.pop(top_element)
                 update = proto.gnmi_pb2.Update()
-                update.path.CopyFrom(self.parse_xpath_to_gnmi_path(element, origin))
+                update.path.CopyFrom(self.parse_xpath_to_gnmi_path(top_element))
+                config = config.pop(top_element)
                 if ietf:
                     update.val.json_ietf_val = json.dumps(config).encode("utf-8")
                 else: