README added

Miha

2017-01-17 ab06932d8e6534679f805ba43e851cc154d78cd1

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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
import paramiko
import os.path
 
class Connection:
    def __init__(self, hostname, port=22, username=None, password=None):
        self.hostname=hostname
        self.port=port
        if(username!=None):
            self.username=username
        else:
            self.username=''
        if(password!=None):
            self.password=password
        else:
            self.password=''
        self.ssh=paramiko.SSHClient()
        self.ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        self.connected=False
        return
    
    def connect(self, Timeout=5):
        if(not self.connected):
            try:
                print("Trying to connect to: "+self.username+"@"+self.hostname+":"+str(self.port)+".")
                self.ssh.connect(self.hostname, username=self.username, password=self.password, port=self.port, timeout=Timeout)
                self.connected=True
            except:
                print("Error establishing connection with "+self.username+"@"+self.hostname+":"+str(self.port)+".")
                exit(1)
        else:
            print("Already connected!")
        return
 
    def disconnect(self):
        if(self.connected):
            try:
                self.ssh.close()
            except:
                print("Cannot disconect. Unknown error.")
        else:
            print("Cannot disconect. Already disconnected.")
        self.connected=False
 
    def execute(self,command):
        if(self.connected):
            try:
                stdin,stdout,stderr=self.ssh.exec_command(command)
                output=stdout.readlines()
                errors=stderr.readlines()
            #    print(errors)
                return(output)
            except:
                print("Cannot execute remote commands")
        else:
            print("Cannot execute remote commands. Connect first.")
 
    def send_file(self, local, remote):
        sftp=self.ssh.open_sftp()
        sftp.put(local,remote)
        sftp.close()
 
    def receive_file(self,remote,local):
        sftp=self.ssh.open_sftp()
        sftp.get(remote,local)
        sftp.close()
 
    def mkdir_remote(self,directory):
        sftp=self.ssh.open_sftp()
        sftp.mkdir(directory)
        sftp.close()
 
    def mkdir_p(self,sftp, remote_directory):
        """Change to this directory, recursively making new folders if needed.
        Returns True if any folders were created. Recursive algorithm."""
        if remote_directory == '/':
            # absolute path so change directory to root
            sftp.chdir('/')
            return
        if remote_directory == '':
            # top-level relative directory must exist
            return
        try:
            sftp.chdir(remote_directory) # sub-directory exists
        except IOError:
            dirname, basename = os.path.split(remote_directory.rstrip('/'))
            self.mkdir_p(sftp, dirname) # make parent directories
            sftp.mkdir(basename) # sub-directory missing, so created it
            sftp.chdir(basename)
        return True
    
    def send_multiple_files_in_directory(self,local_files,directory):
        sftp=self.ssh.open_sftp()
#        try:
#            sftp.chdir(directory)  # Test if remote_path exists
#        except (IOError,FileNotFoundError):
#            sftp.mkdir(directory)  # Create remote_path
#            sftp.chdir(directory)
        self.mkdir_p(sftp, directory)
        for f in set(local_files):    
            sftp.put(f, f)
        sftp.close()