43 lines
1.6 KiB
Python
43 lines
1.6 KiB
Python
from .models import *
|
|
from projects.models import *
|
|
from rest_framework import serializers
|
|
|
|
from common.models.choiceModels import NotificationType
|
|
|
|
|
|
NOTIFICATION_REL_MAP_REGISTRY = {
|
|
# "model" : "model_cls",
|
|
# "serializer" : "serializer_cls"
|
|
}
|
|
|
|
def register_notification_serializer(note_type:NotificationType, model_cls):
|
|
def wrapper(serializer_cls):
|
|
NOTIFICATION_REL_MAP_REGISTRY[note_type] = {'model' : model_cls, 'serializer' : serializer_cls}
|
|
return serializer_cls
|
|
return wrapper
|
|
|
|
@register_notification_serializer(NotificationType.PROJECT_INVITE, ProjectInvitation)
|
|
class ProjectInvitationMetaSerializer(serializers.ModelSerializer):
|
|
project_invitation_id = serializers.CharField(source='id')
|
|
project_title = serializers.CharField(source='project.title')
|
|
from_user_nickname = serializers.CharField(source='from_user.nickname')
|
|
|
|
class Meta:
|
|
model = ProjectInvitation
|
|
fields = ['project_invitation_id', 'project_title', 'from_user_nickname', 'status']
|
|
|
|
class NotificationSerializer(serializers.ModelSerializer):
|
|
meta = serializers.SerializerMethodField()
|
|
|
|
class Meta:
|
|
model = Notification
|
|
fields = ['id', 'content', 'note_type', 'is_read', 'created_at', 'meta']
|
|
|
|
def get_meta(self, obj):
|
|
note_type = obj.note_type
|
|
rel_name = NOTIFICATION_REL_MAP_REGISTRY[note_type]['model']._meta.get_field('notification').remote_field.related_name
|
|
rel_obj = getattr(obj, rel_name, None)
|
|
if rel_obj is not None:
|
|
return NOTIFICATION_REL_MAP_REGISTRY[note_type]['serializer'](rel_obj).data
|
|
return None
|
|
|