Fix flake8

This commit is contained in:
Valentin Chemiere 2015-04-27 11:34:07 +02:00
parent 6e9e9fff5a
commit ade3a27b7d
5 changed files with 76 additions and 15 deletions

61
external_file_location/$ Normal file
View File

@ -0,0 +1,61 @@
# -*- coding: utf-8 -*-
###############################################################################
#
# Copyright (C) 2015 Akretion (http://www.akretion.com).
# @author Valentin CHEMIERE <valentin.chemiere@akretion.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
###############################################################################
import mock
from contextlib import contextmanager
class ConnMock(object):
def __init__(self, response):
import ipdb; ipdb.set_trace()
self.response = response
def isfile(self, path):
return self.response['isfile']
def makedir(self, path):
return self.response['makedir']
def exists(self, path):
return self.response['exists']
def open(self, path, mode):
return self.response['open']
def listdir(self, path):
return self.response['listdir']
def remove(self, path):
return self.response['remove']
def rename(self, source_path, target_path):
return self.response['rename']
def __call__(self, *args, **kwargs):
return self
def __exit__(self, *args, **kwargs):
pass
@contextmanager
def sftp_mock(response):
with mock.patch('fs.sftpfs.SFTPFS', ConnMock(response)) as SFTPFS:
yield

View File

@ -1 +1,3 @@
<EFBFBD>}q.
}q(Utests/test_sftp.pyq}U/usr/lib/python2.7/tempfile.py}qUNamedTemporaryFilecrope.base.oi.memorydb
ScopeInfo
q)<29>q}q(Unone…qhhhhUunknown…tUinstanceUdefinedU/usr/lib/python2.7/tempfile.pyU_TemporaryFileWrapper‡†s}q†bsu.

View File

@ -42,7 +42,8 @@ class FtpImportTask(FtpTask):
_synchronize_type = 'import'
def run(self):
with ftpfs.FTPFS(self.host, self.user, self.pwd, port=self.port) as ftp_conn:
with ftpfs.FTPFS(self.host, self.user, self.pwd,
port=self.port) as ftp_conn:
files_to_process = self._get_files(ftp_conn, self.path)
for file_to_process in files_to_process:
self._process_file(ftp_conn, file_to_process)
@ -57,10 +58,8 @@ class FtpExportTask(FtpTask):
if attachment.state in ('pending', 'failed'):
self.attachment_id = attachment
with ftpfs.FTPFS(self.host, self.user, self.pwd,
port=self.port) as ftp_conn:
port=self.port) as ftp_conn:
self._upload_file(ftp_conn, self.host, self.port,
self.user,
self.pwd,
self.path,
self.user, self.pwd, self.path,
attachment.datas_fname,
b64decode(attachment.datas))

View File

@ -44,8 +44,10 @@ class SftpImportTask(SftpTask):
def run(self):
connection_string = "{}:{}".format(self.host, self.port)
root = "/home/{}".format(self.user)
with sftpfs.SFTPFS(connection=connection_string, root_path=root,
username=self.user, password=self.pwd) as sftp_conn:
with sftpfs.SFTPFS(connection=connection_string,
root_path=root,
username=self.user,
password=self.pwd) as sftp_conn:
files_to_process = self._get_files(sftp_conn, self.path)
for file_to_process in files_to_process:
self._process_file(sftp_conn, file_to_process)
@ -61,12 +63,9 @@ class SftpExportTask(SftpTask):
self.attachment_id = attachment
connection_string = "{}:{}".format(self.host, self.port)
with sftpfs.SFTPFS(connection=connection_string,
username=self.user,
password=self.pwd) as sftp_conn:
username=self.user,
password=self.pwd) as sftp_conn:
datas = b64decode(attachment.datas)
self._upload_file(sftp_conn, self.host, self.port,
self.user,
self.pwd,
self.path,
attachment.datas_fname,
datas)
self.user, self.pwd, self.path,
attachment.datas_fname, datas)