Spaces:
Sleeping
Sleeping
File size: 1,393 Bytes
9dc6849 8083ca2 9dc6849 5ac84de 86614b9 9dc6849 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
from fastapi import FastAPI, Request
from fastapi.staticfiles import StaticFiles
from fastapi.responses import HTMLResponse
from itables import to_html_datatable
from pathlib import Path
import pandas as pd
ROOT = Path(".")
app = FastAPI()
app.mount(path="/", app=StaticFiles(directory=ROOT), name="HOME")
@app.middleware("http")
async def file_system(request: Request, call_next):
url = request.url
if url.path.endswith("/"):
return HTMLResponse(content=files_in_folder(url.path.lstrip("/")))
response = await call_next(request)
return response
def files_in_folder(path: str):
"""List files to render as file index."""
folder = ROOT / path
print(folder)
path_glob = folder.glob("*")
res = pd.DataFrame(
[
(
f.name,
f.stat().st_size,
pd.Timestamp(int(f.stat().st_mtime), unit="s"),
f.stat().st_nlink > 1,
)
for f in path_glob
],
columns=["path", "size", "mtime", "folder"],
).sort_values(["folder"], ascending=False)
res["path"] = res.apply(
lambda x: f"<a href=/{path}{x.path}{'/' if x.folder else ''}>{x.path}</a>",
axis=1,
)
return f"<h1>{folder}</h1><br>" + to_html_datatable(
df=res.drop(columns="folder"),
classes="display compact",
showIndex=False,
)
|