Spaces:
Running
Running
File size: 1,093 Bytes
1a56106 e0b4f7e a3bb72f e0b4f7e 1a56106 e0b4f7e a3bb72f |
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 51 |
from typing import Union
import fastapi
import pydantic
import networkx as nx
class Position(pydantic.BaseModel):
x: float
y: float
class WorkspaceNode(pydantic.BaseModel):
id: str
title: str
type: str
position: Position
class WorkspaceConnection(pydantic.BaseModel):
id: str
# Baklava.js calls it "from", but that's a reserved keyword in Python.
src: str = pydantic.Field(None, alias='from')
dst: str = pydantic.Field(None, alias='to')
class WorkspaceGraph(pydantic.BaseModel):
nodes: list[WorkspaceNode]
connections: list[WorkspaceConnection]
panning: Position
scaling: float
nodes: list[WorkspaceNode]
class Workspace(pydantic.BaseModel):
graph: WorkspaceGraph
app = fastapi.FastAPI()
@app.get("/")
def read_root():
return {"Hello": "World"}
@app.get("/items/{item_id}")
def read_item(item_id: int, q: Union[str, None] = None):
return {"item_id": item_id, "q": q}
@app.post("/api/save")
def save(ws: Workspace):
print(ws)
G = nx.scale_free_graph(100)
return {"graph": list(nx.to_edgelist(G))}
|