Skip to content

Attributes

Classes

spark_expectations.notifications.plugins.teams.SparkExpectationsTeamsPluginImpl

Bases: SparkExpectationsNotification

This class implements/supports functionality to send teams notification

Functions

send_notification(_context: SparkExpectationsContext, _config_args: Dict[Union[str], Union[str, bool]]) -> None

function to send the teams notification for assigned channel Args: _context: SparkExpectationsContext class object _config_args: dict

Returns: None

Source code in spark_expectations/notifications/plugins/teams.py 
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
@spark_expectations_notification_impl
def send_notification(
    self,
    _context: SparkExpectationsContext,
    _config_args: Dict[Union[str], Union[str, bool]],
) -> None:
    """
    function to send the teams notification for assigned channel
    Args:
        _context: SparkExpectationsContext class object
        _config_args: dict

    Returns: None

    """
    try:
        if _context.get_enable_teams is True:
            # payload = {"token": "{token}", "channel": kwargs['channel'], "text": kwargs['message']}

            message = _config_args.get("message")

            # Format Message for Teams
            if isinstance(message, str):
                message = message.replace("\n", "\n\n").replace("            ", "")

            payload = {
                "title": "SE Notification",
                "themeColor": "008000",
                "text": message,
            }

            response = requests.post(_context.get_teams_webhook_url, json=payload, timeout=10)

            # Check the response for success or failure
            if response.status_code == 200:
                _log.info("Message posted successfully!")
            else:
                _log.info("Failed to post message")
                raise SparkExpectationsTeamsNotificationException(
                    "error occurred while sending teams notification from spark expectations project"
                )

    except Exception as e:
        raise SparkExpectationsTeamsNotificationException(e)