beenull/simple-login
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

do not hardcode spamd user

Browse source
This commit is contained in:
Son NK 2020-09-30 14:01:49 +02:00
parent 6a4622fca9
commit 770b15aba3
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
app/spamassassin_utils.py
View file

@ -9,9 +9,10 @@ first_line_pattern = re.compile(br"^SPAMD/[^ ]+ 0 EX_OK$")
class SpamAssassin(object):
def __init__(self, message, timeout=20, host="127.0.0.1"):
def __init__(self, message, timeout=20, host="127.0.0.1", spamd_user="spamd"):
self.score = None
self.symbols = None
self.spamd_user = spamd_user
# Connecting
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
@ -48,7 +49,7 @@ class SpamAssassin(object):
data_len = str(len(message)).encode()
reqfp.write(b"REPORT SPAMC/1.2\r\n")
reqfp.write(b"Content-Length: " + data_len + b"\r\n")
reqfp.write(b"User: spamd\r\n\r\n")
reqfp.write(f"User: {self.spamd_user}\r\n\r\n".encode())
reqfp.write(message)
return reqfp.getvalue()