File size: 3,112 Bytes
b75a90e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 | import os
from flask import Flask
from webdav3.client import Client
import paramiko
import logging
from requests.auth import HTTPBasicAuth
# 配置日志
logging.basicConfig(level=logging.INFO)
# 从环境变量获取配置信息
WEBDAV_URL = os.getenv('WEBDAV_URL')
WEBDAV_USERNAME = os.getenv('WEBDAV_USERNAME')
WEBDAV_PASSWORD = os.getenv('WEBDAV_PASSWORD')
WEBDAV_PATH = os.getenv('WEBDAV_PATH')
SSH_HOST = os.getenv('SSH_HOST')
SSH_PORT = 22
SSH_USERNAME = os.getenv('SSH_USERNAME')
SSH_PASSWORD = os.getenv('SSH_PASSWORD')
SSH_PATH = os.getenv('SSH_PATH')
app = Flask(__name__)
def safe_download(webdav_client, remote_path, local_path):
try:
webdav_client.download(remote_path, local_path)
except KeyError as e:
if str(e) == "'content-length'":
logging.warning(f"'content-length' missing for {remote_path}. Using alternative download method.")
# 使用基本认证进行请求
auth = HTTPBasicAuth(WEBDAV_USERNAME, WEBDAV_PASSWORD)
response = webdav_client.session.get(webdav_client.get_url(remote_path), stream=True, auth=auth)
if response.status_code != 200:
logging.error(f"Failed to download {remote_path}. Status code: {response.status_code}")
return
with open(local_path, 'wb') as f:
for chunk in response.iter_content(chunk_size=8192):
if chunk:
f.write(chunk)
logging.info(f"Downloaded file size: {os.path.getsize(local_path)} bytes")
else:
raise
@app.route('/')
def transfer_images():
options = {
'webdav_hostname': WEBDAV_URL,
'webdav_login': WEBDAV_USERNAME,
'webdav_password': WEBDAV_PASSWORD
}
webdav_client = Client(options)
# 创建SSH客户端
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect(SSH_HOST, port=SSH_PORT, username=SSH_USERNAME, password=SSH_PASSWORD)
sftp = ssh.open_sftp()
# 获取WebDAV中的文件列表
files = webdav_client.list(WEBDAV_PATH)
# 转移文件
for file in files:
if file.endswith('.jpg'):
file_name = os.path.basename(file)
remote_path = os.path.join(SSH_PATH, file_name)
logging.info(f"Checking if file {file_name} exists on remote server...")
try:
sftp.stat(remote_path)
logging.info(f"File {file_name} already exists. Skipping...")
except FileNotFoundError:
local_tmp_file = '/tmp/' + file_name
safe_download(webdav_client, f"{WEBDAV_PATH}/{file_name}", local_tmp_file)
if os.path.exists(local_tmp_file) and os.path.getsize(local_tmp_file) > 0:
sftp.put(local_tmp_file, remote_path)
logging.info(f"File {file_name} transferred with size: {os.path.getsize(local_tmp_file)} bytes")
sftp.close()
ssh.close()
return "Transfer complete!"
if __name__ == "__main__":
app.run(host='0.0.0.0', port=8080) |