administrateur/mkdocs-auto
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: patching webdav client

Browse Source
This commit is contained in:
administrateur 2025-03-29 19:33:13 +01:00
parent cd6522b99c
commit f546480afa
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
script_creator.py
View File

@ -7,24 +7,24 @@ import redis
import shutil import shutil
from threading import Lock from threading import Lock
from webdav3.client import Client from webdav3.client import Client
from requests.structures import CaseInsensitiveDict
from webdav3.client import WebDavClient
# ✅ Patch direct de la méthode `download_file` dans Client
def safe_download_file(self, remote_path, local_path, progress=None, progress_args=()): def safe_download_file(self, remote_path, local_path, progress=None, progress_args=()):
response = self.execute_request(action='download', path=remote_path, headers={'accept-encoding': 'identity'}) response = self.execute_request(action='download', path=remote_path, headers={'accept-encoding': 'identity'})
total = response.headers.get('content-length') content_length = response.headers.get('content-length')
if total is None: if content_length is None:
log_debug(f"⚠️ Skipping file without content-length: {remote_path}") print(f"⚠️ Skipping file without content-length: {remote_path}")
return return
total = int(total) total = int(content_length)
os.makedirs(os.path.dirname(local_path), exist_ok=True)
with open(local_path, 'wb') as f: with open(local_path, 'wb') as f:
for chunk in response.iter_content(chunk_size=1024): for chunk in response.iter_content(chunk_size=1024):
if chunk: if chunk:
f.write(chunk) f.write(chunk)
log_debug(f"✅ Downloaded file: {remote_path} → {local_path}")
WebDavClient.download_file = safe_download_file # 🔁 Surcharge
Client.download_file = safe_download_file
app = Flask(__name__) app = Flask(__name__)