import peter

database/tinformationsource/curd.py

@@ -0,0 +1,31 @@
+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