mirror of https://github.com/jumpserver/jumpserver
117 lines
3.4 KiB
Python
117 lines
3.4 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
#
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
from ftplib import FTP, error_perm
|
|||
|
|
from .base import ObjectStorage
|
|||
|
|
|
|||
|
|
|
|||
|
|
class FTPStorage(ObjectStorage):
|
|||
|
|
|
|||
|
|
def __init__(self, config):
|
|||
|
|
self.host = config.get("HOST", None)
|
|||
|
|
self.port = int(config.get("PORT", 21))
|
|||
|
|
self.username = config.get("USERNAME", None)
|
|||
|
|
self.password = config.get("PASSWORD", None)
|
|||
|
|
self.pasv = bool(config.get("PASV", False))
|
|||
|
|
self.dir = config.get("DIR", "replay")
|
|||
|
|
self.client = FTP()
|
|||
|
|
self.client.encoding = 'utf-8'
|
|||
|
|
self.client.set_pasv(self.pasv)
|
|||
|
|
self.pwd = '.'
|
|||
|
|
self.connect()
|
|||
|
|
|
|||
|
|
def connect(self, timeout=-999, source_address=None):
|
|||
|
|
self.client.connect(self.host, self.port, timeout, source_address)
|
|||
|
|
self.client.login(self.username, self.password)
|
|||
|
|
if not self.check_dir_exist(self.dir):
|
|||
|
|
self.mkdir(self.dir)
|
|||
|
|
self.client.cwd(self.dir)
|
|||
|
|
self.pwd = self.client.pwd()
|
|||
|
|
|
|||
|
|
def confirm_connected(self):
|
|||
|
|
try:
|
|||
|
|
self.client.pwd()
|
|||
|
|
except Exception:
|
|||
|
|
self.connect()
|
|||
|
|
|
|||
|
|
def upload(self, src, target):
|
|||
|
|
self.confirm_connected()
|
|||
|
|
target_dir = os.path.dirname(target)
|
|||
|
|
exist = self.check_dir_exist(target_dir)
|
|||
|
|
if not exist:
|
|||
|
|
ok = self.mkdir(target_dir)
|
|||
|
|
if not ok:
|
|||
|
|
raise PermissionError('Dir create error: %s' % target)
|
|||
|
|
try:
|
|||
|
|
with open(src, 'rb') as f:
|
|||
|
|
self.client.storbinary('STOR '+target, f)
|
|||
|
|
return True, None
|
|||
|
|
except Exception as e:
|
|||
|
|
return False, e
|
|||
|
|
|
|||
|
|
def download(self, src, target):
|
|||
|
|
self.confirm_connected()
|
|||
|
|
try:
|
|||
|
|
os.makedirs(os.path.dirname(target), 0o755, exist_ok=True)
|
|||
|
|
with open(target, 'wb') as f:
|
|||
|
|
self.client.retrbinary('RETR ' + src, f.write)
|
|||
|
|
return True, None
|
|||
|
|
except Exception as e:
|
|||
|
|
return False, e
|
|||
|
|
|
|||
|
|
def delete(self, path):
|
|||
|
|
self.confirm_connected()
|
|||
|
|
if not self.exists(path):
|
|||
|
|
raise FileNotFoundError('File not exist error(%s)' % path)
|
|||
|
|
try:
|
|||
|
|
self.client.delete(path)
|
|||
|
|
return True, None
|
|||
|
|
except Exception as e:
|
|||
|
|
return False, e
|
|||
|
|
|
|||
|
|
def check_dir_exist(self, d):
|
|||
|
|
pwd = self.client.pwd()
|
|||
|
|
try:
|
|||
|
|
self.client.cwd(d)
|
|||
|
|
self.client.cwd(pwd)
|
|||
|
|
return True
|
|||
|
|
except error_perm:
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def mkdir(self, dirs):
|
|||
|
|
self.confirm_connected()
|
|||
|
|
# 创建多级目录，ftplib不支持一次创建多级目录
|
|||
|
|
dir_list = dirs.split('/')
|
|||
|
|
pwd = self.client.pwd()
|
|||
|
|
try:
|
|||
|
|
for d in dir_list:
|
|||
|
|
if not d or d in ['.']:
|
|||
|
|
continue
|
|||
|
|
# 尝试切换目录
|
|||
|
|
try:
|
|||
|
|
self.client.cwd(d)
|
|||
|
|
continue
|
|||
|
|
except:
|
|||
|
|
pass
|
|||
|
|
# 切换失败创建这个目录，再切换
|
|||
|
|
try:
|
|||
|
|
self.client.mkd(d)
|
|||
|
|
self.client.cwd(d)
|
|||
|
|
except:
|
|||
|
|
return False
|
|||
|
|
return True
|
|||
|
|
finally:
|
|||
|
|
self.client.cwd(pwd)
|
|||
|
|
|
|||
|
|
def exists(self, target):
|
|||
|
|
self.confirm_connected()
|
|||
|
|
try:
|
|||
|
|
self.client.size(target)
|
|||
|
|
return True
|
|||
|
|
except:
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def close(self):
|
|||
|
|
self.client.close()
|