Qlockify/qlockify-backend-deployment
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(pricing): add workspace user rates and price units

Browse Source
This commit is contained in:
Amirhossein Khalili
2026-04-26 10:19:04 +03:30
parent f960ca8221
commit fadf898486
19 changed files with 731 additions and 266 deletions
Download Patch File Download Diff File Expand all files Collapse all files

81
apps/workspaces/api/serializers.py
View File

@@ -1,9 +1,11 @@
from decimal import Decimal
from rest_framework import serializers
from apps.notifications.services import notify_workspace_membership_added
from apps.users.models import User
from core.serializers.base import BaseModelSerializer
from apps.workspaces.models import Workspace, WorkspaceMembership
from apps.workspaces.models import PriceUnit, Workspace, WorkspaceMembership, WorkspaceUserRate
from core.serializers.mini import UserMiniSerializer
@@ -73,7 +75,7 @@ class WorkspaceSerializer(BaseModelSerializer):
return workspace
class WorkspaceMembershipSerializer(BaseModelSerializer):
class WorkspaceMembershipSerializer(BaseModelSerializer):
class Meta:
model = WorkspaceMembership
fields = BaseModelSerializer.Meta.fields + (
@@ -85,8 +87,73 @@ class WorkspaceMembershipSerializer(BaseModelSerializer):
def to_representation(self, instance):
data = super().to_representation(instance)
data["user"] = UserMiniSerializer(
instance.user,
context=self.context
).data
return data
data["user"] = UserMiniSerializer(
instance.user,
context=self.context
).data
return data
class PriceUnitSerializer(BaseModelSerializer):
class Meta:
model = PriceUnit
fields = BaseModelSerializer.Meta.fields + (
"code",
"name",
"local_name",
"symbol",
)
read_only_fields = fields
class WorkspaceUserRateSerializer(BaseModelSerializer):
user_details = UserMiniSerializer(source="user", read_only=True)
price_unit = serializers.SerializerMethodField()
class Meta:
model = WorkspaceUserRate
fields = BaseModelSerializer.Meta.fields + (
"workspace",
"user",
"user_details",
"hourly_rate",
"currency",
"price_unit",
"effective_from",
)
read_only_fields = fields
def get_price_unit(self, obj):
unit = PriceUnit.objects.filter(code=obj.currency, is_deleted=False).first()
if not unit:
return None
return PriceUnitSerializer(unit, context=self.context).data
class WorkspaceUserRateCreateSerializer(serializers.Serializer):
workspace_id = serializers.UUIDField()
user_id = serializers.UUIDField()
hourly_rate = serializers.DecimalField(max_digits=10, decimal_places=2, min_value=Decimal("0.01"))
currency = serializers.CharField(max_length=3, default="USD")
def validate_currency(self, value):
code = value.upper()
if not PriceUnit.objects.filter(code=code, is_deleted=False).exists():
raise serializers.ValidationError("Selected price unit is invalid.")
return code
class WorkspaceUserRateUpdateSerializer(serializers.Serializer):
hourly_rate = serializers.DecimalField(
max_digits=10,
decimal_places=2,
min_value=Decimal("0.01"),
required=False,
)
currency = serializers.CharField(max_length=3, required=False)
def validate_currency(self, value):
code = value.upper()
if not PriceUnit.objects.filter(code=code, is_deleted=False).exists():
raise serializers.ValidationError("Selected price unit is invalid.")
return code

17
apps/workspaces/api/urls.py
View File

@@ -1,11 +1,18 @@
from django.urls import path, include
from rest_framework.routers import DefaultRouter
from apps.workspaces.api.views import WorkspaceViewSet, WorkspaceMembershipViewSet
router = DefaultRouter()
router.register(r'workspaces', WorkspaceViewSet, basename='workspace')
router.register(r'workspace-memberships', WorkspaceMembershipViewSet, basename='workspace-membership')
from apps.workspaces.api.views import (
PriceUnitViewSet,
WorkspaceViewSet,
WorkspaceMembershipViewSet,
WorkspaceUserRateViewSet,
)
router = DefaultRouter()
router.register(r'workspaces', WorkspaceViewSet, basename='workspace')
router.register(r'workspace-memberships', WorkspaceMembershipViewSet, basename='workspace-membership')
router.register(r'price-units', PriceUnitViewSet, basename='price-unit')
router.register(r'workspace-user-rates', WorkspaceUserRateViewSet, basename='workspace-user-rate')
urlpatterns = [
path('', include(router.urls)),

117
apps/workspaces/api/views.py
View File

@@ -1,7 +1,8 @@
from django.db.models import Q
from django.shortcuts import get_object_or_404
from rest_framework import status
from rest_framework.response import Response
from rest_framework import status
from rest_framework.exceptions import PermissionDenied
from rest_framework.response import Response
from rest_framework.filters import OrderingFilter, SearchFilter
from django_filters.rest_framework import DjangoFilterBackend
from rest_framework.viewsets import ModelViewSet
@@ -19,14 +20,24 @@ from apps.workspaces.api.permissions import (
IsWorkspaceMember,
IsWorkspaceOwner,
)
from apps.workspaces.api.serializers import WorkspaceMembershipSerializer, WorkspaceSerializer
from apps.workspaces.api.serializers import (
PriceUnitSerializer,
WorkspaceMembershipSerializer,
WorkspaceSerializer,
WorkspaceUserRateSerializer,
WorkspaceUserRateCreateSerializer,
WorkspaceUserRateUpdateSerializer,
)
from apps.workspaces.api.filters import WorkspaceFilter, WorkspaceMembershipFilter
from apps.workspaces.models import Workspace, WorkspaceMembership
from apps.workspaces.models import PriceUnit, Workspace, WorkspaceMembership, WorkspaceUserRate
from apps.workspaces.services import (
WORKSPACE_MEMBERS_VIEW,
WORKSPACE_EDIT,
can_assign_workspace_role,
can_change_workspace_membership,
has_workspace_capability,
upsert_workspace_user_rate,
update_workspace_user_rate,
)
from core.paginations.limit_offset import CustomLimitOffsetPagination
@@ -65,7 +76,7 @@ class WorkspaceViewSet(ModelViewSet):
serializer.save(owner=self.request.user)
class WorkspaceMembershipViewSet(ModelViewSet):
class WorkspaceMembershipViewSet(ModelViewSet):
serializer_class = WorkspaceMembershipSerializer
pagination_class = CustomLimitOffsetPagination
filter_backends = (DjangoFilterBackend, OrderingFilter, SearchFilter)
@@ -236,3 +247,99 @@ class WorkspaceMembershipViewSet(ModelViewSet):
role=role,
)
return Response(status=status.HTTP_204_NO_CONTENT)
class PriceUnitViewSet(ModelViewSet):
serializer_class = PriceUnitSerializer
permission_classes = [IsAuthenticated]
http_method_names = ["get", "head", "options"]
pagination_class = None
filter_backends = (SearchFilter, OrderingFilter)
search_fields = ("code", "name", "local_name", "symbol")
ordering_fields = ("code", "name")
ordering = ("code",)
def get_queryset(self):
return PriceUnit.objects.filter(is_deleted=False)
class WorkspaceUserRateViewSet(ModelViewSet):
serializer_class = WorkspaceUserRateSerializer
pagination_class = CustomLimitOffsetPagination
filter_backends = (DjangoFilterBackend, OrderingFilter)
filterset_fields = ("workspace", "user", "currency")
ordering_fields = ("effective_from", "updated_at", "created_at")
ordering = ("-effective_from", "-updated_at")
def get_queryset(self):
user = self.request.user
if not user.is_authenticated:
return WorkspaceUserRate.objects.none()
return WorkspaceUserRate.objects.filter(
workspace__memberships__user=user,
workspace__memberships__is_active=True,
is_deleted=False,
).distinct()
def get_serializer_class(self):
if self.action == "create":
return WorkspaceUserRateCreateSerializer
if self.action in ["update", "partial_update"]:
return WorkspaceUserRateUpdateSerializer
return WorkspaceUserRateSerializer
def _ensure_manage_access(self, user, workspace):
if not has_workspace_capability(user, workspace, WORKSPACE_EDIT):
raise PermissionDenied("You do not have permission to manage workspace rates.")
def list(self, request, *args, **kwargs):
workspace_id = request.query_params.get("workspace")
if not workspace_id:
return Response(
{"detail": "workspace query parameter is required."},
status=status.HTTP_400_BAD_REQUEST,
)
workspace = get_object_or_404(Workspace, id=workspace_id, is_deleted=False)
self._ensure_manage_access(request.user, workspace)
return super().list(request, *args, **kwargs)
def create(self, request, *args, **kwargs):
serializer = self.get_serializer(data=request.data)
serializer.is_valid(raise_exception=True)
workspace = get_object_or_404(
Workspace,
id=serializer.validated_data["workspace_id"],
is_deleted=False,
)
self._ensure_manage_access(request.user, workspace)
rate = upsert_workspace_user_rate(
workspace=workspace,
user_id=serializer.validated_data["user_id"],
hourly_rate=serializer.validated_data["hourly_rate"],
currency=serializer.validated_data.get("currency", "USD"),
)
return Response(
WorkspaceUserRateSerializer(rate, context=self.get_serializer_context()).data,
status=status.HTTP_201_CREATED,
)
def update(self, request, *args, **kwargs):
rate = self.get_object()
self._ensure_manage_access(request.user, rate.workspace)
serializer = self.get_serializer(data=request.data, partial=kwargs.pop("partial", False))
serializer.is_valid(raise_exception=True)
updated_rate = update_workspace_user_rate(rate, **serializer.validated_data)
return Response(
WorkspaceUserRateSerializer(updated_rate, context=self.get_serializer_context()).data,
status=status.HTTP_200_OK,
)
def partial_update(self, request, *args, **kwargs):
kwargs["partial"] = True
return self.update(request, *args, **kwargs)
def destroy(self, request, *args, **kwargs):
rate = self.get_object()
self._ensure_manage_access(request.user, rate.workspace)
rate.delete()
return Response(status=status.HTTP_204_NO_CONTENT)