api done

gramup/main.py

@@ -1,13 +1,17 @@
 import asyncio
+import glob
+import os
+from typing import List, Literal
 
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException, Request
 from starlette import status
 from starlette.middleware.cors import CORSMiddleware
 from starlette.staticfiles import StaticFiles
 from starlette.websockets import WebSocket
 from telethon import TelegramClient
 
 from constants import API_ID, API_HASH
+from gramup.models import File, Task, TaskRequest
 
 client = TelegramClient('anon', API_ID, API_HASH)
 app = FastAPI()
@@ -48,11 +52,35 @@ async def websocket_endpoint(websocket: WebSocket):
 
 
 @app.get("/api/files/")
-async def files(path=""):
+async def files(path="") -> List[File]:
     if not client.is_connected() or not await client.is_user_authorized():
         raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail="User not authorized")
 
     return []
 
 
+@app.get("/api/local_files/")
+async def local_files(path="") -> List[File]:
+    return [
+        File(
+            folder=os.path.isdir(file),
+            name=os.path.basename(file),
+            path=file,
+            id=""
+        )
+        for file in glob.glob(f"{path}/*")
+    ]
+
+
+@app.post("/api/tasks/")
+async def tasks(request: Request) -> Task:
+    body = TaskRequest(**(await request.json()))
+
+    return Task(
+        id="",
+        name="",
+        status="running",
+    )
+
+
 app.mount("/", StaticFiles(directory="static"), name="static")

gramup/models.py

@@ -0,0 +1,19 @@
+from typing import Literal, TypedDict
+
+from pydantic import BaseModel
+
+
+class File(BaseModel):
+    folder: bool
+    name: str
+    path: str
+    id: str
+
+
+class Task(BaseModel):
+    id: str
+    name: str
+    status: Literal["running", "stopped", "finished"]
+
+
+TaskRequest = TypedDict("TaskRequest", {"path": str, "type": Literal["sync", "upload"]})

requirements.txt

@@ -1,3 +1,5 @@
 fastapi==0.96.0
 starlette~=0.27.0
 Telethon~=1.28.5
+
+pydantic~=1.10.8