From 75215e2e6dd1727916179584c14a7463dd5394c5 Mon Sep 17 00:00:00 2001
From: muskit <15199219+muskit@users.noreply.github.com>
Date: Sun, 20 Aug 2023 16:13:38 -0700
Subject: [PATCH] clean up create_ttweet_from_image()

---
 src/util.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/util.py b/src/util.py
index 02e381f..5c8a17e 100644
--- a/src/util.py
+++ b/src/util.py
@@ -53,19 +53,19 @@ def get_key_from_value(d: dict, val):
 # FIXME: web_auth_token under rate-limitation will fail to screenshot
 async def create_ttweet_image(ttweet):
     tc = TweetCapture()
-    tc.cookies = [{'name': 'auth_token', 'value': dotenv_values()['web_auth_token']}]
+    auth_token = dotenv_values().get('web_auth_token')
+    if auth_token:
+        tc.cookies = [{'name': 'auth_token', 'value': auth_token}]
     if 'linux' in sys.platform:
         # Linux chromedriver path
         tc.driver_path = '/usr/bin/chromedriver'
     filename = f'{get_project_dir()}/img.png'
-    url = ttweet.url()
     img = None
-    print(url)
     try: os.remove(filename)
     except: pass
     try:
         img = await tc.screenshot(
-            url=url,
+            url=ttweet.url(),
             path=filename,
             mode=4,
             night_mode=1,
@@ -76,9 +76,9 @@ async def create_ttweet_image(ttweet):
         print('unable to create tweet image')
         traceback.print_exc()
         return None
-    else:
-        print(f'successfully saved {img}')
-        return img
+    
+    print(f'successfully saved {img}')
+    return img
 
 def get_tweet_url(id, username):
     return f'https://www.twitter.com/{username}/status/{id}'