From 36a76c4af7c4921a33fcd37bbeb65735309fa1ca Mon Sep 17 00:00:00 2001
From: pablusha <pablusha@pablusha.ru>
Date: Fri, 15 Aug 2025 00:27:32 +0300
Subject: [PATCH] added ability to change banlist file

---
 config.py | 1 +
 main.py   | 2 +-
 utils.py  | 2 +-
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/config.py b/config.py
index 2fa2043..8e436f1 100644
--- a/config.py
+++ b/config.py
@@ -6,6 +6,7 @@ port=7856
 
 log_file=None
 preset_file="preset.html"
+banlist_file="banlist.lol"
 directory="files/"
 
 read_buffer=16384
diff --git a/main.py b/main.py
index d29ceff..82fd008 100644
--- a/main.py
+++ b/main.py
@@ -96,7 +96,7 @@ class WebServer:
             return
         
 
-        if addr[0] in utils.get_banlist(): # banlist for pidors
+        if addr[0] in utils.get_banlist(config.banlist_file): # banlist for pidors
             await self.log(config.err_msgs[403], addr, file_path)
             file_size = os.path.getsize(config.err_files[403])
             await self.send_headers(writer, 403, file_size)
diff --git a/utils.py b/utils.py
index a2ba424..ce66892 100644
--- a/utils.py
+++ b/utils.py
@@ -13,5 +13,5 @@ def get_create_time(path: str, format: str="%a %b %e %H:%M:%S %Z %Y") -> str:
     return create_datetime.strftime(format)
 
 
-def get_banlist() -> list:
+def get_banlist(path: str) -> list:
     return list(set(open("banlist.lol").readlines()))
\ No newline at end of file