fix: get conversations not working for more than 20

jupyter-naas · Mar 14, 2024 · 3d71575 · 3d71575
1 parent 094531e
commit 3d71575
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/naas_drivers/tools/linkedin.py b/naas_drivers/tools/linkedin.py
@@ -1016,8 +1016,9 @@ def get_conversations(
             count = limit
         if limit > limit_max:
             limit = limit_max
+
+        params = {"count": count}
         while True:
-            params = {"count": count}
             req_url = f"{LINKEDIN_API}/message/getConversations?{urllib.parse.urlencode(params, safe='(),')}"
             res = requests.post(req_url, json=self.cookies, headers=HEADERS)
             res.raise_for_status()
@@ -1036,8 +1037,8 @@ def get_conversations(
             if len(df) >= limit:
                 break
             # Set created before params
-            last_message_sent_at = tmp_df["LAST_MESSAGE_SENT_AT"].iloc[0]
-            created_before = datetime.strptime(last_message_sent_at, DATETIME_FORMAT)
+            last_message_sent_at = tmp_df["LAST_MESSAGE_SENT_AT"].iloc[-1]
+            created_before = int(datetime.strptime(last_message_sent_at, DATETIME_FORMAT).strftime("%s") + "000")
             params["created_before"] = created_before
             if sleep:
                 time.sleep(TIME_SLEEP)