def slack_webhook_hook_demo():
slack = SlackWebhookHook(http_conn_id='slack',
message='Hello, world! (SlackWebhookHook)',
username='******',
icon_emoji=':cat:',
link_names=True)
slack.execute()
def test_url_generated_by_endpoint(self, request_mock):
hook = SlackWebhookHook(webhook_token=self.expected_url)
try:
hook.execute()
except MissingSchema:
pass
request_mock.assert_called_once_with(self.expected_method,
self.expected_url,
headers=mock.ANY,
data=mock.ANY)
request_mock.reset_mock()
def test_url_generated_by_http_conn_id(self, request_mock):
hook = SlackWebhookHook(http_conn_id='slack-webhook-url')
try:
hook.execute()
except MissingSchema:
pass
request_mock.assert_called_once_with(self.expected_method,
self.expected_url,
headers=mock.ANY,
data=mock.ANY)
request_mock.reset_mock()
def test_url_generated_by_http_conn_id_and_endpoint(
self, request_mock, session_mock):
hook = SlackWebhookHook(http_conn_id='slack-webhook-host',
webhook_token='B000/XXX')
try:
hook.execute()
except MissingSchema:
pass
request_mock.assert_called_once_with(self.expected_method,
self.expected_url,
headers=mock.ANY,
data=mock.ANY)
request_mock.reset_mock()
def test_url_generated_by_endpoint(self, request_mock):
hook = SlackWebhookHook(webhook_token=self.expected_url)
try:
hook.execute()
except MissingSchema:
pass
request_mock.assert_called_once_with(
self.expected_method,
self.expected_url,
headers=mock.ANY,
data=mock.ANY
)
request_mock.reset_mock()
def test_url_generated_by_http_conn_id(self, request_mock):
hook = SlackWebhookHook(http_conn_id='slack-webhook-url')
try:
hook.execute()
except MissingSchema:
pass
request_mock.assert_called_once_with(
self.expected_method,
self.expected_url,
headers=mock.ANY,
data=mock.ANY
)
request_mock.reset_mock()
class SlackWebhookOperator(SimpleHttpOperator):
"""
This operator allows you to post messages to Slack using incoming webhooks.
Takes both Slack webhook token directly and connection that has Slack webhook token.
If both supplied, http_conn_id will be used as base_url,
and webhook_token will be taken as endpoint, the relative path of the url.
Each Slack webhook token can be pre-configured to use a specific channel, username and
icon. You can override these defaults in this hook.
:param http_conn_id: connection that has Slack webhook token in the extra field
:type http_conn_id: str
:param webhook_token: Slack webhook token
:type webhook_token: str
:param message: The message you want to send on Slack
:type message: str
:param attachments: The attachments to send on Slack. Should be a list of
dictionaries representing Slack attachments.
:type attachments: list
:param blocks: The blocks to send on Slack. Should be a list of
dictionaries representing Slack blocks.
:type blocks: list
:param channel: The channel the message should be posted to
:type channel: str
:param username: The username to post to slack with
:type username: str
:param icon_emoji: The emoji to use as icon for the user posting to Slack
:type icon_emoji: str
:param icon_url: The icon image URL string to use in place of the default icon.
:type icon_url: str
:param link_names: Whether or not to find and link channel and usernames in your
message
:type link_names: bool
:param proxy: Proxy to use to make the Slack webhook call
:type proxy: str
:param extra_options: Extra options for http hook
:type extra_options: dict
"""
template_fields = [
'webhook_token',
'message',
'attachments',
'blocks',
'channel',
'username',
'proxy',
'extra_options',
]
@apply_defaults
def __init__(self,
http_conn_id=None,
webhook_token=None,
message="",
attachments=None,
blocks=None,
channel=None,
username=None,
icon_emoji=None,
icon_url=None,
link_names=False,
extra_options=None,
proxy=None,
*args,
**kwargs):
super(SlackWebhookOperator, self).__init__(endpoint=webhook_token,
*args,
**kwargs)
self.http_conn_id = http_conn_id
self.webhook_token = webhook_token
self.message = message
self.attachments = attachments
self.blocks = blocks
self.channel = channel
self.username = username
self.icon_emoji = icon_emoji
self.icon_url = icon_url
self.link_names = link_names
self.proxy = proxy
self.hook = None
self.extra_options = extra_options
def execute(self, context):
"""
Call the SlackWebhookHook to post the provided Slack message
"""
self.hook = SlackWebhookHook(self.http_conn_id, self.webhook_token,
self.message, self.attachments,
self.blocks, self.channel, self.username,
self.icon_emoji, self.icon_url,
self.link_names, self.proxy,
self.extra_options)
self.hook.execute()
class SlackWebhookOperator(SimpleHttpOperator):
"""
This operator allows you to post messages to Slack using incoming webhooks.
Takes both Slack webhook token directly and connection that has Slack webhook token.
If both supplied, Slack webhook token will be used.
Each Slack webhook token can be pre-configured to use a specific channel, username and
icon. You can override these defaults in this hook.
:param http_conn_id: connection that has Slack webhook token in the extra field
:type http_conn_id: str
:param webhook_token: Slack webhook token
:type webhook_token: str
:param message: The message you want to send on Slack
:type message: str
:param attachments: The attachments to send on Slack. Should be a list of
dictionaries representing Slack attachments.
:type attachments: list
:param channel: The channel the message should be posted to
:type channel: str
:param username: The username to post to slack with
:type username: str
:param icon_emoji: The emoji to use as icon for the user posting to Slack
:type icon_emoji: str
:param link_names: Whether or not to find and link channel and usernames in your
message
:type link_names: bool
:param proxy: Proxy to use to make the Slack webhook call
:type proxy: str
"""
@apply_defaults
def __init__(self,
http_conn_id=None,
webhook_token=None,
message="",
attachments=None,
channel=None,
username=None,
icon_emoji=None,
link_names=False,
proxy=None,
*args,
**kwargs):
super(SlackWebhookOperator, self).__init__(endpoint=webhook_token,
*args,
**kwargs)
self.http_conn_id = http_conn_id
self.webhook_token = webhook_token
self.message = message
self.attachments = attachments
self.channel = channel
self.username = username
self.icon_emoji = icon_emoji
self.link_names = link_names
self.proxy = proxy
self.hook = None
def execute(self, context):
"""
Call the SlackWebhookHook to post the provided Slack message
"""
self.hook = SlackWebhookHook(
self.http_conn_id,
self.webhook_token,
self.message,
self.attachments,
self.channel,
self.username,
self.icon_emoji,
self.link_names,
self.proxy
)
self.hook.execute()
class SlackWebhookOperator(SimpleHttpOperator):
"""
This operator allows you to post messages to Slack using incoming webhooks.
Takes both Slack webhook token directly and connection that has Slack webhook token.
If both supplied, Slack webhook token will be used.
Each Slack webhook token can be pre-configured to use a specific channel, username and
icon. You can override these defaults in this hook.
:param http_conn_id: connection that has Slack webhook token in the extra field
:type http_conn_id: str
:param webhook_token: Slack webhook token
:type webhook_token: str
:param message: The message you want to send on Slack
:type message: str
:param channel: The channel the message should be posted to
:type channel: str
:param username: The username to post to slack with
:type username: str
:param icon_emoji: The emoji to use as icon for the user posting to Slack
:type icon_emoji: str
:param link_names: Whether or not to find and link channel and usernames in your
message
:type link_names: bool
:param proxy: Proxy to use to make the Slack webhook call
:type proxy: str
"""
@apply_defaults
def __init__(self,
http_conn_id=None,
webhook_token=None,
message="",
channel=None,
username=None,
icon_emoji=None,
link_names=False,
proxy=None,
*args,
**kwargs):
super(SlackWebhookOperator, self).__init__(endpoint=webhook_token,
*args,
**kwargs)
self.http_conn_id = http_conn_id
self.webhook_token = webhook_token
self.message = message
self.channel = channel
self.username = username
self.icon_emoji = icon_emoji
self.link_names = link_names
self.proxy = proxy
self.hook = None
def execute(self, context):
"""
Call the SlackWebhookHook to post the provided Slack message
"""
self.hook = SlackWebhookHook(self.http_conn_id, self.webhook_token,
self.message, self.channel, self.username,
self.icon_emoji, self.link_names,
self.proxy)
self.hook.execute()
