File size: 909 Bytes
cce8120 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 | from sqlalchemy.orm import Session
from . import models, schemas
def get_test(db: Session, test_id: int):
return db.query(models.Test).filter(models.Test.id == test_id).first()
def get_tests(db: Session, skip: int = 0, limit: int = 100):
return db.query(models.Test).offset(skip).limit(limit).all()
def create_test(db: Session, test: schemas.TestCreate):
db_test = models.Test(name=test.name, description=test.description)
db.add(db_test)
db.commit()
db.refresh(db_test)
return db_test
def update_test(db: Session, db_test: models.Test, updates: schemas.TestUpdate):
if updates.name is not None:
db_test.name = updates.name
if updates.description is not None:
db_test.description = updates.description
db.commit()
db.refresh(db_test)
return db_test
def delete_test(db: Session, db_test: models.Test):
db.delete(db_test)
db.commit()
|