Initial Codebase

from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
from datetime import datetime
from typing import Optional
 
from models import Journey, Scooter, Ticket
from request_models import OffboardRequest
 
app = FastAPI()
 
@app.post("/offboard-scooter")
async def offboard_scooter(request: OffboardRequest):
try: # Get journey and validate it belongs to the user and scooter
journey = await Journey.get(
id=request.journey_id,
user_id=request.user_id,
scooter_id=request.scooter_id,
status=Journey.STARTED
)
 
        if not journey:
            raise HTTPException(status_code=404, detail="Journey not found")
 
        # Update journey status
        journey.status = Journey.COMPLETED_REVIEWING
        journey.end_time = datetime.now()
        await journey.save()
 
        # Mark scooter as available
        scooter = journey.scooter
        scooter.is_available = True
        await scooter.save()
 
        ###########################
        # TODO: Create a Ticket entity for the user if the scooter is not parked properly
        # according to the offboarding photo
        ###########################
 
        return {
            'status': 'success',
            'message': 'Scooter offboarded application got successfully',
            'journey_id': request.journey_id
        }
 
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))
 

from pydantic import BaseModel, Field
 
class OffboardRequest(BaseModel):
    user_id: str
    scooter_id: str
    journey_id: str
 

from pydantic import BaseModel
from datetime import datetime
from typing import Optional
from beanie import Document, Indexed
 
class Scooter(Document):
id: str = Indexed()
is_available: bool = True
last_location: Optional[str] = None
 
    class Settings:
        name = "scooters"
 
class Journey(Document):
STARTED = 'started'
COMPLETED = 'completed'
COMPLETED_REVIEWING = 'completed_reviewing'
COMPLETED_TICKET = 'completed_ticket'
 
    id: str = Indexed()
    user_id: str
    scooter_id: str
    status: str = STARTED
    start_time: datetime = datetime.now()
    end_time: Optional[datetime] = None
 
    class Settings:
        name = "journeys"
 
class Ticket(Document):
id: str = Indexed()
user_id: str
journey_id: str
amount: float
created_at: datetime = datetime.now()
 
    class Settings:
        name = "tickets"