taiga-back/greenmine/scrum/serializers.py

161 lines
3.4 KiB
Python

from rest_framework import serializers
from greenmine.scrum.models import *
from picklefield.fields import dbsafe_encode, dbsafe_decode
import json, reversion
class PickleField(serializers.WritableField):
"""
Pickle objects serializer.
"""
def to_native(self, obj):
return obj
def from_native(self, data):
return data
class PointsSerializer(serializers.ModelSerializer):
class Meta:
model = Points
fields = ()
class ProjectSerializer(serializers.ModelSerializer):
tags = PickleField()
class Meta:
model = Project
fields = ()
class UserStorySerializer(serializers.ModelSerializer):
tags = PickleField()
is_closed = serializers.Field(source='is_closed')
class Meta:
model = UserStory
fields = ()
depth = 0
class MilestoneSerializer(serializers.ModelSerializer):
user_stories = UserStorySerializer(many=True, required=False)
class Meta:
model = Milestone
fields = ()
class ChangeSerializer(serializers.ModelSerializer):
tags = PickleField()
class Meta:
model = Change
fields = ()
class ChangeAttachmentSerializer(serializers.ModelSerializer):
tags = PickleField()
class Meta:
model = ChangeAttachment
fields = ()
class TaskSerializer(serializers.ModelSerializer):
tags = PickleField()
class Meta:
model = Task
fields = ()
class IssueSerializer(serializers.ModelSerializer):
tags = PickleField()
comment = serializers.SerializerMethodField('get_comment')
history = serializers.SerializerMethodField('get_history')
class Meta:
model = Issue
fields = ()
def get_comment(self, obj):
return ''
def get_issues_diff(self, old_issue_version, new_issue_version):
old_obj = old_issue_version.field_dict
new_obj = new_issue_version.field_dict
diff_dict = {
'modified_date': new_obj['modified_date'],
'by': old_issue_version.revision.user,
'comment': old_issue_version.revision.comment,
}
for key in old_obj.keys():
if key == 'modified_date':
continue
if old_obj[key] == new_obj[key]:
continue
diff_dict[key] = {
'old': old_obj[key],
'new': new_obj[key],
}
return diff_dict
def get_history(self, obj):
diff_list = []
current = None
for version in reversed(list(reversion.get_for_object(obj))):
if current:
issues_diff = self.get_issues_diff(version, current)
diff_list.append(issues_diff)
current = version
return diff_list
class SeveritySerializer(serializers.ModelSerializer):
class Meta:
model = Severity
fields = ()
class IssueStatusSerializer(serializers.ModelSerializer):
class Meta:
model = IssueStatus
fields = ()
class TaskStatusSerializer(serializers.ModelSerializer):
class Meta:
model = TaskStatus
fields = ()
class UserStoryStatusSerializer(serializers.ModelSerializer):
class Meta:
model = UserStoryStatus
fields = ()
class PrioritySerializer(serializers.ModelSerializer):
class Meta:
model = Priority
fields = ()
class IssueTypeSerializer(serializers.ModelSerializer):
class Meta:
model = IssueType
fields = ()