Add WIP for whitelist and blacklist

discord_rss_bot/feeds.py

@@ -24,7 +24,7 @@ Exceptions:
 from typing import Iterable
 
 from discord_webhook import DiscordWebhook
-from reader import Entry, Reader
+from reader import Entry, Reader, TagNotFoundError
 from requests import Response
 
 from discord_rss_bot.settings import get_reader
@@ -57,17 +57,79 @@ def send_to_discord(custom_reader: Reader | None = None, feed=None, do_once=Fals
 
     for entry in entries:
         # Set the webhook to read, so we don't send it again.
-        reader.set_entry_read(entry, True)
+        reader.set_entry_read(entry, True)  # type: ignore
 
         # Get the webhook from the feed.
-        webhook_url: str = str(reader.get_tag(entry.feed_url, "webhook"))
+        try:
+            webhook_url: str = str(reader.get_tag(entry.feed_url, "webhook"))
+        except TagNotFoundError:
+            print(f"Webhook not found for feed {entry.feed_url}")
+            continue
+
         webhook_message: str = f":robot: :mega: {entry.title}\n{entry.link}"
         webhook: DiscordWebhook = DiscordWebhook(url=webhook_url, content=webhook_message, rate_limit_retry=True)
 
-        # Send the webhook.
-        response: Response = webhook.execute()
-        if not response.ok:
-            reader.set_entry_read(entry, False)
+        try:
+            whitelist_title = reader.get_tag(feed, "whitelist_title")
+        except TagNotFoundError:
+            whitelist_title = ""
+        try:
+            whitelist_summary = reader.get_tag(feed, "whitelist_summary")
+        except TagNotFoundError:
+            whitelist_summary = ""
+        try:
+            whitelist_content = reader.get_tag(feed, "whitelist_content")
+        except TagNotFoundError:
+            whitelist_content = ""
+
+        try:
+            blacklist_title = reader.get_tag(feed, "blacklist_title")
+        except TagNotFoundError:
+            blacklist_title = ""
+        try:
+            blacklist_summary = reader.get_tag(feed, "blacklist_summary")
+        except TagNotFoundError:
+            blacklist_summary = ""
+        try:
+            blacklist_content = reader.get_tag(feed, "blacklist_content")
+        except TagNotFoundError:
+            blacklist_content = ""
+
+        # Check if the entry should be sent. If on the blacklist, mark as read and continue.
+        if whitelist_title:
+            if whitelist_title.lower() in entry.title.lower():
+                print(f"Whitelisted because of title: {entry.title}")
+                response: Response = webhook.execute()
+
+                if not response.ok:
+                    print(f"Error: {response.status_code} {response.reason}")
+                    reader.set_entry_read(entry, False)  # type: ignore
+        if whitelist_summary:
+            if whitelist_summary.lower() in entry.summary.lower():
+                print(f"Whitelisted because of summary: {entry.title}")
+                response: Response = webhook.execute()
+
+                if not response.ok:
+                    print(f"Error: {response.status_code} {response.reason}")
+                    reader.set_entry_read(entry, False)  # type: ignore
+        # if whitelist_content.lower() in entry.content.lower():
+
+        if blacklist_title:
+            if blacklist_title.lower() in entry.title.lower():
+                print(f"Blacklisted because of title: {entry.title}")
+                reader.set_entry_read(entry, True)  # type: ignore
+        if blacklist_summary:
+            if blacklist_summary.lower() in entry.summary.lower():
+                print(f"Blacklisted because of summary: {entry.title}")
+                reader.set_entry_read(entry, True)  # type: ignore
+        # if blacklist_content.lower() in entry.content.lower():
+
+        else:
+            response: Response = webhook.execute()
+
+            if not response.ok:
+                print(f"Error: {response.status_code} {response.reason}")
+                reader.set_entry_read(entry, False)  # type: ignore
 
         # If we only want to send one entry, we will break the loop. This is used when testing this function.
         if do_once: