def test_override_settings(self):
with override_settings(SFTP_STORAGE_ROOT='foo1'):
storage = sftpstorage.SFTPStorage()
self.assertEqual(storage._root_path, 'foo1')
with override_settings(SFTP_STORAGE_ROOT='foo2'):
storage = sftpstorage.SFTPStorage()
self.assertEqual(storage._root_path, 'foo2')
def save_upload_file_to_sftp(PostFile, FilePath):
sftp_class = sftpstorage.SFTPStorage()
save_sftp = sftp_class._save(FilePath,PostFile)
if save_sftp != FilePath:
return u"写入文件错误"
# print(save_sftp)
# logging.info("sftp return name is ----->"+save_sftp)
else:
return u"SUCCESS"
def save_scrawl_file_to_sftp(request, savepath, filename):
savepath = 'scram_data/' + savepath
content = request.POST.get(USettings.UEditorUploadSettings.get("scrawlFieldName", "upfile"))
sftp_class = sftpstorage.SFTPStorage()
save_sftp = sftp_class._save(savepath,filename,base64.b64decode(content))
if save_sftp != savepath:
return u"写入文件错误"
else:
return u"SUCCESS"
def setUp(self):
self.storage = sftpstorage.SFTPStorage('foo')
def setUp(self):
self.storage = sftpstorage.SFTPStorage('foo')
self.file = sftpstorage.SFTPStorageFile('bar', self.storage, 'wb')
def test_override_init_argument(self):
storage = sftpstorage.SFTPStorage(root_path='foo1')
self.assertEqual(storage._root_path, 'foo1')
storage = sftpstorage.SFTPStorage(root_path='foo2')
self.assertEqual(storage._root_path, 'foo2')
