153 lines
3.2 KiB
Python
153 lines
3.2 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()
|
|
history = serializers.SerializerMethodField('get_history')
|
|
|
|
class Meta:
|
|
model = Issue
|
|
fields = ()
|
|
|
|
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']}
|
|
|
|
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):
|
|
#TODO: add comments info
|
|
diff_list = []
|
|
current = None
|
|
|
|
for version in reversed(list(reversion.get_for_object(obj))):
|
|
if current:
|
|
issues_diff = self.get_issues_diff(current, version)
|
|
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 = ()
|
|
|
|
|