fix: avg sentiment calculation

Gowtham1729 · Dec 18, 2023 · 6fdc3b1 · 6fdc3b1
1 parent 0d1ee70
commit 6fdc3b1
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/applications/data_analyzer/analyzer.py b/applications/data_analyzer/analyzer.py
@@ -258,7 +258,17 @@ def update_daily_analysis(self, news: News, date: str):
             """
             UPDATE news_analysis
             SET 
-                average_sentiment = (SELECT AVG(sentiment) FROM news_news WHERE sentiment!=0), 
+                average_sentiment = 
+                (
+                    SELECT 
+                        AVG(sentiment) 
+                    FROM 
+                        news_news 
+                    WHERE 
+                        sentiment!=0 AND 
+                        symbol = %s AND 
+                        DATE(publish_time) = %s
+                ), 
                 total_news = total_news + 1, 
                 positive_news = positive_news + %s, 
                 negative_news = negative_news + %s, 
@@ -267,6 +277,8 @@ def update_daily_analysis(self, news: News, date: str):
             WHERE symbol = %s AND date = %s
             """,
             (
+                news.symbol,
+                date,
                 1 if news.sentiment > 0 else 0,
                 1 if news.sentiment < 0 else 0,
                 1 if news.sentiment == 0 else 0,