Update func_facebook.py

func_facebook.py

@@ -15,15 +15,23 @@ def hide_negative_comments(token):
         posts_data = graph.get_object(id='me/posts', fields='id,message,created_time')
         return posts_data.get('data', [])
 
-    def get_comments_for_all_posts(posts):
-        all_comments = []
+
+
+    def get_comments_for_post(post_id):
         graph = init_facebook_client(token)
-        for post in posts:
-            comments_data = graph.get_object(id=f'{post["id"]}/comments', fields='message,created_time,id,is_hidden')
-            visible_comments = [comment for comment in comments_data.get('data', []) if
-                                not comment.get('is_hidden', False)]
-            all_comments.extend(visible_comments)
-        return all_comments
+        comments = []
+        url = f'{post_id}/comments'
+        params = {'fields': 'id,message,is_hidden'}
+        while True:
+            comments_data = graph.get_object(id=url, **params)
+            visible_comments = [comment for comment in comments_data.get('data', []) if not comment.get('is_hidden', False)]
+            comments.extend(visible_comments)
+            if 'paging' in comments_data and 'next' in comments_data['paging']:
+                url = comments_data['paging']['next']
+                params = {}  # The 'next' URL already includes access token and fields
+            else:
+                break
+        return comments
 
     def filter_comments(comments, sentiments):
         print("Фильтруем негативные комментарии.")
@@ -48,40 +56,35 @@ def hide_negative_comments(token):
         print("Нет постов для обработки.")
         return []
 
-    comments = get_comments_for_all_posts(posts)
-    if not comments:
-        print("Нет комментариев для обработки.")
-        return []
-
-    comments_text = [comment['message'] for comment in comments]
-    sentiments = analyze_sentiment(comments_text)
-    negative_comments = filter_comments(comments, sentiments)
-
-    hidden_comments = []
-    for comment in negative_comments:
-        if hide_comment(comment['id']):
-            hidden_comments.append({
-                'id': comment['id'],
-                'message': comment['message']
+    hidden_comments_per_post = []
+    for post in posts:
+        post_id = post['id']
+        post_message = post.get('message', '')
+        comments = get_comments_for_post(post_id)
+        if not comments:
+            print(f"Нет комментариев для поста {post_id}.")
+            continue
+
+        comments_text = [comment['message'] for comment in comments]
+        sentiments = analyze_sentiment(comments_text)
+        negative_comments = filter_comments(comments, sentiments)
+
+        hidden_comments = []
+        for comment in negative_comments:
+            if hide_comment(comment['id']):
+                hidden_comments.append({
+                    'id': comment['id'],
+                    'message': comment['message']
+                })
+
+        if hidden_comments:
+            hidden_comments_per_post.append({
+                'post_id': post_id,
+                'post_message': post_message,
+                'hidden_comments': hidden_comments
             })
 
-    return hidden_comments
-
-
-def get_facebook_comments(post_id, ACCESS_TOKEN):
-    print(f"Получаем комментарии для поста: {post_id}")
-    url = f"{GRAPH_API_URL}/{post_id}/comments"
-    params = {
-        'access_token': ACCESS_TOKEN,
-        'filter': 'stream'
-    }
-    response = requests.get(url, params=params)
-    if response.status_code == 200:
-        print("Комментарии успешно получены.")
-        return response.json().get('data', [])
-    else:
-        print(f"Ошибка при получении комментариев: {response.text}")
-        return []
+    return hidden_comments_per_post
 
 
 def get_page_id(page_access_token):
@@ -102,8 +105,7 @@ def get_posts(page_id, page_access_token):
     url = f"{GRAPH_API_URL}/{page_id}/posts"
     params = {
         "access_token": page_access_token,
-        "fields": "id"
-        # Максимальное количество постов за запрос
+        "fields": "id,message"
     }
     posts = []
     while True:
@@ -126,7 +128,6 @@ def get_comments(post_id, page_access_token):
     params = {
         "access_token": page_access_token,
         "fields": "id,from,message,is_hidden",
-
     }
     comments = []
     while True:
@@ -138,7 +139,7 @@ def get_comments(post_id, page_access_token):
         comments.extend(data.get("data", []))
         if 'paging' in data and 'next' in data['paging']:
             url = data['paging']['next']
-            params = {}  # Следующие запросы уже содержат все параметры в URL
+            params = {}
         else:
             break
     return comments
@@ -149,7 +150,6 @@ def has_page_replied(comment_id, page_id, page_access_token):
     params = {
         "access_token": page_access_token,
         "fields": "from{id}",
-
     }
     while True:
         response = requests.get(url, params=params)
@@ -175,13 +175,16 @@ def get_unanswered_comments(page_access_token):
 
     print(f"ID Страницы: {page_id}")
     posts = get_posts(page_id, page_access_token)
-    unanswered_comments = []
+    posts_with_unanswered_comments = []
 
     for post in posts:
         post_id = post['id']
+        post_message = post.get('message', '')
         print(f"Обработка поста: {post_id}")
         comments = get_comments(post_id, page_access_token)
 
+        unanswered_comments = []
+
         for comment in comments:
             if comment.get('is_hidden', False):
                 continue
@@ -190,7 +193,14 @@ def get_unanswered_comments(page_access_token):
             if not has_page_replied(comment_id, page_id, page_access_token):
                 unanswered_comments.append(comment)
 
-    return unanswered_comments
+        if unanswered_comments:
+            posts_with_unanswered_comments.append({
+                'post_id': post_id,
+                'post_message': post_message,
+                'unanswered_comments': unanswered_comments
+            })
+
+    return posts_with_unanswered_comments
 
 
 def reply_comment(comment_id, message, token):