def get_all_node_threats(db: Session = Depends(get_db)):
return crud.read_all(db_table=NodeThreat, db=db)
def get_all_observables(db: Session = Depends(get_db)):
return crud.read_all(db_table=Observable, db=db)
def get_all_event_statuses(db: Session = Depends(get_db)):
return crud.read_all(db_table=EventStatus, db=db)
sys.exit()
data = None
with open("etc/defaults.yml", "r") as yamlfile:
data = yaml.load(yamlfile, Loader=yaml.FullLoader)
# Quit early if the config file is empty
if data is None:
print("etc/defaults.yml is empty!")
sys.exit()
# Get a database connection
db: Session = next(get_db())
# Add the objects to the database but only if their respective tables do not have any items already.
if "alert_disposition" in data and not crud.read_all(db_table=AlertDisposition,
db=db):
for rank, value in enumerate(data["alert_disposition"]):
db.add(AlertDisposition(rank=rank, value=value))
print(f"Adding alert disposition: {rank}:{value}")
if not crud.read_all(db_table=AlertQueue, db=db):
if "alert_queue" in data:
# Make sure there is always a "default" queue
if "default" not in data["alert_queue"]:
data["alert_queue"].append("default")
for value in data["alert_queue"]:
db.add(AlertQueue(value=value))
print(f"Adding alert queue: {value}")
else:
# Make sure there is always a "default" queue
def get_all_alert_tools(db: Session = Depends(get_db)):
return crud.read_all(db_table=AlertTool, db=db)
def get_all_user_roles(db: Session = Depends(get_db)):
return crud.read_all(db_table=UserRole, db=db)
def get_all_node_history_actions(db: Session = Depends(get_db)):
return crud.read_all(db_table=NodeHistoryAction, db=db)
def get_all_dispositions(db: Session = Depends(get_db)):
return crud.read_all(db_table=AlertDisposition, db=db)
def get_all_event_prevention_tools(db: Session = Depends(get_db)):
return crud.read_all(db_table=EventPreventionTool, db=db)
def get_all_event_risk_levels(db: Session = Depends(get_db)):
return crud.read_all(db_table=EventRiskLevel, db=db)
def get_all_analysis_module_types(db: Session = Depends(get_db)):
return crud.read_all(db_table=AnalysisModuleType, db=db)
def get_all_event_sources(db: Session = Depends(get_db)):
return crud.read_all(db_table=EventSource, db=db)
def get_all_alert_queues(db: Session = Depends(get_db)):
return crud.read_all(db_table=AlertQueue, db=db)
def get_all_event_remediations(db: Session = Depends(get_db)):
return crud.read_all(db_table=EventRemediation, db=db)
def get_all_event_vectors(db: Session = Depends(get_db)):
return crud.read_all(db_table=EventVector, db=db)
def get_all_event_types(db: Session = Depends(get_db)):
return crud.read_all(db_table=EventType, db=db)