Update utils/instaloader_utils.py

utils/instaloader_utils.py

@@ -1,6 +1,6 @@
 import instaloader
-from typing import List, Dict
-import time
+from typing import List, Dict, Optional
+import instaloader.exceptions
 
 # Initialize Instaloader
 L = instaloader.Instaloader()
@@ -10,47 +10,36 @@ def fetch_user_posts(username: str, max_posts: int = 50) -> List[Dict]:
     Fetch posts from a given Instagram profile (public data only).
 
     Args:
-        username (str): Instagram username.
+        username (str): The username of the Instagram profile.
         max_posts (int): Maximum number of posts to fetch.
 
     Returns:
         List[Dict]: A list of dictionaries containing post details.
     """
     try:
-        # Load the profile
         profile = instaloader.Profile.from_username(L.context, username)
         posts = []
-
-        # Fetch posts
         for post in profile.get_posts():
             posts.append({
                 "username": username,
-                "caption": post.caption,
-                "hashtags": post.caption_hashtags,
+                "caption": post.caption or "",  # Handle None captions
+                "hashtags": post.caption_hashtags if post.caption else [],
                 "likes": post.likes,
                 "comments": post.comments,
                 "date": post.date_utc.isoformat(),
                 "image_url": post.url
             })
-
-            # Stop if we've reached the maximum number of posts
-            if len(posts) >= max_posts:
+            if len(posts) >= max_posts:  # Limit the number of posts
                 break
-
-            # Add a delay to avoid hitting Instagram's rate limits
-            time.sleep(2)  # 2-second delay between requests
-
         return posts
-
     except instaloader.exceptions.ProfileNotExistsException:
-        print(f"Error: The profile '{username}' does not exist.")
-        return []
-    except instaloader.exceptions.QueryReturnedBadRequestException:
-        print("Error: Instagram blocked the request. Please wait before trying again.")
-        return []
+        print(f"Error: Profile '{username}' does not exist.")
+    except instaloader.exceptions.ConnectionException:
+        print("Error: Unable to connect to Instagram. Check your connection.")
     except Exception as e:
-        print(f"Unexpected error fetching posts for {username}: {e}")
-        return []
+        print(f"Unexpected error while fetching posts for {username}: {e}")
+    return []
+
 
 
 def find_similar_accounts(username: str, rapidapi_key: str) -> List[str]: