allow to edit files
This commit is contained in:
+3
-52
@@ -1,55 +1,14 @@
|
||||
from my_modules.decoratory.header import login_required
|
||||
from my_modules.expiry import parse_expires, ensure_utc
|
||||
from my_modules.file_meta import iso_stamp_filename, format_size
|
||||
|
||||
from quart import Blueprint, request, jsonify, current_app
|
||||
from datetime import datetime, timedelta, timezone
|
||||
import aiofiles, asyncio, re
|
||||
import asyncio
|
||||
|
||||
upload_bp = Blueprint("upload_bp", __name__)
|
||||
|
||||
# --- Helpers -----------------------------------------------------
|
||||
|
||||
PRESET_H = re.compile(r"^(\d+)h$")
|
||||
PRESET_D = re.compile(r"^(\d+)d$")
|
||||
|
||||
def iso_stamp_filename(prefix: str, ext: str) -> str:
|
||||
"""Generate timestamped filename, e.g. pasted-2025-10-23T121212Z.png"""
|
||||
ts = datetime.now(timezone.utc).isoformat()
|
||||
ts = ts.replace(":", "").split(".")[0]
|
||||
if ts.endswith("+00:00"):
|
||||
ts = ts.replace("+00:00", "Z")
|
||||
return f"{prefix}-{ts}.{ext}"
|
||||
|
||||
def parse_expires(value: str | None) -> datetime | None:
|
||||
"""Parse expiration presets or ISO datetime."""
|
||||
if not value:
|
||||
return None
|
||||
value = value.strip()
|
||||
if m := PRESET_H.match(value):
|
||||
return datetime.now(timezone.utc) + timedelta(hours=int(m.group(1)))
|
||||
if m := PRESET_D.match(value):
|
||||
return datetime.now(timezone.utc) + timedelta(days=int(m.group(1)))
|
||||
try:
|
||||
return datetime.fromisoformat(value.replace("Z", "+00:00")).astimezone(timezone.utc)
|
||||
except Exception:
|
||||
return None
|
||||
|
||||
def format_size(num_bytes: int) -> str:
|
||||
"""Return a human-readable file size (e.g., '2.3 MB', '10 Bytes')."""
|
||||
if num_bytes < 1024:
|
||||
return f"{num_bytes} Byte{'s' if num_bytes != 1 else ''}"
|
||||
|
||||
units = ["KB", "MB", "GB", "TB", "PB", "EB"]
|
||||
size = float(num_bytes)
|
||||
for unit in units:
|
||||
size /= 1024.0
|
||||
if size < 1024.0 or unit == units[-1]:
|
||||
# 1 decimal place; drop trailing .0 (optional)
|
||||
val = f"{size:.1f}"
|
||||
if val.endswith(".0"):
|
||||
val = val[:-2]
|
||||
return f"{val} {unit}"
|
||||
return f"{num_bytes} Bytes" # fallback
|
||||
|
||||
async def read_all(uploaded) -> bytes:
|
||||
"""Read all bytes from an uploaded file, handling sync or async .read()."""
|
||||
reader = getattr(uploaded, "read", None)
|
||||
@@ -63,14 +22,6 @@ async def read_all(uploaded) -> bytes:
|
||||
return await data
|
||||
return data
|
||||
|
||||
def ensure_utc(dt:datetime):
|
||||
"""Ensure a timezone-aware UTC datetime or None."""
|
||||
if dt is None:
|
||||
return None
|
||||
if dt.tzinfo is None:
|
||||
return dt.replace(tzinfo=timezone.utc)
|
||||
return dt.astimezone(timezone.utc)
|
||||
|
||||
# --- Routes ------------------------------------------------------
|
||||
|
||||
@upload_bp.post("/api/upload")
|
||||
|
||||
Reference in New Issue
Block a user