diff --git a/Whatsapp_Chat_Exporter/android_handler.py b/Whatsapp_Chat_Exporter/android_handler.py index 3374c8c..ea89455 100644 --- a/Whatsapp_Chat_Exporter/android_handler.py +++ b/Whatsapp_Chat_Exporter/android_handler.py @@ -661,6 +661,9 @@ def create_html( for current, contact in enumerate(data): current_chat = data.get_chat(contact) + if len(current_chat) == 0: + # Skip empty chats + continue safe_file_name, name = get_file_name(contact, current_chat) if maximum_size is not None: @@ -733,6 +736,8 @@ def create_html( def create_txt(data, output): os.makedirs(output, exist_ok=True) for jik, chat in data.items(): + if len(chat) == 0: + continue if chat.name is not None: contact = chat.name.replace('/', '') else: diff --git a/Whatsapp_Chat_Exporter/data_model.py b/Whatsapp_Chat_Exporter/data_model.py index 578713a..d724279 100644 --- a/Whatsapp_Chat_Exporter/data_model.py +++ b/Whatsapp_Chat_Exporter/data_model.py @@ -182,6 +182,10 @@ class ChatStore: self.their_avatar_thumb = None self.status = None self.media_base = "" + + def __len__(self) -> int: + """Get number of chats. Required for dict-like access.""" + return len(self._messages) def add_message(self, id: str, message: 'Message') -> None: """Add a message to the chat store."""