mirror of
https://github.com/PostHog/posthog.git
synced 2024-11-28 00:46:45 +01:00
56 lines
2.2 KiB
Python
56 lines
2.2 KiB
Python
import json
|
|
from typing import Optional
|
|
|
|
from django.db import models
|
|
from django.db.models.signals import post_delete, post_save
|
|
from django.dispatch.dispatcher import receiver
|
|
from rest_hooks.models import AbstractHook
|
|
from statshog.defaults.django import statsd
|
|
|
|
from ee.tasks.hooks import DeliverHook
|
|
from posthog.constants import AvailableFeature
|
|
from posthog.models.signals import mutable_receiver
|
|
from posthog.models.team import Team
|
|
from posthog.models.utils import generate_random_token
|
|
from posthog.redis import get_client
|
|
|
|
|
|
class Hook(AbstractHook):
|
|
id = models.CharField(primary_key=True, max_length=50, default=generate_random_token)
|
|
user = models.ForeignKey("posthog.User", related_name="rest_hooks", on_delete=models.CASCADE)
|
|
team = models.ForeignKey("posthog.Team", related_name="rest_hooks", on_delete=models.CASCADE)
|
|
resource_id = models.IntegerField(null=True, blank=True)
|
|
|
|
|
|
def find_and_fire_hook(
|
|
event_name: str, instance: models.Model, user_override: Team, payload_override: Optional[dict] = None
|
|
):
|
|
if not user_override.organization.is_feature_available(AvailableFeature.ZAPIER):
|
|
return
|
|
hooks = Hook.objects.filter(event=event_name, team=user_override)
|
|
if event_name == "action_performed":
|
|
# action_performed is a resource_id-filterable hook
|
|
hooks = hooks.filter(models.Q(resource_id=instance.pk))
|
|
for hook in hooks:
|
|
statsd.incr("posthog_cloud_hooks_rest_fired")
|
|
hook.deliver_hook(instance, payload_override)
|
|
|
|
|
|
def deliver_hook_wrapper(target, payload, instance, hook):
|
|
# pass IDs not objects because using pickle for objects is a bad thing
|
|
DeliverHook.apply_async(kwargs=dict(target=target, payload=payload, hook_id=hook.id))
|
|
|
|
|
|
@receiver(post_save, sender=Hook)
|
|
def hook_saved(sender, instance: Hook, created, **kwargs):
|
|
if instance.event == "action_performed":
|
|
get_client().publish(
|
|
"reload-action", json.dumps({"teamId": instance.team_id, "actionId": instance.resource_id})
|
|
)
|
|
|
|
|
|
@mutable_receiver(post_delete, sender=Hook)
|
|
def hook_deleted(sender, instance: Hook, **kwargs):
|
|
if instance.event == "action_performed":
|
|
get_client().publish("drop-action", json.dumps({"teamId": instance.team_id, "actionId": instance.resource_id}))
|