summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorDaniil Baturin <daniil@baturin.org>2018-01-26 12:43:53 +0700
committerGitHub <noreply@github.com>2018-01-26 12:43:53 +0700
commitb8f383f1bad5439f05a537f46cd8d04abb74b2c9 (patch)
tree02c9b2c68aeb7600bf6e1eb651a42bea5bfeb3e6
parent82dea5997dc05cc124c8723d99801df0dd9b5018 (diff)
parentfde0d9fab1a7d1c1e075bd6b8a71407b97ad63fc (diff)
downloadpython-vyos-mgmt-b8f383f1bad5439f05a537f46cd8d04abb74b2c9.tar.gz
python-vyos-mgmt-b8f383f1bad5439f05a537f46cd8d04abb74b2c9.zip
T527: Merge pull request #1 from tomodachi/ssh_key_support
Add support for ssh key authentication.
-rw-r--r--vymgmt/router.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/vymgmt/router.py b/vymgmt/router.py
index f4379c3..05af3ab 100644
--- a/vymgmt/router.py
+++ b/vymgmt/router.py
@@ -59,18 +59,20 @@ class ConfigLocked(CommitError):
class Router(object):
""" Router configuration interface class
"""
- def __init__(self, address, user, password='', port=22):
+ def __init__(self, address, user, password='', port=22, ssh_key=''):
""" Router configuration interface class
:param address: Router address,example:'192.0.2.1'
:param user: Router user
:param password: Router user's password
:param port: SSH port
+ :param ssh_key: SSH private key
"""
self.__address = address
self.__user = user
self.__password = password
self.__port = port
+ self.__ssh_key = ssh_key
# Session flags
self.__logged_in = False
@@ -120,7 +122,7 @@ class Router(object):
# There may or may not be a better way to handle it
self.__conn = pxssh.pxssh()
- self.__conn.login(self.__address, self.__user, password=self.__password, port=self.__port)
+ self.__conn.login(self.__address, self.__user, password=self.__password, port=self.__port, ssh_key=self.__ssh_key)
self.__logged_in = True
def logout(self):