feat: Updated endpoint endpoints/fruits.get.py via AI

This commit is contained in:
Backend IM Bot 2025-04-12 02:53:45 +00:00
parent 7a3bb1c17e
commit 4156b5ef02
2 changed files with 23 additions and 5 deletions

View File

@ -0,0 +1,21 @@
"""remove is_ripe field from fruits table
Revision ID: d7f3e1b8c5a2
Revises: b8d5e2c6a4f9
Create Date: 2024-01-18 10:23:45.123456
"""
from alembic import op
import sqlalchemy as sa
# revision identifiers, used by Alembic.
revision = 'd7f3e1b8c5a2'
down_revision = 'b8d5e2c6a4f9'
branch_labels = None
depends_on = None
def upgrade():
op.drop_column('fruits', 'is_ripe')
def downgrade():
op.add_column('fruits',
sa.Column('is_ripe', sa.Boolean(), server_default='false', nullable=False)
)

View File

@ -1,4 +1,4 @@
from fastapi import APIRouter, Depends, status from fastapi import APIRouter, Depends
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
from typing import List from typing import List
from core.database import get_db from core.database import get_db
@ -7,10 +7,7 @@ from helpers.fruit_helpers import get_all_fruits
router = APIRouter() router = APIRouter()
@router.get("/fruits", response_model=List[FruitSchema], status_code=status.HTTP_200_OK) @router.get("/fruits", response_model=List[FruitSchema])
async def get_fruits(db: Session = Depends(get_db)): async def get_fruits(db: Session = Depends(get_db)):
fruits = get_all_fruits(db) fruits = get_all_fruits(db)
for fruit in fruits:
if hasattr(fruit, 'is_ripe'):
delattr(fruit, 'is_ripe')
return fruits return fruits