def get_all(session: FileOperator, hardware: str, base: str, test_type: str) -> list: if test_type != 'all': return session.query(Types.Test())\ .filter(lambda Test: (Test.base == base and Test.hardware == hardware and Test.type == test_type))\ .all() else: return session.query(Types.Test())\ .filter(lambda Test: (Test.base == base and Test.hardware == hardware))\ .all()
def update(session: FileOperator, name: str, hardware: str, base: str, type: str, definition: dict, is_enabled: bool): test = session.query(Types.Test())\ .filter(lambda Test: (Test.name == name and Test.hardware == hardware and Test.base == base))\ .one() test.definition = definition test.is_enabled = is_enabled session.commit()
def get(session: FileOperator, name: str, hardware: str, base: str) -> DBObject: return session.query(Types.Test())\ .filter(lambda Test: (Test.name == name and Test.hardware == hardware and Test.base == base))\ .one()
def get_all_for_hardware(session: FileOperator, hardware: str) -> list: return session.query(Types.Test()).filter(lambda Test: Test.hardware == hardware).all()
def add(session: FileOperator, name: str, hardware: str, base: str, test_type: str, definition: dict, is_enabled: bool): test = {"name": name, "hardware": hardware, "base": base, "type": test_type, "definition": definition, "is_enabled": is_enabled} session.query(Types.Test()).add(test) session.commit()
def remove(session: FileOperator, name: str): session.query(Types.Test())\ .filter(lambda Test: Test.name == name)\ .delete() session.commit()