Merge pull request #58 from it-goats/feature/ITG-133-134

[ITG-133] [ITG-134] Notification for tasks by sendgrid.

bode/bode/app.py

@@ -5,6 +5,8 @@
 from flask_migrate import upgrade
 from flask_smorest import Blueprint
 
+from bode.mail_service.sending import start_notify
+
 from .config import Config
 from .extensions import api, db, migrate
 from .resources import tags, task_relations, tasks
@@ -28,6 +30,8 @@ def create_app():
         if CONFIG.AUTO_MIGRATE:
             upgrade()
 
+    start_notify(app)
+
     return app
 
 

bode/bode/config.py

@@ -17,3 +17,6 @@ def get_db_uri():
 class Config:
     DATABASE_URI = get_db_uri()
     AUTO_MIGRATE = os.getenv("AUTO_MIGRATE", False)
+
+
+sendgrid_key = os.getenv("SENDGRID_API_KEY")

bode/bode/mail_service/sending.py

@@ -0,0 +1,56 @@
+import time
+
+from datetime import datetime, timedelta
+from threading import Thread
+
+import schedule
+
+from dateutil import tz
+from sendgrid import SendGridAPIClient
+from sendgrid.helpers.mail import Mail
+
+from bode.config import sendgrid_key
+from bode.extensions import db
+from bode.models.enums import TaskStatus
+from bode.models.task.model import Task
+
+client = SendGridAPIClient(sendgrid_key)
+
+
+def send_notification(address, title, due_date, link):
+    message = Mail(
+        from_email="itgoatsteam@gmail.com",
+        to_emails=address,
+        subject="Notification",
+        html_content=f"<h3>The deadline is approaching for task:</h3><h4>TITLE: </h4>{title}<br /><h4>"
+        f"DUE DATE: </h4>{due_date}<br /><a href='{link}'><h4>CLICK</h4></a>",
+    )
+    client.send(message)
+
+
+def start_notify(app):
+    def check_task_to_notify():
+        nowutc = datetime.utcnow().replace(second=0, microsecond=0)
+        with app.app_context():
+            tasks = db.session.query(Task).filter(Task.status == TaskStatus.TODO.value).all()
+        address = "itgoatsteam@gmail.com"  # get email from database
+        for task in tasks:
+            notify_time = 60  # in minutes, get this information from task, if none => not notify
+            if notify_time is None:
+                continue
+            dt = nowutc + timedelta(minutes=notify_time)
+            due_date = datetime.fromisoformat(str(task.due_date)).replace(second=0, microsecond=0)
+            delta = dt - due_date.replace(tzinfo=None)
+            if delta.total_seconds() != 0:
+                continue
+            link = f"https://it-goats.netlify.app/task/{task.id}"
+            due_date_string = due_date.astimezone(tz.gettz("Europe/Warsaw")).strftime("%d-%m-%Y %H:%M")
+            send_notification(address, task.title, due_date_string, link)
+
+    def start_scheduler():
+        schedule.every().minute.at(":00").do(check_task_to_notify)
+        while True:
+            schedule.run_pending()
+            time.sleep(1)
+
+    Thread(target=start_scheduler, daemon=True).start()

bode/bode/models/tag/model.py

@@ -2,7 +2,7 @@
 
 from sqlalchemy.dialects.postgresql import UUID
 
-from bode.app import db
+from bode.extensions import db
 
 task_tag = db.Table(
     "task_tag", db.Column("task_id", db.ForeignKey("tasks.id")), db.Column("tag_id", db.ForeignKey("tags.id"))

bode/bode/models/task/model.py

@@ -3,7 +3,7 @@
 from sqlalchemy import Enum
 from sqlalchemy.dialects.postgresql import UUID
 
-from bode.app import db
+from bode.extensions import db
 from bode.models.enums import TaskStatus
 from bode.models.tag.model import task_tag
 from bode.models.utc_datetime import UTCDateTime

bode/pyproject.toml

@@ -23,6 +23,8 @@ psycopg2 = "^2.9.3"
 flake8-print = "^4.0.0"
 python-dateutil = "^2.8.2"
 gunicorn = "^20.1.0"
+schedule = "^1.1.0"
+sendgrid = "^6.9.7"
 
 [tool.poetry.dev-dependencies]
 pytest = "^5.2"

docker-compose.yaml

@@ -8,6 +8,7 @@ services:
       - "4000:4000"
     env_file:
       - .env.example
+      - .env
     volumes:
       - ./bode:/bode/