from database.tinformationsource.model import TInformationSource


def create_information_source(db, information_source: TInformationSource):
    db.add(information_source)
    db.commit()
    db.refresh(information_source)
    return information_source

def get_information_source_by_id(db, information_source_id: int):
    return db.query(TInformationSource).filter(TInformationSource.id == information_source_id).first()

def get_active_information_sources(db) -> list:
    return db.query(TInformationSource).filter(TInformationSource.active == True).all()

def update_information_source(db, information_source_id: int, updates: dict):
    update_information = db.query(TInformationSource).filter(TInformationSource.id == information_source_id).first()
    if update_information:
        for key, value in updates.items():
            setattr(update_information, key, value)
        db.commit()
        db.refresh(update_information)
    return update_information


def delete_update_information(db, information_source_id: int):
    update_information = db.query(TInformationSource).filter(TInformationSource.id == information_source_id).first()
    if update_information:
        db.delete(update_information)
        db.commit()
    return update_information