zcdsj
/
factory
5
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: stlogserializer增加校验

This commit is contained in:
caoqianming 2023-11-22 16:33:41 +08:00
parent 2030812c18
commit e78f864666
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
apps/wpm/serializers.py
View File

@ -50,11 +50,11 @@ class StLogSerializer(CustomModelSerializer):
validated_data.pop('mgroup', None) validated_data.pop('mgroup', None)
validated_data.pop('start_time', None) validated_data.pop('start_time', None)
end_time = validated_data.pop('end_time', None) end_time = validated_data.pop('end_time', None)
if end_time: # 需要把涉及到的sflog都拉入
cal_exp_duration_hour(instance.id)
current_sflog = validated_data.pop('current_sflog') current_sflog = validated_data.pop('current_sflog')
current_note = validated_data.pop('current_note', '') current_note = validated_data.pop('current_note', '')
instance = super().update(instance, validated_data) instance = super().update(instance, validated_data)
if end_time: # 需要把涉及到的sflog都拉入
cal_exp_duration_hour(instance.id)
try: try:
sflogexp = SfLogExp.objects.get( sflogexp = SfLogExp.objects.get(
stlog=instance.stlog, sflog=current_sflog) stlog=instance.stlog, sflog=current_sflog)
@ -65,9 +65,14 @@ class StLogSerializer(CustomModelSerializer):
return instance return instance
def validate(self, attrs): def validate(self, attrs):
now = timezone.now()
start_time: datetime = attrs['start_time'] start_time: datetime = attrs['start_time']
end_time: datetime = attrs.get('end_time', None) end_time: datetime = attrs.get('end_time', None)
if start_time > now:
raise ParseError('开始时间应为历史时间')
if end_time: if end_time:
if end_time > now:
raise ParseError('开始时间应为历史时间')
if end_time > start_time: if end_time > start_time:
attrs['duration'] = ( attrs['duration'] = (
end_time - start_time).total_seconds / 3600 end_time - start_time).total_seconds / 3600