From 5cfda324cb8a6766fd9bfc6ba45ecc49e32600b2 Mon Sep 17 00:00:00 2001
From: Domiziano Scarcelli <scarcelli.domiziano@gmail.com>
Date: Sat, 4 Jun 2022 12:22:40 +0200
Subject: [PATCH] Comments can be filtered by lenght

---
 .env.template       |  5 +++++
 reddit/subreddit.py | 19 ++++++++++++-------
 2 files changed, 17 insertions(+), 7 deletions(-)

diff --git a/.env.template b/.env.template
index 5ab4ae4..bd2cbe4 100644
--- a/.env.template
+++ b/.env.template
@@ -9,3 +9,8 @@ REDDIT_2FA=""
 THEME=""
 
 SUBREDDIT=""
+
+# Filters the comments by range of lenght (min and max characters)
+# Min has to be less or equal to max
+# DO NOT INSERT ANY SPACES BETWEEN THE COMMA AND THE VALUES
+COMMENT_LENGHT_RANGE = "min,max"
\ No newline at end of file
diff --git a/reddit/subreddit.py b/reddit/subreddit.py
index 5d020fe..5212375 100644
--- a/reddit/subreddit.py
+++ b/reddit/subreddit.py
@@ -1,3 +1,4 @@
+from numpy import Infinity
 from utils.console import print_markdown, print_step, print_substep
 import praw
 import random
@@ -58,13 +59,17 @@ def get_subreddit_threads():
         content["comments"] = []
 
         for top_level_comment in submission.comments:
-            content["comments"].append(
-                {
-                    "comment_body": top_level_comment.body,
-                    "comment_url": top_level_comment.permalink,
-                    "comment_id": top_level_comment.id,
-                }
-            )
+            COMMENT_LENGHT_RANGE = [0, Infinity]
+            if os.getenv("COMMENT_LENGHT_RANGE"):
+                COMMENT_LENGHT_RANGE = [int(i) for i in os.getenv("COMMENT_LENGHT_RANGE").split(",")]                
+            if COMMENT_LENGHT_RANGE[0] <= len(top_level_comment.body) <= COMMENT_LENGHT_RANGE[1]:
+                content["comments"].append(
+                    {
+                        "comment_body": top_level_comment.body,
+                        "comment_url": top_level_comment.permalink,
+                        "comment_id": top_level_comment.id,
+                    }
+                )
 
     except AttributeError as e:
         pass