from rest_framework.response import Response

from .services import record_adventure_listing_click, record_equipment_listing_click


class EquipmentListingClickTrackingMixin:
    def retrieve(self, request, *args, **kwargs):
        instance = self.get_object()
        click = record_equipment_listing_click(request, instance)
        serializer = self.get_serializer(instance)
        data = dict(serializer.data)
        data["marketing_click_id"] = click.id
        data["click_traffic_type"] = click.traffic_type
        return Response(data)


class AdventureListingClickTrackingMixin:
    def retrieve(self, request, *args, **kwargs):
        instance = self.get_object()
        click = record_adventure_listing_click(request, instance)
        serializer = self.get_serializer(instance)
        data = dict(serializer.data)
        data["marketing_click_id"] = click.id
        data["click_traffic_type"] = click.traffic_type
        return Response(data)