# Entity: Drug

from fastapi import APIRouter, Depends, HTTPException, status
from app.models import Drug
from app.schemas import DrugSchema
from sqlalchemy.orm import Session
from app.api.db.database import get_db

router = APIRouter()

@router.get("/cocaine", response_model=list[DrugSchema])
async def get_hard_drugs(db: Session = Depends(get_db)):
    hard_drugs = db.query(Drug).filter(Drug.is_hard == True).all()
    return hard_drugs