99 lines
2.8 KiB
Python
99 lines
2.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import json, reversion
|
|
from rest_framework import serializers
|
|
|
|
from greenmine.base.serializers import PickleField
|
|
from . import models
|
|
|
|
|
|
|
|
class PointsSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = models.Points
|
|
|
|
|
|
class RolePointsField(serializers.WritableField):
|
|
def to_native(self, obj):
|
|
return {str(o.role.id): o.points.order for o in obj.all()}
|
|
|
|
def from_native(self, obj):
|
|
if isinstance(obj, dict):
|
|
return obj
|
|
return json.loads(obj)
|
|
|
|
|
|
class UserStoryStatusSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = models.UserStoryStatus
|
|
|
|
|
|
class UserStorySerializer(serializers.ModelSerializer):
|
|
tags = PickleField(blank=True, default=[])
|
|
is_closed = serializers.Field(source="is_closed")
|
|
points = RolePointsField(source="role_points")
|
|
total_points = serializers.SerializerMethodField("get_total_points")
|
|
comment = serializers.SerializerMethodField("get_comment")
|
|
history = serializers.SerializerMethodField("get_history")
|
|
|
|
class Meta:
|
|
model = models.UserStory
|
|
depth = 0
|
|
|
|
def save_object(self, obj, **kwargs):
|
|
role_points = obj._related_data.pop("role_points", None)
|
|
super(UserStorySerializer, self).save_object(obj, **kwargs)
|
|
obj.project.update_role_points()
|
|
|
|
if role_points:
|
|
for role_id, points_order in role_points.items():
|
|
role_points = obj.role_points.get(role__id=role_id)
|
|
role_points.points = models.Points.objects.get(project=obj.project,
|
|
order=points_order)
|
|
role_points.save()
|
|
|
|
def get_total_points(self, obj):
|
|
return obj.get_total_points()
|
|
|
|
def get_comment(self, obj):
|
|
# TODO
|
|
return ""
|
|
|
|
def get_user_stories_diff(self, old_us_version, new_us_version):
|
|
old_obj = old_us_version.field_dict
|
|
new_obj = new_us_version.field_dict
|
|
|
|
diff_dict = {
|
|
"modified_date": new_obj["modified_date"],
|
|
"by": new_us_version.revision.user,
|
|
"comment": new_us_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
|
|
|
|
if obj:
|
|
for version in reversed(list(reversion.get_for_object(obj))):
|
|
if current:
|
|
us_diff = self.get_user_stories_diff(current, version)
|
|
diff_list.append(us_diff)
|
|
|
|
current = version
|
|
|
|
return diff_list
|