simple-login/app
1
1
Fork 0
mirror of https://github.com/simple-login/app.git synced 2025-11-11 18:30:49 +08:00
Code Issues Projects Releases Packages Wiki Activity

able to set a different host than 127.0.0.1 and apply black format

Browse source
This commit is contained in:
Son NK 2020-09-30 10:32:21 +02:00
parent 078368362c
commit 91e3cc5dcb
1 changed files with 30 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
app/spamassassin_utils.py
View file

@ -4,19 +4,19 @@ https://github.com/petermat/spamassassin_client
import socket, select, re, logging import socket, select, re, logging
from io import BytesIO from io import BytesIO
divider_pattern = re.compile(br'^(.*?)\r?\n(.*?)\r?\n\r?\n', re.DOTALL) divider_pattern = re.compile(br"^(.*?)\r?\n(.*?)\r?\n\r?\n", re.DOTALL)
first_line_pattern = re.compile(br'^SPAMD/[^ ]+ 0 EX_OK$') first_line_pattern = re.compile(br"^SPAMD/[^ ]+ 0 EX_OK$")
class SpamAssassin(object): class SpamAssassin(object):
def __init__(self, message, timeout=20): def __init__(self, message, timeout=20, host="127.0.0.1"):
self.score = None self.score = None
self.symbols = None self.symbols = None
# Connecting # Connecting
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM) client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
client.settimeout(timeout) client.settimeout(timeout)
client.connect(('127.0.0.1', 783)) client.connect((host, 783))
# Sending # Sending
client.sendall(self._build_message(message)) client.sendall(self._build_message(message))
@ -28,11 +28,11 @@ class SpamAssassin(object):
ready = select.select([client], [], [], timeout) ready = select.select([client], [], [], timeout)
if ready[0] is None: if ready[0] is None:
# Kill with Timeout! # Kill with Timeout!
logging.info('[SpamAssassin] - Timeout ({0}s)!'.format(str(timeout))) logging.info("[SpamAssassin] - Timeout ({0}s)!".format(str(timeout)))
break break
data = client.recv(4096) data = client.recv(4096)
if data == b'': if data == b"":
break break
resfp.write(data) resfp.write(data)
@ -46,14 +46,14 @@ class SpamAssassin(object):
def _build_message(self, message): def _build_message(self, message):
reqfp = BytesIO() reqfp = BytesIO()
data_len = str(len(message)).encode() data_len = str(len(message)).encode()
reqfp.write(b'REPORT SPAMC/1.2\r\n') reqfp.write(b"REPORT SPAMC/1.2\r\n")
reqfp.write(b'Content-Length: ' + data_len + b'\r\n') reqfp.write(b"Content-Length: " + data_len + b"\r\n")
reqfp.write(b'User: cx42\r\n\r\n') reqfp.write(b"User: cx42\r\n\r\n")
reqfp.write(message) reqfp.write(message)
return reqfp.getvalue() return reqfp.getvalue()
def _parse_response(self, response): def _parse_response(self, response):
if response == b'': if response == b"":
logging.info("[SPAM ASSASSIN] Empty response") logging.info("[SPAM ASSASSIN] Empty response")
return None return None
@ -65,7 +65,7 @@ class SpamAssassin(object):
first_line = match.group(1) first_line = match.group(1)
headers = match.group(2) headers = match.group(2)
body = response[match.end(0):] body = response[match.end(0) :]
# Checking response is good # Checking response is good
match = first_line_pattern.match(first_line) match = first_line_pattern.match(first_line)
@ -74,19 +74,20 @@ class SpamAssassin(object):
logging.error(first_line) logging.error(first_line)
return None return None
report_list = [s.strip() for s in body.decode('utf-8').strip().split('\n')] report_list = [s.strip() for s in body.decode("utf-8").strip().split("\n")]
linebreak_num = report_list.index([s for s in report_list if "---" in s][0]) linebreak_num = report_list.index([s for s in report_list if "---" in s][0])
tablelists = [s for s in report_list[linebreak_num + 1:]] tablelists = [s for s in report_list[linebreak_num + 1 :]]
self.report_fulltext = '\n'.join(report_list)
self.report_fulltext = "\n".join(report_list)
# join line when current one is only wrap of previous # join line when current one is only wrap of previous
tablelists_temp = [] tablelists_temp = []
if tablelists: if tablelists:
for counter, tablelist in enumerate(tablelists): for counter, tablelist in enumerate(tablelists):
if len(tablelist)>1: if len(tablelist) > 1:
if (tablelist[0].isnumeric() or tablelist[0] == '-') and (tablelist[1].isnumeric() or tablelist[1] == '.'): if (tablelist[0].isnumeric() or tablelist[0] == "-") and (
tablelist[1].isnumeric() or tablelist[1] == "."
):
tablelists_temp.append(tablelist) tablelists_temp.append(tablelist)
else: else:
if tablelists_temp: if tablelists_temp:
@ -96,10 +97,19 @@ class SpamAssassin(object):
# create final json # create final json
self.report_json = dict() self.report_json = dict()
for tablelist in tablelists: for tablelist in tablelists:
wordlist = re.split('\s+', tablelist) wordlist = re.split("\s+", tablelist)
self.report_json[wordlist[1]] = {'partscore': float(wordlist[0]), 'description': ' '.join(wordlist[1:])} self.report_json[wordlist[1]] = {
"partscore": float(wordlist[0]),
"description": " ".join(wordlist[1:]),
}
headers = headers.decode('utf-8').replace(' ', '').replace(':', ';').replace('/', ';').split(';') headers = (
headers.decode("utf-8")
.replace(" ", "")
.replace(":", ";")
.replace("/", ";")
.split(";")
)
self.score = float(headers[2]) self.score = float(headers[2])
def get_report_json(self): def get_report_json(self):
@ -113,6 +123,3 @@ class SpamAssassin(object):
def get_fulltext(self): def get_fulltext(self):
return self.report_fulltext return self.report_fulltext