pssecret-server/src/rectes/main.py

28 lines
541 B
Python
Raw Normal View History

2022-06-12 10:47:07 +00:00
from fastapi import FastAPI
from pydantic import BaseModel
from rectes.redis_db import redis
from rectes.utils import get_new_key
2022-06-12 10:47:07 +00:00
app = FastAPI()
class Secret(BaseModel):
data: str
2022-12-17 13:33:26 +00:00
class SecretSaveResult(BaseModel):
status: str
retrieval_url: str
@app.post("/secret", response_model=SecretSaveResult)
async def set_secret(data: Secret):
new_key = await get_new_key()
await redis.setex(new_key, 60 * 60 * 24, data.data)
return {
"status": "saved",
"retrieval_url": f"/secret/{new_key}",
}