key for cache

mBaratta96 · Aug 6, 2023 · c038e3a · c038e3a
1 parent 0e4d3d3
commit c038e3a
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/letterboxd_stats/web_scraper.py b/letterboxd_stats/web_scraper.py
@@ -113,12 +113,13 @@ def _get_tmdb_id_from_web(link: str, is_diary: bool):
 
 def get_tmdb_id(link: str, is_diary=False):
     tmdb_id_cache = shelve.open(cache_path, writeback=False, protocol=5)
-    if link in tmdb_id_cache:
-        id = tmdb_id_cache[link]
+    key = link.split("/")[-1]
+    if key in tmdb_id_cache:
+        id = tmdb_id_cache[key]
     else:
         try:
             id = _get_tmdb_id_from_web(link, is_diary)
-            tmdb_id_cache[link] = id
+            tmdb_id_cache[key] = id
         except ValueError as e:
             print(e)
             id = None