| from fastapi import FastAPI, Request, HTTPException, UploadFile, File |
| from fastapi.responses import HTMLResponse, FileResponse, RedirectResponse |
| from fastapi.staticfiles import StaticFiles |
| from fastapi.templating import Jinja2Templates |
| import os |
| from typing import List |
|
|
| root_path = os.getcwd() |
| abs_path = './all_file' |
| upload_path = os.path.join(root_path,'all_file') |
| varlist = ['folder','image','audio','video','pdf','file'] |
|
|
| def retlist(path=""): |
| all_list = [] |
| for i in range(0,7): |
| all_list.append([]) |
|
|
| new_path = os.path.join(abs_path,path) |
| files = os.listdir(os.path.join(abs_path,path)) |
| |
| files.sort() |
| for f in files: |
| if os.path.isdir(os.path.join(new_path,f)): |
| all_list[0].append(f) |
| if os.path.isfile(os.path.join(new_path,f)): |
| filetype = os.path.splitext(os.path.join(new_path,f)) |
| if [x for x in ['jpg','png','bmp','jpeg','gif'] if x in filetype[1]]: |
| all_list[1].append(f) |
| elif [x for x in ['mp3','wav'] if x in filetype[1]]: |
| all_list[2].append(f) |
| elif [x for x in ['mp4','mkv','mpg','mpeg','avi'] if x in filetype[1]]: |
| all_list[3].append(f) |
| elif [x for x in ['pdf'] if x in filetype[1]]: |
| all_list[4].append(f) |
| else: |
| all_list[5].append(f) |
| return all_list |
|
|
|
|
| |
| app = FastAPI() |
|
|
| |
| app.mount("/static", StaticFiles(directory=os.path.join(root_path,'static')), name="static") |
|
|
| |
| templates = Jinja2Templates(directory=os.path.join(root_path,'templates')) |
|
|
| @app.get("/", response_class=HTMLResponse) |
| async def index(request: Request): |
| lst = retlist() |
| return templates.TemplateResponse("index.html", { |
| "request": request, |
| "path": "", |
| "all_list": lst, |
| "varlist": varlist |
| }) |
|
|
| @app.get("/{path:path}", response_class=HTMLResponse) |
| async def dir_listing(path: str, request: Request): |
| full_path = os.path.join(abs_path, path) |
| |
| if os.path.isdir(full_path): |
| lst = retlist(path) |
| return templates.TemplateResponse("index.html", { |
| "request": request, |
| "path": path, |
| "all_list": lst, |
| "varlist": varlist |
| }) |
| else: |
| if os.path.exists(full_path): |
| return FileResponse(full_path) |
| else: |
| raise HTTPException(status_code=404, detail="File not found") |
|
|
| ''' |
| # Uncomment if you need this functionality |
| import pyotp, glob |
| |
| @app.get("/rm/{code}/{path:path}") |
| async def rm_file(code: str, path: str): |
| otp_chk = pyotp.TOTP('') |
| if otp_chk.now() == code: |
| files = glob.glob(f'{abs_path}/{path}') |
| for file in files: |
| os.remove(file) |
| return RedirectResponse(url="/") |
| ''' |
| @app.post("/uploadfiles") |
| async def upload_image(file: List[UploadFile] = File(...)): |
| for fi in file: |
| file_path = os.path.join(abs_path, fi.filename) |
| with open(file_path, "wb") as f: |
| f.write(await fi.read()) |
| return RedirectResponse(url="/", status_code=303) |
|
|
| |
| |
| |
| |
|
|
