CIRCL
/
AIL-framework
mirror of https://github.com/CIRCL/AIL-framework
2
1
Fork 0
Code Issues Releases Wiki Activity

fix: [chat forum] fix get years daterange

master
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
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
bin/lib/objects/abstract_chat_object.py
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 = {}