forked from github/plane
dev: webhooks
This commit is contained in:
parent
2e513af4f3
commit
e275740a45
@ -12,6 +12,6 @@ from .issue import (
|
|||||||
IssueExpandSerializer,
|
IssueExpandSerializer,
|
||||||
)
|
)
|
||||||
from .state import StateLiteSerializer, StateSerializer
|
from .state import StateLiteSerializer, StateSerializer
|
||||||
from .cycle import CycleSerializer, CycleIssueSerializer
|
from .cycle import CycleSerializer, CycleIssueSerializer, CycleLiteSerializer
|
||||||
from .module import ModuleSerializer, ModuleIssueSerializer
|
from .module import ModuleSerializer, ModuleIssueSerializer, ModuleLiteSerializer
|
||||||
from .inbox import InboxIssueSerializer
|
from .inbox import InboxIssueSerializer
|
@ -46,4 +46,11 @@ class CycleIssueSerializer(BaseSerializer):
|
|||||||
"workspace",
|
"workspace",
|
||||||
"project",
|
"project",
|
||||||
"cycle",
|
"cycle",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
|
class CycleLiteSerializer(BaseSerializer):
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
model = Cycle
|
||||||
|
fields = "__all__"
|
@ -19,8 +19,9 @@ from plane.db.models import (
|
|||||||
ProjectMember,
|
ProjectMember,
|
||||||
)
|
)
|
||||||
from .base import BaseSerializer
|
from .base import BaseSerializer
|
||||||
from .cycle import CycleSerializer
|
from .cycle import CycleSerializer, CycleLiteSerializer
|
||||||
from .module import ModuleSerializer
|
from .module import ModuleSerializer, ModuleLiteSerializer
|
||||||
|
|
||||||
|
|
||||||
class IssueSerializer(BaseSerializer):
|
class IssueSerializer(BaseSerializer):
|
||||||
assignees = serializers.ListField(
|
assignees = serializers.ListField(
|
||||||
@ -312,10 +313,30 @@ class IssueActivitySerializer(BaseSerializer):
|
|||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
class IssueExpandSerializer(BaseSerializer):
|
class CycleIssueSerializer(BaseSerializer):
|
||||||
|
cycle = CycleSerializer(read_only=True)
|
||||||
|
|
||||||
cycle = CycleSerializer(source="issue_cycle.cycle", many=True)
|
class Meta:
|
||||||
module = ModuleSerializer(source="issue_module.module", many=True)
|
fields = [
|
||||||
|
"cycle",
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
class ModuleIssueSerializer(BaseSerializer):
|
||||||
|
module = ModuleSerializer(read_only=True)
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
fields = [
|
||||||
|
"module",
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
class IssueExpandSerializer(BaseSerializer):
|
||||||
|
# Serialize the related cycle. It's a OneToOne relation.
|
||||||
|
cycle = CycleLiteSerializer(source="issue_cycle.cycle", read_only=True)
|
||||||
|
|
||||||
|
# Serialize the related module. It's a OneToOne relation.
|
||||||
|
module = ModuleLiteSerializer(source="issue_module.module", read_only=True)
|
||||||
|
|
||||||
class Meta:
|
class Meta:
|
||||||
model = Issue
|
model = Issue
|
||||||
|
@ -21,7 +21,6 @@ class ModuleSerializer(BaseSerializer):
|
|||||||
write_only=True,
|
write_only=True,
|
||||||
required=False,
|
required=False,
|
||||||
)
|
)
|
||||||
is_favorite = serializers.BooleanField(read_only=True)
|
|
||||||
total_issues = serializers.IntegerField(read_only=True)
|
total_issues = serializers.IntegerField(read_only=True)
|
||||||
cancelled_issues = serializers.IntegerField(read_only=True)
|
cancelled_issues = serializers.IntegerField(read_only=True)
|
||||||
completed_issues = serializers.IntegerField(read_only=True)
|
completed_issues = serializers.IntegerField(read_only=True)
|
||||||
@ -153,4 +152,11 @@ class ModuleLinkSerializer(BaseSerializer):
|
|||||||
raise serializers.ValidationError(
|
raise serializers.ValidationError(
|
||||||
{"error": "URL already exists for this Issue"}
|
{"error": "URL already exists for this Issue"}
|
||||||
)
|
)
|
||||||
return ModuleLink.objects.create(**validated_data)
|
return ModuleLink.objects.create(**validated_data)
|
||||||
|
|
||||||
|
|
||||||
|
class ModuleLiteSerializer(BaseSerializer):
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
model = Module
|
||||||
|
fields = "__all__"
|
@ -52,7 +52,7 @@ class WebhookMixin:
|
|||||||
# Push the object to delay
|
# Push the object to delay
|
||||||
send_webhook.delay(
|
send_webhook.delay(
|
||||||
event=self.webhook_event,
|
event=self.webhook_event,
|
||||||
event_data=response.data,
|
payload=response.data,
|
||||||
kw=self.kwargs,
|
kw=self.kwargs,
|
||||||
action=self.request.method,
|
action=self.request.method,
|
||||||
slug=self.workspace_slug,
|
slug=self.workspace_slug,
|
||||||
|
@ -311,6 +311,7 @@ class CycleIssueAPIEndpoint(WebhookMixin, BaseAPIView):
|
|||||||
serializer_class = CycleIssueSerializer
|
serializer_class = CycleIssueSerializer
|
||||||
model = CycleIssue
|
model = CycleIssue
|
||||||
webhook_event = "cycle_issue"
|
webhook_event = "cycle_issue"
|
||||||
|
bulk = True
|
||||||
permission_classes = [
|
permission_classes = [
|
||||||
ProjectEntityPermission,
|
ProjectEntityPermission,
|
||||||
]
|
]
|
||||||
|
@ -43,27 +43,28 @@ class TimezoneMixin:
|
|||||||
|
|
||||||
class WebhookMixin:
|
class WebhookMixin:
|
||||||
webhook_event = None
|
webhook_event = None
|
||||||
|
bulk = False
|
||||||
|
|
||||||
def finalize_response(self, request, response, *args, **kwargs):
|
def finalize_response(self, request, response, *args, **kwargs):
|
||||||
response = super().finalize_response(request, response, *args, **kwargs)
|
response = super().finalize_response(request, response, *args, **kwargs)
|
||||||
|
|
||||||
|
# Check for the case should webhook be sent
|
||||||
if (
|
if (
|
||||||
self.webhook_event
|
self.webhook_event
|
||||||
and self.request.method in ["POST", "PATCH"]
|
and self.request.method in ["POST", "PATCH", "DELETE"]
|
||||||
and response.status_code in [200, 201, 204]
|
and response.status_code in [200, 201, 204]
|
||||||
):
|
):
|
||||||
# Get the id
|
# Push the object to delay
|
||||||
object_id = (
|
|
||||||
response.data.get("id") if isinstance(response.data, dict) else None
|
|
||||||
)
|
|
||||||
|
|
||||||
send_webhook.delay(
|
send_webhook.delay(
|
||||||
event=self.webhook_event,
|
event=self.webhook_event,
|
||||||
event_id=object_id,
|
payload=response.data,
|
||||||
|
kw=self.kwargs,
|
||||||
action=self.request.method,
|
action=self.request.method,
|
||||||
slug=self.workspace_slug,
|
slug=self.workspace_slug,
|
||||||
|
bulk=self.bulk,
|
||||||
)
|
)
|
||||||
return response
|
|
||||||
|
|
||||||
|
return response
|
||||||
|
|
||||||
|
|
||||||
class BaseViewSet(TimezoneMixin, ModelViewSet, BasePaginator):
|
class BaseViewSet(TimezoneMixin, ModelViewSet, BasePaginator):
|
||||||
|
@ -502,7 +502,10 @@ class CycleViewSet(WebhookMixin, BaseViewSet):
|
|||||||
class CycleIssueViewSet(WebhookMixin, BaseViewSet):
|
class CycleIssueViewSet(WebhookMixin, BaseViewSet):
|
||||||
serializer_class = CycleIssueSerializer
|
serializer_class = CycleIssueSerializer
|
||||||
model = CycleIssue
|
model = CycleIssue
|
||||||
|
|
||||||
webhook_event = "cycle_issue"
|
webhook_event = "cycle_issue"
|
||||||
|
bulk = True
|
||||||
|
|
||||||
permission_classes = [
|
permission_classes = [
|
||||||
ProjectEntityPermission,
|
ProjectEntityPermission,
|
||||||
]
|
]
|
||||||
|
@ -287,6 +287,8 @@ class ModuleIssueViewSet(WebhookMixin, BaseViewSet):
|
|||||||
serializer_class = ModuleIssueSerializer
|
serializer_class = ModuleIssueSerializer
|
||||||
model = ModuleIssue
|
model = ModuleIssue
|
||||||
webhook_event = "module_issue"
|
webhook_event = "module_issue"
|
||||||
|
bulk = True
|
||||||
|
|
||||||
|
|
||||||
filterset_fields = [
|
filterset_fields = [
|
||||||
"issue__labels__id",
|
"issue__labels__id",
|
||||||
|
@ -55,11 +55,14 @@ MODEL_MAPPER = {
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
def get_model_data(event, event_id, many=True):
|
def get_model_data(event, event_id, many=False):
|
||||||
model = MODEL_MAPPER.get(event)
|
model = MODEL_MAPPER.get(event)
|
||||||
queryset = model.objects.get(pk=event_id)
|
if many:
|
||||||
|
queryset = model.objects.filter(pk__in=event_id)
|
||||||
|
else:
|
||||||
|
queryset = model.objects.get(pk=event_id)
|
||||||
serializer = SERIALIZER_MAPPER.get(event)
|
serializer = SERIALIZER_MAPPER.get(event)
|
||||||
return serializer(queryset).data
|
return serializer(queryset, many=many).data
|
||||||
|
|
||||||
|
|
||||||
@shared_task(
|
@shared_task(
|
||||||
@ -89,11 +92,11 @@ def webhook_task(self, webhook, slug, event, event_data, action):
|
|||||||
|
|
||||||
# Use HMAC for generating signature
|
# Use HMAC for generating signature
|
||||||
if webhook.secret_key:
|
if webhook.secret_key:
|
||||||
event_data_json = json.dumps(event_data) if event_data is not None else '{}'
|
event_data_json = json.dumps(event_data) if event_data is not None else "{}"
|
||||||
hmac_signature = hmac.new(
|
hmac_signature = hmac.new(
|
||||||
webhook.secret_key.encode("utf-8"),
|
webhook.secret_key.encode("utf-8"),
|
||||||
event_data_json.encode("utf-8"),
|
event_data_json.encode("utf-8"),
|
||||||
hashlib.sha256
|
hashlib.sha256,
|
||||||
)
|
)
|
||||||
signature = hmac_signature.hexdigest()
|
signature = hmac_signature.hexdigest()
|
||||||
headers["X-Plane-Signature"] = signature
|
headers["X-Plane-Signature"] = signature
|
||||||
@ -157,7 +160,6 @@ def webhook_task(self, webhook, slug, event, event_data, action):
|
|||||||
raise requests.RequestException()
|
raise requests.RequestException()
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(e)
|
|
||||||
if settings.DEBUG:
|
if settings.DEBUG:
|
||||||
print(e)
|
print(e)
|
||||||
capture_exception(e)
|
capture_exception(e)
|
||||||
@ -165,7 +167,7 @@ def webhook_task(self, webhook, slug, event, event_data, action):
|
|||||||
|
|
||||||
|
|
||||||
@shared_task()
|
@shared_task()
|
||||||
def send_webhook(event, event_data, kw, action, slug, bulk):
|
def send_webhook(event, payload, kw, action, slug, bulk):
|
||||||
try:
|
try:
|
||||||
webhooks = Webhook.objects.filter(workspace__slug=slug, is_active=True)
|
webhooks = Webhook.objects.filter(workspace__slug=slug, is_active=True)
|
||||||
|
|
||||||
@ -184,22 +186,39 @@ def send_webhook(event, event_data, kw, action, slug, bulk):
|
|||||||
if event == "issue_comment":
|
if event == "issue_comment":
|
||||||
webhooks = webhooks.filter(issue_comment=True)
|
webhooks = webhooks.filter(issue_comment=True)
|
||||||
|
|
||||||
|
|
||||||
if webhooks:
|
if webhooks:
|
||||||
if action in ["POST", "PATCH"]:
|
if action in ["POST", "PATCH"]:
|
||||||
if bulk:
|
if bulk and event in ["cycle_issue", "module_issue"]:
|
||||||
event_data= []
|
event_data = IssueExpandSerializer(
|
||||||
if event in ["cycle_issue", "module_issue"]:
|
Issue.objects.filter(
|
||||||
pass
|
pk__in=[
|
||||||
|
str(event.get("issue")) for event in payload
|
||||||
|
]
|
||||||
|
).prefetch_related("issue_cycle", "issue_module"), many=True
|
||||||
|
).data
|
||||||
|
event = "issue"
|
||||||
|
action = "PATCH"
|
||||||
else:
|
else:
|
||||||
event_id = event_data.get("id") if isinstance(event_data, dict) else None
|
event_data = [
|
||||||
event_data = [get_model_data(event=event, event_id=event_id, many=isinstance(event_id, list))]
|
get_model_data(
|
||||||
|
event=event,
|
||||||
|
event_id=payload.get("id") if isinstance(payload, dict) else None,
|
||||||
|
many=False,
|
||||||
|
)
|
||||||
|
]
|
||||||
|
|
||||||
if action == "DELETE":
|
if action == "DELETE":
|
||||||
event_data = [{"id": kw.get("pk")}]
|
event_data = [{"id": kw.get("pk")}]
|
||||||
|
|
||||||
for webhook in webhooks:
|
for webhook in webhooks:
|
||||||
for data in event_data:
|
for data in event_data:
|
||||||
webhook_task.delay(webhook=webhook.id, slug=slug, event=event, event_data=data, action=action,)
|
webhook_task.delay(
|
||||||
|
webhook=webhook.id,
|
||||||
|
slug=slug,
|
||||||
|
event=event,
|
||||||
|
event_data=data,
|
||||||
|
action=action,
|
||||||
|
)
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
if settings.DEBUG:
|
if settings.DEBUG:
|
||||||
|
@ -51,9 +51,9 @@ class Module(ProjectBaseModel):
|
|||||||
|
|
||||||
def save(self, *args, **kwargs):
|
def save(self, *args, **kwargs):
|
||||||
if self._state.adding:
|
if self._state.adding:
|
||||||
smallest_sort_order = Module.objects.filter(
|
smallest_sort_order = Module.objects.filter(project=self.project).aggregate(
|
||||||
project=self.project
|
smallest=models.Min("sort_order")
|
||||||
).aggregate(smallest=models.Min("sort_order"))["smallest"]
|
)["smallest"]
|
||||||
|
|
||||||
if smallest_sort_order is not None:
|
if smallest_sort_order is not None:
|
||||||
self.sort_order = smallest_sort_order - 10000
|
self.sort_order = smallest_sort_order - 10000
|
||||||
|
Loading…
Reference in New Issue
Block a user