def restartConnection(self): try: self.sr = sr.Connection(self.name, self.conn) except RuntimeError as r: if self.conn == sr.SR_CONN_DAEMON_REQUIRED and r.message == "The peer disconnected": sleep(1) #wait for daemon to start self.sr = sr.Connection(self.name, self.conn) else: raise r self.session = sr.Session(self.sr, self.ds)
def test_move_last_first(self): conn = sr.Connection("move_test4") self.session = sr.Session(conn, sr.SR_DS_STARTUP) self.session.move_item("/test-module:user[name='nameC']", sr.SR_MOVE_LAST) items = self.session.get_items("/test-module:user") self.compareListItems(items, ["A", "B", "D", "C"])
def test_move_after_unknown(self): conn = sr.Connection("move_test3") self.session = sr.Session(conn, sr.SR_DS_STARTUP) with self.assertRaises(RuntimeError): self.session.move_item("/test-module:user[name='nameB']", sr.SR_MOVE_AFTER, "/test-module:user[name='nameXY']")
def test_move_before_first(self): conn = sr.Connection("move_test2") self.session = sr.Session(conn, sr.SR_DS_STARTUP) self.session.move_item("/test-module:user[name='nameC']", sr.SR_MOVE_BEFORE, "/test-module:user[name='nameA']") items = self.session.get_items("/test-module:user") self.compareListItems(items, ["C", "A", "B", "D"])
def test_move_after_last(self): conn = sr.Connection("move_test1") self.session = sr.Session(conn, sr.SR_DS_STARTUP) print("move") self.session.move_item("/test-module:user[name='nameB']", sr.SR_MOVE_AFTER, "/test-module:user[name='nameD']") items = self.session.get_items("/test-module:user") self.compareListItems(items, ["A", "C", "D", "B"])
def setUp(self): conn = sr.Connection("move_test") session = sr.Session(conn, sr.SR_DS_STARTUP) session.delete_item("/test-module:*") session.set_item("/test-module:user[name='nameA']", None) session.set_item("/test-module:user[name='nameB']", None) session.set_item("/test-module:user[name='nameC']", None) session.set_item("/test-module:user[name='nameD']", None) session.commit()
def sysrepo_setup(state): """ Initialize sysrepo context which program uses """ state['connection'] = sr.Connection("measure performance") assert state['connection'] is not None return
def create_example_module(): connection = sr.Connection("test-module") session = sr.Session(connection, sr.SR_DS_STARTUP) session.delete_item("/example-module:*") v = sr.Val("Leaf value", sr.SR_STRING_T) session.set_item( "/example-module:container/list[key1='key1'][key2='key2']/leaf", v) session.commit()
def load_startup_datastore(ctx): conn = sr.Connection(ctx["yang_model"]) sess = sr.Session(conn, sr.SR_DS_STARTUP, sr.SR_SESS_CONFIG_ONLY) if None == conn or None == sess: return sr.SR_ERR_INTERNAL ctx["startup_conn"] = conn ctx["startup_sess"] = sess return sr.SR_ERR_OK
def test_commit_empty(self): TestModule.create_test_module() connection = sr.Connection("name") session = sr.Session(self.conn, sr.SR_DS_STARTUP) v_old = self.session.get_item("/test-module:main/string") self.session.delete_item("/test-module:*") self.session.commit() #test random leaf that was deleted v_none = self.session.get_item("/test-module:main/string") self.assertIsNone(v_none) self.session.set_item("/test-module:main/string", v_old) self.session.commit() TestModule.create_test_module()
def createDataTreeLargeExampleModule(count, datastore): """ Add data to example-module. """ conn = sr.Connection("load test") sess = sr.Session(conn, datastore) subs = sr.Subscribe(sess) for i in xrange(count): xpath = "/example-module:container/list[key1='key" + str(i) + "'][key2='key" + str(i) +"']/leaf" val = sr.Val("leaf" + str(i), sr.SR_STRING_T) sess.set_item(xpath, val) sess.commit()
def createDataTreeLargeIETFinterfacesModule(count, datastore): """ Add data to ietf-interfaces. """ conn = sr.Connection("load test") sess = sr.Session(conn, datastore) subs = sr.Subscribe(sess) for i in xrange(count): xpath = "/ietf-interfaces:interfaces/interface[name='eth" + str( i) + "']" xpath_ip = xpath + "/ietf-ip:ipv4/address[ip='192.168.1." + str( i) + "]" x_name = xpath + "/name" x_type = xpath + "/type" x_desc = xpath + "/description" x_enabled = xpath + "/enabled" x_ipv4_enabled = xpath + "/ietf-ip:ipv4/enabled" x_ipv4_mtu = xpath + "/ietf-ip:ipv4/mtu" x_ipv4_forward = xpath + "/ietf-ip:ipv4/forwarding" x_prefix_len = xpath_ip + "/prefix-length" val = sr.Val("Ethernet 0", sr.SR_STRING_T) sess.set_item(x_desc, val) val = sr.Val("ethernetCsmacd", sr.SR_IDENTITYREF_T) sess.set_item(x_type, val) val = sr.Val(True, sr.SR_BOOL_T) sess.set_item(x_enabled, val) val = sr.Val(True, sr.SR_BOOL_T) sess.set_item(x_ipv4_enabled, val) val = sr.Val(1500, sr.SR_UINT16_T) sess.set_item(x_ipv4_mtu, val) val = sr.Val(False, sr.SR_BOOL_T) sess.set_item(x_ipv4_forward, val) sess.commit()
def clearDataTree(module_name, datastore): """ Clear yang model. """ conn = sr.Connection("clear") sess = sr.Session(conn, datastore) subs = sr.Subscribe(sess) xpath = "/" + module_name + ":*//*" values = sess.get_items(xpath) if values == None: return for i in xrange(values.val_cnt()): sess.delete_item(values.val(i).xpath()) sess.commit()
def setup(self): if self.autoconnect: self.sr = sr.Connection(self.name, self.conn) self.session = sr.Session(self.sr, self.ds)
# subscribe for changes in running config */ ctx["sub"].module_change_subscribe(ctx["yang_model"], module_change_cb, ctx) print "\n\n ========== READING STARTUP CONFIG: ==========\n" try: print_current_config(session, yang_model); except Exception as e: print e print "\n\n ========== STARTUP CONFIG APPLIED AS RUNNING ==========\n" sr.global_loop() return rc # Main client function try: # connect to sysrepo conn = sr.Connection("example_application") # start session sess = sr.Session(conn) rc = sr_plugin_init_cb(sess) if (sr.SR_ERR_OK != rc): print "error" print "Application exit requested, exiting.\n"; except Exception as e: print e
def create_test_module(): connection = sr.Connection("test-module") session = sr.Session(connection, sr.SR_DS_STARTUP) session.delete_item("/test-module:*") v = sr.Val(XP_TEST_MODULE_ENUM_VALUE, sr.SR_ENUM_T) session.set_item(XP_TEST_MODULE_ENUM, v) v = sr.Val(XP_TEST_MODULE_RAW_VALUE, sr.SR_BINARY_T) session.set_item(XP_TEST_MODULE_RAW, v) v = sr.Val(XP_TEST_MODULE_BITS_VALUE, sr.SR_BITS_T) session.set_item(XP_TEST_MODULE_BITS, v) v = sr.Val(XP_TEST_MODULE_BOOL_VALUE_T, sr.SR_BOOL_T) session.set_item(XP_TEST_MODULE_BOOL, v) v = sr.Val(XP_TEST_MODULE_DEC64_VALUE_T) session.set_item(XP_TEST_MODULE_DEC64, v) v = sr.Val(None, sr.SR_LEAF_EMPTY_T) session.set_item(XP_TEST_MODULE_EMPTY, v) v = sr.Val(XP_TEST_MODULE_IDREF_VALUE, sr.SR_IDENTITYREF_T) session.set_item(XP_TEST_MODULE_IDREF, v) v = sr.Val(XP_TEST_MODULE_STRING_VALUE, sr.SR_STRING_T) session.set_item(XP_TEST_MODULE_STRING, v) v = sr.Val(XP_TEST_MODULE_INT8_VALUE_T, sr.SR_INT8_T) session.set_item(XP_TEST_MODULE_INT8, v) v = sr.Val(XP_TEST_MODULE_INT16_VALUE_T, sr.SR_INT16_T) session.set_item(XP_TEST_MODULE_INT16, v) v = sr.Val(XP_TEST_MODULE_INT32_VALUE_T, sr.SR_INT32_T) session.set_item(XP_TEST_MODULE_INT32, v) v = sr.Val(XP_TEST_MODULE_INT64_VALUE_T, sr.SR_INT64_T) session.set_item(XP_TEST_MODULE_INT64, v) v = sr.Val(XP_TEST_MODULE_UINT8_VALUE_T, sr.SR_UINT8_T) session.set_item(XP_TEST_MODULE_UINT8, v) v = sr.Val(XP_TEST_MODULE_UINT16_VALUE_T, sr.SR_UINT16_T) session.set_item(XP_TEST_MODULE_UINT16, v) v = sr.Val(XP_TEST_MODULE_UINT32_VALUE_T, sr.SR_UINT32_T) session.set_item(XP_TEST_MODULE_UINT32, v) v = sr.Val(XP_TEST_MODULE_UINT64_VALUE_T, sr.SR_UINT64_T) session.set_item(XP_TEST_MODULE_UINT64, v) for num in [1, 2, 42]: v = sr.Val(num, sr.SR_UINT8_T) session.set_item("/test-module:main/numbers", v) session.set_item("/test-module:list[key='k1']", None) v = sr.Val("id_1", sr.SR_IDENTITYREF_T) session.set_item("/test-module:list[key='k1']/id_ref", v) v = sr.Val(None, sr.SR_CONTAINER_PRESENCE_T) session.set_item("/test-module:list[key='k1']/wireless", v) session.set_item("/test-module:list[key='k2']", None) v = sr.Val("id_2", sr.SR_IDENTITYREF_T) session.set_item("/test-module:list[key='k2']/id_ref", v) session.commit()
def create_ietf_interfaces(): connection = sr.Connection("ietf-interfaces") session = sr.Session(connection, sr.SR_DS_STARTUP) session.delete_item("/ietf-interfaces:*") v = sr.Val("iana-if-type:ethernetCsmacd", sr.SR_IDENTITYREF_T) session.set_item("/ietf-interfaces:interfaces/interface[name='eth0']/type", v) v = sr.Val("Ethernet 0", sr.SR_STRING_T) session.set_item( "/ietf-interfaces:interfaces/interface[name='eth0']/description", v) v = sr.Val(True, sr.SR_BOOL_T) session.set_item( "/ietf-interfaces:interfaces/interface[name='eth0']/enabled", v) v = sr.Val(24, sr.SR_UINT8_T) session.set_item( "/ietf-interfaces:interfaces/interface[name='eth0']/ietf-ip:ipv4/address[ip='192.168.2.100']/prefix-length", v) v = sr.Val(True, sr.SR_BOOL_T) session.set_item( "/ietf-interfaces:interfaces/interface[name='eth0']/ietf-ip:ipv4/enabled", v) v = sr.Val(1500, sr.SR_UINT16_T) session.set_item( "/ietf-interfaces:interfaces/interface[name='eth0']/ietf-ip:ipv4/mtu", v) v = sr.Val("iana-if-type:ethernetCsmacd", sr.SR_IDENTITYREF_T) session.set_item("/ietf-interfaces:interfaces/interface[name='eth1']/type", v) v = sr.Val("Ethernet 1", sr.SR_STRING_T) session.set_item( "/ietf-interfaces:interfaces/interface[name='eth1']/description", v) v = sr.Val(True, sr.SR_BOOL_T) session.set_item( "/ietf-interfaces:interfaces/interface[name='eth1']/ietf-ip:ipv4/enabled", v) v = sr.Val(1500, sr.SR_UINT16_T) session.set_item( "/ietf-interfaces:interfaces/interface[name='eth1']/ietf-ip:ipv4/mtu", v) v = sr.Val("iana-if-type:ethernetCsmacd", sr.SR_IDENTITYREF_T) session.set_item( "/ietf-interfaces:interfaces/interface[name='gigaeth0']/type", v) v = sr.Val("GigabitEthernet 0", sr.SR_STRING_T) session.set_item( "/ietf-interfaces:interfaces/interface[name='gigaeth0']/description", v) v = sr.Val(False, sr.SR_BOOL_T) session.set_item( "/ietf-interfaces:interfaces/interface[name='gigaeth0']/enabled", v) session.commit()
# Notable difference between c implementation is using exception mechanism for open handling unexpected events. # Here it is useful because `Conenction`, `Session` and `Subscribe` could throw an exception. if __name__ == "__main__": if len(sys.argv) < 2: print("Usage: %s path_to_subscribe [output_file]\n\n") sys.exit(1) settings = {'xpath': sys.argv[1], 'filename': sys.argv[2]} print("Application will watch for changes under xpath " + settings['xpath'] + "\n") # connect to sysrepo conn = sr.Connection("notification_test_application") # start session sess = sr.Session(conn) # subscribe for changes in running config */ subscribe = sr.Subscribe(sess) # Subscribe for changes in running config - try multiple times because in test # multiple client might try to subscribe to the same model for i in xrange(3): try: subscribe.subtree_change_subscribe(settings['xpath'], subtree_change_cb, settings) except Exception as e: # Multiple clients might try to subscribe to the same model.
def setUpClass(self): TestModule.create_test_module() self.conn = sr.Connection("abc", sr.SR_DS_STARTUP)