mirror of
https://github.com/lrnselfreliance/wrolpi.git
synced 2026-10-26 19:01:39 -04:00
67 lines
1.8 KiB
Python
67 lines
1.8 KiB
Python
import pathlib
|
|
from typing import List
|
|
from urllib.request import Request
|
|
|
|
from sanic import response
|
|
|
|
from wrolpi.common import get_media_directory
|
|
from wrolpi.errors import InvalidFile
|
|
from wrolpi.root_api import get_blueprint, json_response
|
|
from wrolpi.schema import validate_doc
|
|
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('/')
|
|
@validate_doc(
|
|
'List files in a directory',
|
|
consumes=schema.FilesRequest,
|
|
)
|
|
async def get_files(_: Request, data: dict):
|
|
directories = data.get('directories') or []
|
|
|
|
files = lib.list_files(directories)
|
|
files = paths_to_files(files)
|
|
return json_response({'files': files})
|
|
|
|
|
|
@bp.post('/delete')
|
|
@validate_doc(
|
|
'Delete a single file. Returns an error if WROL Mode is enabled.',
|
|
consumes=schema.DeleteRequest,
|
|
)
|
|
async def delete_file(_: Request, data: dict):
|
|
file = data.get('file')
|
|
if not file:
|
|
raise InvalidFile('file cannot be empty')
|
|
lib.delete_file(file)
|
|
return response.empty()
|