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 typing import List
from core.database import get_db
@ -7,10 +7,7 @@ from helpers.fruit_helpers import get_all_fruits
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)):
fruits = get_all_fruits(db)
for fruit in fruits:
if hasattr(fruit, 'is_ripe'):
delattr(fruit, 'is_ripe')
return fruits