26 lines
807 B
Python
26 lines
807 B
Python
# Entity: Migration
|
|
|
|
```python
|
|
from fastapi import APIRouter, Depends, HTTPException, status
|
|
from sqlalchemy.orm import Session
|
|
from core.database import get_db
|
|
from models.migration import Migration
|
|
from schemas.migration import MigrationSchema, MigrationFix
|
|
from helpers.migration_helpers import fix_migration
|
|
|
|
router = APIRouter()
|
|
|
|
@router.post("/migrations/fix", status_code=200, response_model=MigrationSchema)
|
|
async def fix_migration_endpoint(
|
|
migration_data: MigrationFix,
|
|
db: Session = Depends(get_db)
|
|
):
|
|
"""Fix a database migration"""
|
|
fixed_migration = fix_migration(db, migration_data)
|
|
if not fixed_migration:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|
detail="Failed to fix migration"
|
|
)
|
|
return fixed_migration
|
|
``` |