63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
|
import os
|
||
|
|
import subprocess
|
||
|
|
from log_picker.sending.senderbaseclass import SenderBaseClass
|
||
|
|
from log_picker.sending.senderbaseclass import SenderError
|
||
|
|
|
||
|
|
|
||
|
|
class ScpSender(SenderBaseClass):
|
||
|
|
|
||
|
|
def __init__(self, *args, **kwargs):
|
||
|
|
SenderBaseClass.__init__(self, args, kwargs)
|
||
|
|
self.host = None
|
||
|
|
self.port = None
|
||
|
|
self.username = None
|
||
|
|
self.path = "."
|
||
|
|
|
||
|
|
|
||
|
|
def set_host(self, host):
|
||
|
|
if host.find(":") != -1:
|
||
|
|
(self.host, port) = host.split(":")
|
||
|
|
try:
|
||
|
|
self.port = int(port)
|
||
|
|
except ValueError:
|
||
|
|
self.port = None
|
||
|
|
else:
|
||
|
|
self.host = host
|
||
|
|
|
||
|
|
|
||
|
|
def set_login(self, username):
|
||
|
|
self.username = username
|
||
|
|
|
||
|
|
|
||
|
|
def set_path(self, path):
|
||
|
|
if path: self.path = path
|
||
|
|
|
||
|
|
|
||
|
|
def sendfile(self, filename, contenttype):
|
||
|
|
port_args = []
|
||
|
|
if self.port:
|
||
|
|
port_args = ["-P", self.port]
|
||
|
|
|
||
|
|
target = "%s@%s:%s" % (self.username, self.host, self.path)
|
||
|
|
|
||
|
|
command = ["scp",
|
||
|
|
"-q",
|
||
|
|
"-oGSSAPIAuthentication=no",
|
||
|
|
"-oHostbasedAuthentication=no",
|
||
|
|
"-oPubkeyAuthentication=no",
|
||
|
|
"-oChallengeResponseAuthentication=no",
|
||
|
|
"-oPasswordAuthentication=yes",
|
||
|
|
"-oNumberOfPasswordPrompts=1",
|
||
|
|
"-oStrictHostKeyChecking=no",
|
||
|
|
"-oUserKnownHostsFile=/dev/null",
|
||
|
|
] + port_args + [filename, target]
|
||
|
|
|
||
|
|
p = subprocess.Popen(command, stdin=subprocess.PIPE)
|
||
|
|
p.communicate()
|
||
|
|
|
||
|
|
if p.returncode:
|
||
|
|
raise SenderError("Scp sending failed.\n" + \
|
||
|
|
"Possible causes: Bad hostname, bad username, "\
|
||
|
|
"bad password, host is down.")
|
||
|
|
|