From 92fa74553341d4cdac134b0cafdaf16fa94b3afa Mon Sep 17 00:00:00 2001
From: Qetesh <4559341+Qetesh@users.noreply.github.com>
Date: Sat, 12 Oct 2024 01:08:06 +0800
Subject: support Deny-/allow list wildcard url

---
 core/__init__.py     |  0
 core/entry_filter.py | 31 +++++++++++++++++++++++++++++++
 2 files changed, 31 insertions(+)
 create mode 100644 core/__init__.py
 create mode 100644 core/entry_filter.py

(limited to 'core')

diff --git a/core/__init__.py b/core/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/core/entry_filter.py b/core/entry_filter.py
new file mode 100644
index 0000000..62ab9f5
--- /dev/null
+++ b/core/entry_filter.py
@@ -0,0 +1,31 @@
+import fnmatch
+
+def filter_entry(config, agent, entry):
+    start_with_list = [name[1]['title'] for name in config['agents'].items()]
+    style_block = [name[1]['style_block'] for name in config['agents'].items()]
+    [start_with_list.append('<pre') for i in style_block if i]
+
+    # Todo Compatible with whitelist/blacklist parameter, to be removed
+    allow_list = agent[1].get('allow_list') if agent[1].get('allow_list') is not None else agent[1].get('whitelist')
+    deny_list = agent[1]['deny_list'] if agent[1].get('deny_list') is not None else agent[1].get('blacklist')
+
+    # filter, if not content starts with start flag
+    if not entry['content'].startswith(tuple(start_with_list)):
+
+        # filter, if in allow_list
+        if allow_list is not None:
+            if any(fnmatch.fnmatch(entry['feed']['site_url'], pattern) for pattern in allow_list):
+                return True
+
+        # filter, if not in deny_list
+        elif deny_list is not None:
+            if any(fnmatch.fnmatch(entry['feed']['site_url'], pattern) for pattern in deny_list):
+                return False
+            else:
+                return True
+
+        # filter, if allow_list and deny_list are both None
+        elif allow_list is None and deny_list is None:
+            return True
+
+    return False
-- 
cgit v1.2.3