look for tweets.js too

DocNow · Nov 20, 2022 · f6ef9ed · f6ef9ed
1 parent cb8f8b2
commit f6ef9ed
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -1,6 +1,6 @@
 from setuptools import setup
 
-__version__ = "0.0.8"
+__version__ = "0.0.9"
 
 def long_description():
     with open("README.md") as f:

diff --git a/twitter_archive_unshorten.py b/twitter_archive_unshorten.py
@@ -126,8 +126,13 @@ def unshorten(urls, archive_dir):
     else:
         url_map = {}
 
-    # load short/long mapping already present in tweet.js
-    url_map.update(read_url_map(join(archive_dir, "data", "tweet.js")))
+    # load short/long mapping already present in tweet.js or tweets.js
+    tweet_js = join(archive_dir, "data", "tweet.js")
+    if not os.path.isfile(tweet_js):
+        tweet_js = join(archive_dir, "data", "tweets.js")
+
+    if os.path.isfile(tweet_js):
+        url_map.update(read_url_map(tweet_js))
 
     count = 0
     for short_url in urls: