1 files changed, 47 insertions, 15 deletions
diff --git a/example.py b/example.py
index a58b656..3f81de0 100644
--- a/example.py
+++ b/example.py
@@ -1,34 +1,66 @@
-#import warnings
-#warnings.filterwarnings("ignore", category=RuntimeWarning)    
-
+# importing modules
+import warnings
+warnings.filterwarnings("ignore", category=RuntimeWarning)    
 import sys
 import os
-
-# Adicione o diretório raiz do projeto ao sys.path para que possa importar vyapi
-sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__))))
-
+# adding pyvyos to sys.path
+#sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__))))
 
 import unittest
-from vyapi.device import VyDevice
-from vyapi.device import ApiResponse
 from dotenv import load_dotenv
 import pprint 
+import random
+import string
+
+# importing pyvyos modules
+from pyvyos.device import VyDevice
+from pyvyos.device import ApiResponse
 
-load_dotenv()  
 
+# getting env variables
+load_dotenv()  
 hostname = os.getenv('VYDEVICE_HOSTNAME')
-key = os.getenv('VYDEVICE_KEY')
+apikey = os.getenv('VYDEVICE_APIKEY')
 port = os.getenv('VYDEVICE_PORT')
 protocol = os.getenv('VYDEVICE_PROTOCOL')
 verify = os.getenv('VYDEVICE_VERIFY_SSL')
 if verify == "False":
     verify = False
-
 else:
     verify = True
 
+# running example
 if __name__ == '__main__':
-    device = VyDevice(hostname=hostname, key=key, port=port, protocol=protocol, verify=verify)
-    response = device.retrieve_show_config(['system'])
-    pprint.pprint(response)
+    # preparing connection to vyos device
+    device = VyDevice(hostname=hostname, apikey=apikey, port=port, protocol=protocol, verify=verify)
+
+
+
+    #response = device.retrieve_show_config(['system'])
+    #pprint.pprint(response)
+
+    #print("### Generating ssh key ###")
+    #randstring = ''.join(random.choice(string.ascii_letters + string.digits) for _ in range(20))
+    #keyrand =  f'/tmp/key_{randstring}'
+    #response = device.generate(path=["ssh", "client-key", keyrand])
+    #pprint.pprint(response)
+
+
+
+    #response = device.retrieve_return_values(path=["interfaces", "ethernet", "eth0", "address"])
+    #pprint.pprint(response)
+
+    #response = device.reset(path=["conntrack-sync", "internal-cache"])
+    #pprint.pprint(response)    
+
+    #response = device.reboot(path=["now"])
+    #pprint.pprint(response)  
+
+    #response = device.shutdown(path=["now"])
+    #pprint.pprint(response)  
+
+    #response = device.image_add(url="https://github.com/vyos/vyos-rolling-nightly-builds/releases/download/1.5-rolling-202312130023/vyos-1.5-rolling-202312130023-amd64.iso")
+    #pprint.pprint(response)
 
+    response = device.image_delete(name="foo")
+    pprint.pprint(response)