From e05c7fefabb983224c9a3fde63ca6246332214e9 Mon Sep 17 00:00:00 2001
From: Claromes <claromes@hey.com>
Date: Tue, 16 Jul 2024 19:08:45 -0300
Subject: [PATCH] update module viz - add get method

---
 waybacktweets/api/visualize.py | 40 ++++++++++++++++------------------
 1 file changed, 19 insertions(+), 21 deletions(-)

diff --git a/waybacktweets/api/visualize.py b/waybacktweets/api/visualize.py
index 93ad80a..bb4def0 100644
--- a/waybacktweets/api/visualize.py
+++ b/waybacktweets/api/visualize.py
@@ -114,12 +114,12 @@ class HTMLTweetsVisualizer:
                 tweet = self.json_path[index]
                 html += '<div class="tweet">\n'
 
-                if not tweet["available_tweet_text"]:
+                if not tweet.get("available_tweet_text"):
                     iframe_src = {
-                        "Archived Tweet": tweet["archived_tweet_url"],
-                        "Parsed Archived Tweet": tweet["parsed_archived_tweet_url"],
-                        "Original Tweet": tweet["original_tweet_url"],
-                        "Parsed Tweet": tweet["parsed_tweet_url"],
+                        "Archived Tweet": tweet.get("archived_tweet_url"),
+                        "Parsed Archived Tweet": tweet.get("parsed_archived_tweet_url"),
+                        "Original Tweet": tweet.get("original_tweet_url"),
+                        "Parsed Tweet": tweet.get("parsed_tweet_url"),
                     }
 
                     for key, value in iframe_src.items():
@@ -149,25 +149,23 @@ class HTMLTweetsVisualizer:
                             index=index, url=value, key_cleaned=key_cleaned
                         )
 
-                if tweet["available_tweet_text"]:
+                if tweet.get("available_tweet_text"):
                     html += "<br>\n"
-                    html += f'<p><strong class="content">Available Tweet Content:</strong> {tweet["available_tweet_text"]}</p>\n'
-                    html += f'<p><strong class="content">Available Tweet Is Retweet:</strong> {tweet["available_tweet_is_RT"]}</p>\n'
-                    html += f'<p><strong class="content">Available Tweet Username:</strong> {tweet["available_tweet_info"]}</p>\n'
+                    html += f'<p><strong class="content">Available Tweet Content:</strong> {tweet.get("available_tweet_text")}</p>\n'
+                    html += f'<p><strong class="content">Available Tweet Is Retweet:</strong> {tweet.get("available_tweet_is_RT")}</p>\n'
+                    html += f'<p><strong class="content">Available Tweet Username:</strong> {tweet.get("available_tweet_info")}</p>\n'
 
                 html += "<br>\n"
-                html += f'<p><strong>Archived Tweet:</strong> <a href="{tweet["archived_tweet_url"]}" target="_blank">{tweet["archived_tweet_url"]}</a></p>\n'
-                html += f'<p><strong>Parsed Archived Tweet:</strong> <a href="{tweet["parsed_archived_tweet_url"]}" target="_blank">{tweet["parsed_archived_tweet_url"]}</a></p>\n'
-                html += f'<p><strong>Original Tweet:</strong> <a href="{tweet["original_tweet_url"]}" target="_blank">{tweet["original_tweet_url"]}</a></p>\n'
-                html += f'<p><strong>Parsed Tweet:</strong> <a href="{tweet["parsed_tweet_url"]}" target="_blank">{tweet["parsed_tweet_url"]}</a></p>\n'
-                html += f'<p><strong>Archived URL Key:</strong> {tweet["archived_urlkey"]}</p>\n'
-                html += f'<p><strong>Archived Timestamp:</strong> {timestamp_parser(tweet["archived_timestamp"])} ({tweet["archived_timestamp"]})</p>\n'
-                html += f'<p><strong>Archived mimetype:</strong> {tweet["archived_mimetype"]}</p>\n'
-                html += f'<p><strong>Archived Statuscode:</strong> {tweet["archived_statuscode"]}</p>\n'
-                html += (
-                    f'<p><strong>Archived Digest:</strong> {tweet["archived_digest"]}\n'
-                )
-                html += f'<p><strong>Archived Length:</strong> {tweet["archived_length"]}</p>\n'
+                html += f'<p><strong>Archived Tweet:</strong> <a href="{tweet.get("archived_tweet_url")}" target="_blank">{tweet.get("archived_tweet_url")}</a></p>\n'
+                html += f'<p><strong>Parsed Archived Tweet:</strong> <a href="{tweet.get("parsed_archived_tweet_url")}" target="_blank">{tweet.get("parsed_archived_tweet_url")}</a></p>\n'
+                html += f'<p><strong>Original Tweet:</strong> <a href="{tweet.get("original_tweet_url")}" target="_blank">{tweet.get("original_tweet_url")}</a></p>\n'
+                html += f'<p><strong>Parsed Tweet:</strong> <a href="{tweet.get("parsed_tweet_url")}" target="_blank">{tweet.get("parsed_tweet_url")}</a></p>\n'
+                html += f'<p><strong>Archived URL Key:</strong> {tweet.get("archived_urlkey")}</p>\n'
+                html += f'<p><strong>Archived Timestamp:</strong> {timestamp_parser(tweet.get("archived_timestamp"))} ({tweet.get("archived_timestamp")})</p>\n'
+                html += f'<p><strong>Archived mimetype:</strong> {tweet.get("archived_mimetype")}</p>\n'
+                html += f'<p><strong>Archived Statuscode:</strong> {tweet.get("archived_statuscode")}</p>\n'
+                html += f'<p><strong>Archived Digest:</strong> {tweet.get("archived_digest")}\n'
+                html += f'<p><strong>Archived Length:</strong> {tweet.get("archived_length")}</p>\n'
                 html += "</div>\n"
 
             html += "</div>\n</div>\n"  # Closes the page div and the container
-- 
2.34.1