mirror of
https://github.com/PostHog/posthog.git
synced 2024-11-24 18:07:17 +01:00
107 lines
3.7 KiB
Python
107 lines
3.7 KiB
Python
from typing import Dict, List
|
|
|
|
import structlog
|
|
from django.conf import settings
|
|
|
|
from posthog.models.exported_asset import ExportedAsset
|
|
from posthog.models.integration import Integration, SlackIntegration
|
|
from posthog.models.subscription import Subscription
|
|
|
|
logger = structlog.get_logger(__name__)
|
|
|
|
UTM_TAGS_BASE = "utm_source=posthog&utm_campaign=subscription_report"
|
|
|
|
|
|
def _block_for_asset(asset: ExportedAsset) -> Dict:
|
|
image_url = asset.get_public_content_url()
|
|
alt_text = None
|
|
if asset.insight:
|
|
alt_text = asset.insight.name or asset.insight.derived_name
|
|
|
|
if settings.DEBUG:
|
|
image_url = "https://source.unsplash.com/random"
|
|
|
|
return {"type": "image", "image_url": image_url, "alt_text": alt_text}
|
|
|
|
|
|
def send_slack_subscription_report(
|
|
subscription: Subscription, assets: List[ExportedAsset], total_asset_count: int, is_new_subscription: bool = False
|
|
) -> None:
|
|
utm_tags = f"{UTM_TAGS_BASE}&utm_medium=slack"
|
|
|
|
resource_info = subscription.resource_info
|
|
if not resource_info:
|
|
raise NotImplementedError("This type of subscription resource is not supported")
|
|
|
|
integration = Integration.objects.filter(team=subscription.team, kind="slack").first()
|
|
|
|
if not integration:
|
|
# TODO: Write error to subscription...
|
|
logger.error("No Slack integration found for team...")
|
|
return
|
|
|
|
slack_integration = SlackIntegration(integration)
|
|
|
|
channel = subscription.target_value.split("|")[0]
|
|
|
|
first_asset, *other_assets = assets
|
|
|
|
if is_new_subscription:
|
|
title = f"This channel has been subscribed to the {resource_info.kind} *{resource_info.name}* on PostHog! 🎉"
|
|
title += f"\nThis subscription is {subscription.summary}. The next one will be sent on {subscription.next_delivery_date.strftime('%A %B %d, %Y')}"
|
|
else:
|
|
title = f"Your subscription to the {resource_info.kind} *{resource_info.name}* is ready! 🎉"
|
|
|
|
blocks = []
|
|
|
|
blocks.extend([{"type": "section", "text": {"type": "mrkdwn", "text": title}}, _block_for_asset(first_asset)])
|
|
|
|
if other_assets:
|
|
blocks.append({"type": "section", "text": {"type": "mrkdwn", "text": "_See 🧵 for more Insights_"}})
|
|
|
|
blocks.extend(
|
|
[
|
|
{"type": "divider"},
|
|
{
|
|
"type": "actions",
|
|
"elements": [
|
|
{
|
|
"type": "button",
|
|
"text": {"type": "plain_text", "text": "View in PostHog"},
|
|
"url": f"{resource_info.url}?{utm_tags}",
|
|
},
|
|
{
|
|
"type": "button",
|
|
"text": {"type": "plain_text", "text": "Manage Subscription"},
|
|
"url": f"{subscription.url}?{utm_tags}",
|
|
},
|
|
],
|
|
},
|
|
]
|
|
)
|
|
|
|
message_res = slack_integration.client.chat_postMessage(channel=channel, blocks=blocks, text=title)
|
|
|
|
thread_ts = message_res.get("ts")
|
|
|
|
if thread_ts:
|
|
for asset in other_assets:
|
|
slack_integration.client.chat_postMessage(
|
|
channel=channel, thread_ts=thread_ts, blocks=[_block_for_asset(asset)]
|
|
)
|
|
|
|
if total_asset_count > len(assets):
|
|
slack_integration.client.chat_postMessage(
|
|
channel=channel,
|
|
thread_ts=thread_ts,
|
|
blocks=[
|
|
{
|
|
"type": "section",
|
|
"text": {
|
|
"type": "mrkdwn",
|
|
"text": f"Showing {len(assets)} of {total_asset_count} Insights. <{resource_info.url}?{utm_tags}|View the rest in PostHog>",
|
|
},
|
|
}
|
|
],
|
|
)
|