def create_alert_event( name, status, fingerprint, confirm, confirm_user, ): alert_event = AlertEvent( name, status, fingerprint, confirm, confirm_user,) with session_maker() as s: s.add(alert_event) s.commit() s.refresh(alert_event) return alert_event
def query_all_host(_phy=True): with session_maker() as session: res = [] for host in session.query(Host): if _phy == True: if host.node_type in PHYSICAL_HOST_JOB: res.append(host) else: res.append(host) return res
def create_alert_rule( name, group, expr, for_, costom_label, severity, scope, summary, desc): rule = AlertRule( name=name, group=group, expr=expr, for_=for_, costom_label=None if not costom_label else costom_label, severity=severity, scope=None if not scope else scope, summary=summary, desc=desc ) with session_maker() as s: s.add(rule) s.commit() s.refresh(rule) return rule
def create_host(ip, node_type, exporter_port, interval, physical, costom_label, desc, ssh_auth_t, ssh_auth, ssh_port, bmc_url): host = Host(ip=ip, node_type=node_type, state=1, exporter_port=exporter_port, physical=physical, costom_label=costom_label, description=desc, interval=interval, ssh_auth=ssh_auth, ssh_auth_type=ssh_auth_t, ssh_port=ssh_port, bmc_url=bmc_url) with session_maker() as s: s.add(host) s.commit() s.refresh(host) return host
def query_all_alert_event(): with session_maker() as session: return [alert_event for alert_event in session.query(AlertEvent)]
def confirm_alert_event_by_fingerprint(fingerprint, confirmer): with session_maker() as session: event = session.query(AlertEvent).filter(AlertEvent.fingerprint == fingerprint).first() event.confirm = True event.confirmer = confirmer session.commit()
def query_alert_event_by_fingerprint(fingerprint): with session_maker() as session: return session.query(AlertEvent).filter(AlertEvent.fingerprint == fingerprint).first()
def query_alert_event_by_name(name): with session_maker() as session: return session.query(AlertEvent).filter(AlertEvent.name == name).first()
def delete_alert_event(name): with session_maker() as session: alert_event = session.query(AlertEvent).filter(AlertEvent.name == name).first() return alert_event
def query_host(ip): with session_maker() as session: return session.query(Host).filter(Host.ip == ip).first()
def update_host(ip, state): with session_maker() as session: host = session.query(Host).filter(Host.ip == ip).first() host.state = state session.commit() return host
def delete_host(ip, node_type): with session_maker() as session: host = session.query(Host).filter(Host.ip == ip, Host.node_type == node_type).first() return host
def query_all_alert_rule(): with session_maker() as session: return [rule for rule in session.query(AlertRule)]
def query_alert_rule(name): with session_maker() as session: return session.query(AlertRule).filter(AlertRule.name == name).first()
def delete_alert_rule(name): with session_maker() as session: rule = session.query(AlertRule).filter(AlertRule.name == name).first() return rule