fix: Disable mail system (#316)

* fix: Disable mail system

Author: orronai

README.md

@@ -79,6 +79,11 @@ Enter http://127.0.0.1:8080, and the initial credentials should appear in your t
 
 After logging in, use [localhost admin](https://127.0.0.1:8080/admin) to modify entries in the database.
 
+In case you want to enable the mail system:
+
+1. Insert your mail details in the configuration file.
+2. Delete the `DISABLE_MAIL` line.
+
 
 ## Code modification check list
 ### Run flake8 

lms/lmsweb/config.py.example

@@ -22,6 +22,7 @@ MAIL_USE_TLS = False
 MAIL_USERNAME = 'username@gmail.com'
 MAIL_PASSWORD = 'password'
 MAIL_DEFAULT_SENDER = 'username@gmail.com'
+DISABLE_MAIL = True  # On production, delete this line!
 
 # ADMIN PANEL
 FLASK_ADMIN_FLUID_LAYOUT = True

lms/utils/mail.py

@@ -1,12 +1,25 @@
+from functools import wraps
+
 from flask import url_for
 from flask_babel import gettext as _  # type: ignore
 from flask_mail import Message  # type: ignore
 
 from lms.lmsdb.models import User
-from lms.lmsweb import config, webmail
+from lms.lmsweb import config, webapp, webmail
 from lms.models.users import generate_user_token
 
 
+def send_message(func):
+    @wraps(func)
+    def wrapper(*args, **kwargs):
+        msg = func(*args, **kwargs)
+        if not webapp.config.get('DISABLE_MAIL'):
+            webmail.send(msg)
+
+    return wrapper
+
+
+@send_message
 def send_confirmation_mail(user: User) -> Message:
     token = generate_user_token(user)
     subject = _('מייל אימות - %(site_name)s', site_name=config.SITE_NAME)
@@ -18,9 +31,10 @@ def send_confirmation_mail(user: User) -> Message:
         'שלום %(fullname)s,\nלינק האימות שלך למערכת הוא: %(link)s',
         fullname=user.fullname, link=link,
     )
-    webmail.send(msg)
+    return msg
 
 
+@send_message
 def send_reset_password_mail(user: User) -> Message:
     token = generate_user_token(user)
     subject = _('מייל איפוס סיסמה - %(site_name)s', site_name=config.SITE_NAME)
@@ -32,9 +46,10 @@ def send_reset_password_mail(user: User) -> Message:
         'שלום %(fullname)s,\nלינק לצורך איפוס הסיסמה שלך הוא: %(link)s',
         fullname=user.fullname, link=link,
     )
-    webmail.send(msg)
+    return msg
 
 
+@send_message
 def send_change_password_mail(user: User) -> Message:
     subject = _('שינוי סיסמה - %(site_name)s', site_name=config.SITE_NAME)
     msg = Message(subject, recipients=[user.mail_address])
@@ -45,4 +60,4 @@ def send_change_password_mail(user: User) -> Message:
         fullname=user.fullname, site_name=config.SITE_NAME,
         site_mail=config.MAIL_DEFAULT_SENDER,
     )
-    webmail.send(msg)
+    return msg