forked from redsnapper8t8/pyvarnish
-
Notifications
You must be signed in to change notification settings - Fork 0
/
remote.py
49 lines (40 loc) · 1.48 KB
/
remote.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
# -*- coding: utf-8 -*-
__author__ = 'John Moylan'
import sys
from paramiko import SSHClient, SSHConfig, AutoAddPolicy
class Varnish_admin():
def __init__(self, server, ssh_config_file):
self.server = server
self.ssh_config_file = ssh_config_file
self.conf = {
'hostname': server,
'port': 22,
# If these are None, Paramiko will figure out the correct values.
'user': None,
'identityfile': None,
}
self.conf.update(self.config())
def config(self):
sshconfig = SSHConfig()
try:
sshconfig.parse(open(self.ssh_config_file))
except IOError:
print 'SSH config file location invalid.'
sys.exit(1)
conf = sshconfig.lookup(self.server)
if 'port' in conf:
conf['port'] = int(conf['port'])
return conf
def runcmd(self, cmd):
try:
client = SSHClient()
client.load_system_host_keys()
client.set_missing_host_key_policy(AutoAddPolicy())
client.connect(self.conf['hostname'],
port = self.conf['port'],
username = self.conf['user'],
key_filename = self.conf['identityfile'],)
stdin, stdout, stderr = client.exec_command(cmd)
return ''.join([i.rstrip('\r\n ').lstrip() for i in stdout.readlines()])
finally:
client.close()