mirror of
https://github.com/lrnselfreliance/wrolpi.git
synced 2026-10-26 10:51:36 -04:00
63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
import pathlib
|
|
from typing import List
|
|
from urllib.request import Request
|
|
|
|
from sanic import response
|
|
from sanic_ext import validate
|
|
from sanic_ext.extensions.openapi import openapi
|
|
|
|
from wrolpi.common import get_media_directory
|
|
from wrolpi.errors import InvalidFile
|
|
from wrolpi.root_api import get_blueprint, json_response
|
|
from . import lib, schema
|
|
|
|
bp = get_blueprint('Files', '/api/files')
|
|
|
|
|
|
def paths_to_files(paths: List[pathlib.Path]):
|
|
"""Convert Paths to what the React UI expects."""
|
|
media_directory = get_media_directory()
|
|
new_files = []
|
|
for path in paths:
|
|
stat = path.stat()
|
|
key = path.relative_to(media_directory)
|
|
modified = stat.st_mtime
|
|
if path.is_dir():
|
|
key = f'{key}/'
|
|
new_files.append(dict(
|
|
key=key,
|
|
modified=modified,
|
|
url=key,
|
|
name=path.name,
|
|
))
|
|
else:
|
|
# A File should know it's size.
|
|
new_files.append(dict(
|
|
key=key,
|
|
modified=modified,
|
|
size=stat.st_size,
|
|
url=key,
|
|
name=path.name,
|
|
))
|
|
return new_files
|
|
|
|
|
|
@bp.post('/')
|
|
@openapi.description('List files in a directory')
|
|
@validate(schema.FilesRequest)
|
|
async def get_files(_: Request, body: schema.FilesRequest):
|
|
directories = body.directories or []
|
|
|
|
files = lib.list_files(directories)
|
|
files = paths_to_files(files)
|
|
return json_response({'files': files})
|
|
|
|
|
|
@bp.post('/delete')
|
|
@openapi.description('Delete a single file. Returns an error if WROL Mode is enabled.')
|
|
@validate(schema.DeleteRequest)
|
|
async def delete_file(_: Request, body: schema.DeleteRequest):
|
|
if not body.file:
|
|
raise InvalidFile('file cannot be empty')
|
|
lib.delete_file(body.file)
|
|
return response.empty()
|