diff options
-rw-r--r-- | vyroute/Router.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/vyroute/Router.py b/vyroute/Router.py index 86f3c78..476414a 100644 --- a/vyroute/Router.py +++ b/vyroute/Router.py @@ -113,6 +113,7 @@ class BasicRouter(Router): self.__conn.close() self.__status["status"] = "logout" self.__status["configure"] = None + self.__conn = pxssh() return {"Result": "Logout successfully."} except Exception as e: return {"Error": e} @@ -215,6 +216,10 @@ class BasicRouter(Router): return {"Result": "Exit configure mode successfully."} else: return {"Error": "You should save first."} + elif self.__status["commit"] is None: + self.__conn.sendline("exit") + self.__conn.prompt() + return {"Result": "Exit configure mode successfully."} else: return {"Error": "You should commit first."} else: |