Function _dm_store_file

Scripts/ButSystem.py:12475–12507

Store a regular DM attachment (plaintext on disk).

(dm_id: int, file_storage)

    return redirect(url_for("files", p=rel))


UPLOAD_LOCK = threading.RLock()

def _upload_meta_path(upload_id: str) -> str: return os.path.join(TMP_UPLOAD_DIR, f"{upload_id}.json")
def _upload_tmp_path(upload_id: str) -> str: return os.path.join(TMP_UPLOAD_DIR, f"{upload_id}.part")

def _cleanup_stale_file_uploads(max_age: int = 24 * 60 * 60) -> None:
    cutoff = time.time() - max_age
    os.makedirs(TMP_UPLOAD_DIR, exist_ok=True)
    for name in os.listdir(TMP_UPLOAD_DIR):
        if not (name.endswith(".part") or name.endswith(".json") or name.startswith("vault-download-")): continue
        path = os.path.join(TMP_UPLOAD_DIR, name)
        try:
            if os.path.getmtime(path) < cutoff: os.remove(path)
        except Exception: pass


@app.route("/api/upload/init", methods=["POST"])
@login_required
def api_upload_init():
    _cleanup_stale_file_uploads()
    data = request.get_json(silent=True) or {}
    try:
        rel = safe_relpath(data.get("p", "") or ""); filename = _validate_vault_upload_filename((data.get("filename") or "").strip()); total = int(data.get("total_chunks") or 0); total_size = int(data.get("total_size") or 0)
    except Exception: return jsonify({"ok": False, "error": "bad_request"}), 400
    conflict = (data.get("conflict") or "keep").lower()
    if conflict not in ("keep", "replace", "cancel"): conflict = "keep"
    expected = max(1, (total_size + FILE_UPLOAD_CHUNK_BYTES - 1) // FILE_UPLOAD_CHUNK_BYTES)
    if not filename or total != expected or total > 5000 or total_size < 0: return jsonify({"ok": False, "error": "bad_request"}), 400
    if total_size > FILES_MAX_BYTES: return jsonify({"ok": False, "error": "too_large", "max": FILES_MAX_BYTES}), 413
    dest_dir = abs_user_path(current_user(), rel); os.makedirs(dest_dir, exist_ok=True)
    original = os.path.join(dest_dir, filename)
    if os.path.exists(original):
        if conflict == "cancel": return jsonify({"ok": False, "error": "exists"}), 409
        destination = original if conflict == "replace" else _unique_destination(dest_dir, filename)
    else: destination = original