From 4156b5ef02481e096a5b58d8910d44bb196f7b14 Mon Sep 17 00:00:00 2001 From: Backend IM Bot Date: Sat, 12 Apr 2025 02:53:45 +0000 Subject: [PATCH] feat: Updated endpoint endpoints/fruits.get.py via AI --- .../20250412_025343_aa6601e9_update_fruit.py | 21 +++++++++++++++++++ endpoints/fruits.get.py | 7 ++----- 2 files changed, 23 insertions(+), 5 deletions(-) create mode 100644 alembic/versions/20250412_025343_aa6601e9_update_fruit.py diff --git a/alembic/versions/20250412_025343_aa6601e9_update_fruit.py b/alembic/versions/20250412_025343_aa6601e9_update_fruit.py new file mode 100644 index 0000000..7fd542d --- /dev/null +++ b/alembic/versions/20250412_025343_aa6601e9_update_fruit.py @@ -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) + ) \ No newline at end of file diff --git a/endpoints/fruits.get.py b/endpoints/fruits.get.py index 64e8e96..723a551 100644 --- a/endpoints/fruits.get.py +++ b/endpoints/fruits.get.py @@ -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 \ No newline at end of file