actually use alert_config

Author: evanfuller

nightfall/alerts.py

@@ -63,10 +63,10 @@ class AlertConfig:
     def as_dict(self):
         result = {}
         if self.slack:
-            result["slack"] = {"target": self.slack.target}
+            result["slack"] = self.slack.as_dict()
         if self.email:
-            result["email"] = {"address": self.email.address}
+            result["email"] = self.email.as_dict()
         if self.url:
-            result["url"] = {"address": self.url.address}
+            result["url"] = self.url.as_dict()
         return result
 

nightfall/api.py

@@ -101,6 +101,8 @@ def scan_text(self, texts: List[str], policy_uuids: List[str] = None, detection_
             policy["contextBytes"] = context_bytes
         if default_redaction_config:
             policy["defaultRedactionConfig"] = default_redaction_config.as_dict()
+        if alert_config:
+            policy["alertConfig"] = alert_config.as_dict()
 
         request_body = {
             "payload": texts
@@ -175,7 +177,8 @@ def scan_file(self, location: str, webhook_url: Optional[str] = None, policy_uui
                                         detection_rules=detection_rules,
                                         detection_rule_uuids=detection_rule_uuids,
                                         webhook_url=webhook_url, policy_uuid=policy_uuid,
-                                        request_metadata=request_metadata)
+                                        request_metadata=request_metadata,
+                                        alert_config=alert_config)
         _validate_response(response, 200)
         parsed_response = response.json()
 
@@ -222,7 +225,8 @@ def _file_scan_finalize(self, session_id: str):
 
     def _file_scan_scan(self, session_id: str, detection_rules: Optional[List[DetectionRule]] = None,
                         detection_rule_uuids: Optional[List[str]] = None, webhook_url: Optional[str] = None,
-                        policy_uuid: Optional[str] = None, request_metadata: Optional[str] = None) -> requests.Response:
+                        policy_uuid: Optional[str] = None, request_metadata: Optional[str] = None,
+                        alert_config: Optional[AlertConfig] = None) -> requests.Response:
         if policy_uuid:
             data = {"policyUUID": policy_uuid}
         else:
@@ -231,6 +235,8 @@ def _file_scan_scan(self, session_id: str, detection_rules: Optional[List[Detect
                 data["policy"]["detectionRuleUUIDs"] = detection_rule_uuids
             if detection_rules:
                 data["policy"]["detectionRules"] = [d.as_dict() for d in detection_rules]
+            if alert_config:
+                data["policy"]["alertConfig"] = alert_config.as_dict()
 
         if request_metadata:
             data["requestMetadata"] = request_metadata