48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
import os
|
|
import urlparse
|
|
import ftplib
|
|
from log_picker.sending.senderbaseclass import SenderBaseClass
|
|
from log_picker.sending.senderbaseclass import SenderError
|
|
|
|
# This class uses code from module report.plugins.scp
|
|
|
|
class FtpSender(SenderBaseClass):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
SenderBaseClass.__init__(self, args, kwargs)
|
|
self.host = None
|
|
self.username = None
|
|
self.password = None
|
|
|
|
def set_host(self, host):
|
|
if not host.startswith('ftp://'):
|
|
host = 'ftp://' + host
|
|
self.host = host
|
|
|
|
def set_login(self, username, password):
|
|
self.username = username
|
|
self.password = password
|
|
|
|
def sendfile(self, filename, contenttype):
|
|
_, netloc, path, _, _, _ = urlparse.urlparse(self.host)
|
|
if netloc.find(':') > 0:
|
|
netloc, port = netloc.split(':')
|
|
else:
|
|
port = 21
|
|
|
|
try:
|
|
ftp = ftplib.FTP()
|
|
ftp.connect(netloc, port)
|
|
if self.username:
|
|
ftp.login(self.username, self.password)
|
|
else:
|
|
ftp.login()
|
|
ftp.cwd(path)
|
|
ftp.set_pasv(True)
|
|
ftp.storbinary('STOR %s' % os.path.basename(filename), \
|
|
file(filename))
|
|
ftp.quit()
|
|
except ftplib.all_errors as e:
|
|
raise SenderError("FTP upload failed: %(error)s" % {'error':e})
|
|
|