allow to edit files
This commit is contained in:
+86
-7
@@ -2,10 +2,27 @@ from my_modules.decoratory.header import login_required, feature_flag_required
|
||||
from my_modules.functions import get_ip
|
||||
from my_modules.app.setup import LIMITER
|
||||
from my_modules.app.logger import logger
|
||||
from my_modules.expiry import parse_expires
|
||||
|
||||
from quart import Blueprint, request, session, Response, send_file, render_template, abort, current_app
|
||||
from quart import (
|
||||
Blueprint,
|
||||
request,
|
||||
session,
|
||||
Response,
|
||||
send_file,
|
||||
render_template,
|
||||
abort,
|
||||
current_app,
|
||||
jsonify,
|
||||
)
|
||||
|
||||
side_main_bp = Blueprint('side_main', __name__)
|
||||
side_main_bp = Blueprint("side_main", __name__)
|
||||
|
||||
def find_file(files: list[dict], file_id: str):
|
||||
for file_data in files:
|
||||
if file_data.get("file_id") == file_id:
|
||||
return file_data
|
||||
return None
|
||||
|
||||
@side_main_bp.route('/')
|
||||
@LIMITER.limit("10 per minute;50 per hour")
|
||||
@@ -36,15 +53,77 @@ async def files_list(user):
|
||||
@login_required
|
||||
@feature_flag_required("nanoshare_files-info", fallback=False, status_code=404)
|
||||
async def file_info(file_id, user):
|
||||
files_data = await current_app.convex.get_files(user_id=user['sub'])
|
||||
return await render_template("views/webpage/files/info.htm", files=files_data)
|
||||
files_data = await current_app.convex.get_files(user_id=user["sub"])
|
||||
file_data = find_file(files_data, file_id)
|
||||
if not file_data:
|
||||
abort(404)
|
||||
|
||||
@side_main_bp.route('/files/<path:file_id>/edit')
|
||||
access_data = await current_app.convex.get_file_access(file_id=file_id) or []
|
||||
share_url = request.url_root.rstrip("/") + f"/-{file_id}"
|
||||
return await render_template(
|
||||
"views/webpage/files/info.htm",
|
||||
file=file_data,
|
||||
accesses=access_data,
|
||||
share_url=share_url,
|
||||
)
|
||||
|
||||
@side_main_bp.route("/files/<path:file_id>/edit")
|
||||
@login_required
|
||||
@feature_flag_required("nanoshare_files-edit", fallback=False, status_code=404)
|
||||
async def file_edit(file_id, user):
|
||||
files_data = await current_app.convex.get_files(user_id=user['sub'])
|
||||
return await render_template("views/webpage/files/edit.htm", files=files_data)
|
||||
files_data = await current_app.convex.get_files(user_id=user["sub"])
|
||||
file_data = find_file(files_data, file_id)
|
||||
if not file_data:
|
||||
abort(404)
|
||||
|
||||
share_url = request.url_root.rstrip("/") + f"/-{file_id}"
|
||||
return await render_template(
|
||||
"views/webpage/files/edit.htm", file=file_data, share_url=share_url
|
||||
)
|
||||
|
||||
@side_main_bp.post("/api/files/<path:file_id>/edit")
|
||||
@login_required
|
||||
@feature_flag_required("nanoshare_files-edit", fallback=False, status_code=404)
|
||||
async def file_edit_api(file_id, user):
|
||||
files_data = await current_app.convex.get_files(user_id=user["sub"])
|
||||
if not find_file(files_data, file_id):
|
||||
return jsonify({"ok": False, "error": "File not found"}), 404
|
||||
|
||||
payload = await request.get_json(silent=True)
|
||||
if payload is None:
|
||||
payload = await request.form
|
||||
|
||||
file_name = str(payload.get("file_name", "")).strip()
|
||||
note = str(payload.get("note", "")).strip()
|
||||
expires_raw = str(payload.get("expires", "")).strip()
|
||||
|
||||
if not file_name:
|
||||
return jsonify({"ok": False, "error": "Filename is required"}), 400
|
||||
|
||||
expires_at = parse_expires(expires_raw)
|
||||
if expires_raw and expires_raw != "never" and expires_at is None:
|
||||
return jsonify({"ok": False, "error": "Invalid expiration value"}), 400
|
||||
|
||||
await current_app.convex.update_file(
|
||||
file_id=file_id,
|
||||
file_name=file_name,
|
||||
note=note,
|
||||
expires_at=expires_at,
|
||||
user_id=user["sub"],
|
||||
)
|
||||
|
||||
return jsonify({"ok": True})
|
||||
|
||||
@side_main_bp.post("/api/files/<path:file_id>/delete")
|
||||
@login_required
|
||||
@feature_flag_required("nanoshare_files-edit", fallback=False, status_code=404)
|
||||
async def file_delete_api(file_id, user):
|
||||
files_data = await current_app.convex.get_files(user_id=user["sub"])
|
||||
if not find_file(files_data, file_id):
|
||||
return jsonify({"ok": False, "error": "File not found"}), 404
|
||||
|
||||
await current_app.convex.delete_file(file_id=file_id, user_id=user["sub"])
|
||||
return jsonify({"ok": True})
|
||||
|
||||
@side_main_bp.route("/-<file_id>")
|
||||
@LIMITER.limit("10 per minute;500 per hour;")
|
||||
|
||||
Reference in New Issue
Block a user