Support no wa.db

extract.py

@@ -24,6 +24,7 @@ def determine_day(last, current):
 data = {}
 
 # Get contacts
+if os.path.isfile("wa.db"):
     wa = sqlite3.connect("wa.db")
     c = wa.cursor()
     c.execute("""SELECT count() FROM wa_contacts""")
@@ -59,11 +60,17 @@ while content is not None:
         "key_id": content[13]
     }
     if "-" in content[0] and content[2] == 0:
+        name = None
         if content[8] in data:
             name = data[content[8]]["name"]
+            if "@" in content[8]:
+                fallback = content[8].split('@')[0]
             else:
-            name = None
-        data[content[0]]["messages"][content[1]]["sender"] = name or content[8].split('@')[0]
+                fallback = None
+        else:
+            fallback = None
+        
+        data[content[0]]["messages"][content[1]]["sender"] = name or fallback
     else:
         data[content[0]]["messages"][content[1]]["sender"] = None
     

whatsapp.html

@@ -111,7 +111,7 @@
 						<div style="text-align: right; color:#70777c;">{{ msg.time }}</div>
 					</div>
 					<div class="w3-row">
-						<div class="w3-col m2 l2"><img src="{{ their_avatar }}"></div>
+						<div class="w3-col m2 l2"><img src="{{ their_avatar }}" onerror="this.style.display='none'"></div>
 						<div class="w3-col m10 l10">
 							<div style="text-align: left;">
 							{% if msg.reply is not none %}