mirror of
https://github.com/apache/superset.git
synced 2026-04-19 16:14:52 +00:00
29e9f2c70b
* First draft for the slack integration Fix slack another typo another typo Fix slack Add channels to the form another typo Another set of changes Make code more transparent Fix tests Add logging logging use logger import logging import logging import logging add assert more logging Fix channels Fix channels * Address comments * Move slack into a separate module Co-authored-by: bogdan kyryliuk <bogdankyryliuk@dropbox.com>
47 lines
1.7 KiB
Python
47 lines
1.7 KiB
Python
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
import logging
|
|
from io import IOBase
|
|
from typing import cast, Union
|
|
|
|
from retry.api import retry
|
|
from slack import WebClient
|
|
from slack.errors import SlackApiError
|
|
from slack.web.slack_response import SlackResponse
|
|
|
|
from superset import app
|
|
|
|
# Globals
|
|
config = app.config # type: ignore
|
|
logger = logging.getLogger("tasks.slack_util")
|
|
|
|
|
|
@retry(SlackApiError, delay=10, backoff=2, tries=5)
|
|
def deliver_slack_msg(
|
|
slack_channel: str, subject: str, body: str, file: Union[str, IOBase]
|
|
) -> None:
|
|
client = WebClient(token=config["SLACK_API_TOKEN"], proxy=config["SLACK_PROXY"])
|
|
# files_upload returns SlackResponse as we run it in sync mode.
|
|
response = cast(
|
|
SlackResponse,
|
|
client.files_upload(
|
|
channels=slack_channel, file=file, initial_comment=body, title=subject
|
|
),
|
|
)
|
|
logger.info(f"Sent the report to the slack {slack_channel}")
|
|
assert response["file"], str(response) # the uploaded file
|