Spaces:
Paused
Paused
File size: 1,198 Bytes
ec17af3 | 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 | from fastapi import APIRouter, Depends, HTTPException, Response, status
from sqlalchemy.orm import Session
from app import models, schemas
from app.auth import hash_password, issue_token_pair
from app.db import get_db
from app.rate_limit import rate_limit
router = APIRouter(prefix="/setup", tags=["setup"])
@router.get("/status", response_model=schemas.SetupStatus)
def setup_status(db: Session = Depends(get_db)):
user_exists = db.query(models.User.id).first() is not None
return {"setup_required": not user_exists}
@router.post(
"",
response_model=schemas.Token,
status_code=status.HTTP_201_CREATED,
dependencies=[Depends(rate_limit(5))],
)
def initial_setup(payload: schemas.UserCreate, response: Response, db: Session = Depends(get_db)):
user_exists = db.query(models.User.id).first() is not None
if user_exists:
raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="Setup already completed")
user = models.User(
email=payload.email,
hashed_password=hash_password(payload.password),
is_admin=True,
)
db.add(user)
db.commit()
db.refresh(user)
return issue_token_pair(db, response, user)
|