From fff11b26a51c945733f5f7070da756a56ebce04d Mon Sep 17 00:00:00 2001
From: KnugiHK <24708955+KnugiHK@users.noreply.github.com>
Date: Fri, 16 Jun 2023 19:12:06 +0800
Subject: [PATCH] Store the device type

---
 Whatsapp_Chat_Exporter/data_model.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/Whatsapp_Chat_Exporter/data_model.py b/Whatsapp_Chat_Exporter/data_model.py
index f74d45a..9d8b135 100644
--- a/Whatsapp_Chat_Exporter/data_model.py
+++ b/Whatsapp_Chat_Exporter/data_model.py
@@ -10,10 +10,11 @@ class ChatStore():
             raise TypeError("Name must be a string or None")
         self.name = name
         self.messages = {}
+        self.type = type
         if media is not None:
-            if type == Device.IOS:
+            if self.type == Device.IOS:
                 self.my_avatar = os.path.join(media, "Media/Profile/Photo.jpg")
-            elif type == Device.ANDROID:
+            elif self.type == Device.ANDROID:
                 self.my_avatar = None  # TODO: Add Android support
             else:
                 self.my_avatar = None