fix: [chat forum] fix get years daterange

This commit is contained in:
terrtia 2024-12-23 23:35:08 +01:00
parent e19595daa7
commit 66de987c99
No known key found for this signature in database
GPG key ID: 1E1B1F50D84613D0

View file

@ -228,14 +228,17 @@ class AbstractChatObject(AbstractSubtypeObject, ABC):
return stats return stats
def get_message_years(self): def get_message_years(self):
try: timestamp = self.get_timestamp_first_message()
timestamp = datetime.utcfromtimestamp(float(self.get_timestamp_first_message())) if not timestamp:
year_start = int(self.get_first_seen()[0:4])
year_end = int(self.get_last_seen()[0:4])
return list(range(year_start, year_end + 1))
else:
timestamp = datetime.utcfromtimestamp(float(timestamp))
year_start = int(timestamp.strftime('%Y')) year_start = int(timestamp.strftime('%Y'))
timestamp = datetime.utcfromtimestamp(float(self.get_timestamp_last_message())) timestamp = datetime.utcfromtimestamp(float(self.get_timestamp_last_message()))
year_end = int(timestamp.strftime('%Y')) year_end = int(timestamp.strftime('%Y'))
return list(range(year_start, year_end + 1)) return list(range(year_start, year_end + 1))
except:
return []
def get_nb_year_messages(self, year): def get_nb_year_messages(self, year):
nb_year = {} nb_year = {}