144 lines
4.6 KiB
Python
144 lines
4.6 KiB
Python
"""Provide basic settings and views"""
|
|
|
|
import fileinput
|
|
import os
|
|
import re
|
|
import shutil
|
|
from pathlib import Path
|
|
from flask import request, render_template
|
|
from flask_basicauth import BasicAuth
|
|
from jinja2 import Template
|
|
from autoreply_editor import app
|
|
|
|
basic_auth = BasicAuth(app)
|
|
|
|
def qmail_status(user):
|
|
"""Find out whether the filter is currently activated for the given mail user"""
|
|
with open(get_qmailfile(user), encoding="utf8") as qmailfile:
|
|
# trigger to check whether we have to re-open the file for initialisation
|
|
initialise = False
|
|
|
|
|
|
# try to find active filter
|
|
if re.search(
|
|
rf"^{re.escape(get_maildropline(user))}$", qmailfile.read(), re.MULTILINE
|
|
):
|
|
# if get_maildropline(user) in qmailfile.read():
|
|
return True
|
|
else:
|
|
# jump back to top of file because we've read it in the "if"
|
|
qmailfile.seek(0)
|
|
|
|
# filter is deactivated (commented)
|
|
if re.search(
|
|
rf"^#{re.escape(get_maildropline(user))}$",
|
|
qmailfile.read(),
|
|
re.MULTILINE,
|
|
):
|
|
# if f"#{get_maildropline(user)}" in qmailfile.read():
|
|
return False
|
|
else:
|
|
initialise = True
|
|
|
|
# Append a commented filter command to the file
|
|
if initialise:
|
|
with open(
|
|
get_qmailfile(user), mode="a", encoding="utf8"
|
|
) as qmailfile:
|
|
qmailfile.write(f"#{get_maildropline(user)}")
|
|
|
|
return False
|
|
|
|
|
|
def get_messagefile(user):
|
|
"""Return message file of user, and create if necessary"""
|
|
filepath = f"message-{user.split('@')[0]}.txt"
|
|
if not os.path.isfile(filepath):
|
|
Path(filepath).touch()
|
|
|
|
return filepath
|
|
|
|
|
|
def get_qmailfile(user):
|
|
"""Return qmail file of user, and create if necessary based on default qmail"""
|
|
qmail_base = f"{str(Path.home())}/.qmail"
|
|
filepath = f"{qmail_base}-{user.split('@')[0]}"
|
|
if not os.path.isfile(filepath):
|
|
shutil.copy(f"{qmail_base}-default", filepath)
|
|
|
|
return filepath
|
|
|
|
|
|
def get_filterfile(user, name):
|
|
"""Return .filter-autoreply file of user, and create if necessary"""
|
|
filter_base = f"{str(Path.home())}/.filter-autoreply"
|
|
filepath = f"{filter_base}-{user.split('@')[0]}"
|
|
if not os.path.isfile(filepath):
|
|
with open(os.path.join(app.root_path, "templates/filterfile.j2"), encoding="utf-8") as template:
|
|
filterconfig = Template(template.read()).render(
|
|
email=user,
|
|
user=user.split('@')[0],
|
|
name=name,
|
|
messagefile=get_messagefile(user)
|
|
)
|
|
|
|
with open(filepath, "w", encoding="utf-8") as filterfile:
|
|
filterfile.write(filterconfig)
|
|
|
|
return filepath
|
|
|
|
|
|
def get_maildropline(user):
|
|
return f"|maildrop {get_filterfile(user, 'FOOBAR')}"
|
|
|
|
|
|
@app.route("/")
|
|
def index():
|
|
with open(get_messagefile(app.config.get("MAIL_USER")), "r", encoding="utf-8") as messagefile:
|
|
message = messagefile.read()
|
|
|
|
return render_template(
|
|
"index.html",
|
|
message=message,
|
|
qmail_status=qmail_status(app.config.get("MAIL_USER")),
|
|
)
|
|
|
|
|
|
@app.route("/", methods=["POST"])
|
|
def index_post():
|
|
if request.method == "POST":
|
|
if request.form["action"] == "message":
|
|
input_message = request.form["message"]
|
|
with open(get_messagefile(app.config.get("MAIL_USER")), "w", encoding="utf-8") as messagefile:
|
|
messagefile.write(str(input_message))
|
|
result = "Success: The autoreply message has been updated!"
|
|
|
|
if request.form["action"] == "qmail":
|
|
# define whether to set a comment
|
|
if request.form["status"] == "on":
|
|
preis = "#"
|
|
preshould = ""
|
|
else:
|
|
preis = ""
|
|
preshould = "#"
|
|
|
|
with fileinput.FileInput(
|
|
get_qmailfile(app.config.get("MAIL_USER")),
|
|
inplace=True,
|
|
backup=".bak",
|
|
) as qmailfile:
|
|
for line in qmailfile:
|
|
print(
|
|
line.replace(
|
|
f"{preis}{get_maildropline(app.config.get('MAIL_USER'))}", f"{preshould}{get_maildropline(app.config.get('MAIL_USER'))}"
|
|
),
|
|
end="",
|
|
)
|
|
|
|
result = f"Success: the autoreply is now {request.form['status']}."
|
|
|
|
try:
|
|
return render_template("result.html", result=result)
|
|
except UnboundLocalError:
|
|
return render_template("result.html", result="Something went terribly wrong!")
|