32 lines
1.3 KiB
Python
32 lines
1.3 KiB
Python
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
|