| 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 = 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() |
|
|
| |
| 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) |
