fix comparation datas put both in utc time

This commit is contained in:
Cayo Puigdefabregas 2022-03-31 19:40:05 +02:00
parent 6fad773c91
commit 026a88acb8
2 changed files with 4 additions and 3 deletions

View File

@ -17,6 +17,7 @@ from datetime import datetime, timedelta, timezone
from decimal import ROUND_HALF_EVEN, ROUND_UP, Decimal from decimal import ROUND_HALF_EVEN, ROUND_UP, Decimal
from typing import Optional, Set, Union from typing import Optional, Set, Union
from uuid import uuid4 from uuid import uuid4
from dateutil.tz import tzutc
import inflection import inflection
import teal.db import teal.db
@ -273,7 +274,7 @@ class Action(Thing):
super().__init__(**kwargs) super().__init__(**kwargs)
def __lt__(self, other): def __lt__(self, other):
return self.end_time < other.end_time return self.end_time.replace(tzinfo=tzutc()) < other.end_time.replace(tzinfo=tzutc())
def __str__(self) -> str: def __str__(self) -> str:
return '{}'.format(self.severity) return '{}'.format(self.severity)

View File

@ -73,10 +73,10 @@ class Action(Thing):
@validates_schema @validates_schema
def validate_times(self, data: dict): def validate_times(self, data: dict):
unix_time = datetime.fromisoformat("1970-01-02 00:00:00+00:00") unix_time = datetime.fromisoformat("1970-01-02 00:00:00+00:00")
if 'end_time' in data and data['end_time'] < unix_time: if 'end_time' in data and data['end_time'].replace(tzinfo=tzutc()) < unix_time:
data['end_time'] = unix_time data['end_time'] = unix_time
if 'start_time' in data and data['start_time'] < unix_time: if 'start_time' in data and data['start_time'].replace(tzinfo=tzutc()) < unix_time:
data['start_time'] = unix_time data['start_time'] = unix_time
if data.get('end_time') and data.get('start_time'): if data.get('end_time') and data.get('start_time'):