actaully make the emails send at 7 am every day

Author: SharpBit

.env.example

@@ -7,4 +7,7 @@ DB_USERNAME = # Postgresql Username
 DB_PASSWORD = # Postgresql Password
 DB_HOST = # Database host IP
 DB_NAME = # Database name
-BRAWLSTATS_TOKEN = # API Key of the Brawl Stars API
+BRAWLSTATS_TOKEN = # API Key of the Brawl Stars API
+NOREPLY_EMAIL = # Actual gmail email
+CUSTOM_EMAIL = # Custom domain email to use
+NOREPLY_APP_PASSWORD = # 2fa app password

core/__init__.py

@@ -5,6 +5,7 @@
 
 from core.routes import root
 from core.listeners import listeners
+from core.utils import handle_daily_emails
 
 
 def create_app(config_class=Config):
@@ -20,5 +21,6 @@ def create_app(config_class=Config):
     app.blueprint(listeners)
 
     Session(app)  # sanic_session
+    app.add_task(handle_daily_emails)
 
     return app

core/config.py

@@ -18,4 +18,5 @@ class Config:
     DB_HOST = os.getenv('DB_HOST')
     BRAWLSTATS_TOKEN = os.getenv('BRAWLSTATS_TOKEN')
     NOREPLY_EMAIL = os.getenv('NOREPLY_EMAIL')
-    EMAIL_APP_PASSWORD = os.getenv('NOREPLY_APP_PASSWORD')
+    CUSTOM_EMAIL = os.getenv('CUSTOM_EMAIL')
+    EMAIL_APP_PASSWORD = os.getenv('NOREPLY_APP_PASSWORD')

core/routes.py

@@ -10,8 +10,8 @@
 from sanic import Blueprint, response
 from sanic.exceptions import abort
 
-from core.utils import add_message, disable_xss, login_required, open_db_connection, render_template
-from core.utils import daterange, thisweek
+from core.utils import (add_message, disable_xss, get_school_week,
+                        login_required, open_db_connection, render_template)
 
 root = Blueprint('root')
 
@@ -273,77 +273,7 @@ async def schoolweek(request, requested_date_str):
     if not first_day <= requested_date <= date(2021, 6, 11):
         abort(404, message=f'Requested URL {request.path} not found')
 
-    no_school = [
-        date(2020, 9, 28),  # Yom Kippur
-        date(2020, 10, 12),  # Columbus day
-        date(2020, 11, 3),  # Election day
-        date(2020, 11, 11),  # Veteran's day
-        *daterange(date(2020, 11, 25), date(2020, 11, 27)),  # Thanksgiving break
-        *daterange(date(2020, 12, 24), date(2021, 1, 1))  # Holiday break
-    ]
-    special_days = [
-        date(2020, 10, 2)
-    ]
-
-    all_days = []
-    day_map = {
-        1: 'A',
-        0: 'B'
-    }
-
-    next_friday = thisweek(requested_date)[-1]
-    elapsed_dates = daterange(first_day, next_friday)
-    mondays = [d for d in elapsed_dates if d.weekday() == 0 and d not in no_school]
-    cohort_day = 'maroon'
-
-    for d in elapsed_dates:
-        if d in no_school:
-            continue
-        dow = d.weekday()
-        if dow in (5, 6):
-            continue
-        if dow in (1, 3):
-            cohort_day = 'maroon'
-        elif dow in (2, 4):
-            cohort_day = 'gray'
-        elif dow == 0:
-            # Mondays
-            if d not in mondays:
-                continue
-            if mondays.index(d) % 2 == 0:
-                cohort_day = 'maroon'
-            else:
-                cohort_day = 'gray'
-
-        try:
-            prev_day = [day for day in all_days if day['cohort'] == cohort_day][-1]['day']
-        except IndexError:
-            # We are adding 1 for the first day of each cohort so we "start" with a B (0) day
-            prev_day = 0
-
-        if d in special_days:
-            # Skip a day
-            all_days.append({
-                'cohort': cohort_day,
-                'date': d,
-                'day': prev_day + 2})
-        else:
-            all_days.append({
-                'cohort': cohort_day,
-                'date': d,
-                'day': prev_day + 1})
-
-
-    week = thisweek(requested_date)
-
-    week_fmt = []
-    for day in week:
-        try:
-            day_info = list(filter(lambda d: d['date'] == day, all_days))[0]
-        except IndexError:
-            week_fmt.append(f"{day.strftime('%a %m/%d')}<br>NO SCHOOL")
-        else:
-            week_fmt.append(f"{day.strftime('%a %m/%d')}<br>{day_info['cohort'].title()} {day_map[day_info['day'] % 2]} day")
+    week_fmt = await get_school_week(requested_date, first_day, week=True)
 
     return await render_template(
         template='schoolweek',
@@ -363,15 +293,18 @@ async def email_subscribe(request):
     except KeyError:
         return add_message(request, 'error', 'Enter an email in the field.', '/schoolweek')
 
+    async with open_db_connection(request.app) as conn:
+        existing = await conn.fetchrow('SELECT * FROM mailing_list WHERE email = $1', email)
+        if existing:
+            return add_message(request, 'error', 'Email already subscribed.', '/schoolweek')
+
     msg = EmailMessage()
     msg['Subject'] = 'Thank you for subscribing to GCHS Daily Updates!'
-    # msg['Subject'] = f"GCHS Daily Email Notification ({date.today().strftime('%m/%d/%Y')})"
-    msg['From'] = 'gchs-noreply@sharpbit.dev'
+    msg['From'] = request.app.config.CUSTOM_EMAIL
     msg['To'] = email
     body = MIMEText(
         f"If this wasn't you, click <a href=\"http{'s' if not request.app.config.DEV else ''}://{request.app.config.DOMAIN}"
         f"/schoolweek/unsubscribe/{email}\">here</a> to unsubscribe.", 'html')
-    # body = f"Today, {date.today().strftime('%m/%d/%Y')}, is a maroon A day"
     msg.set_content(body)
 
     with smtplib.SMTP_SSL('smtp.gmail.com', 465) as smtp:
@@ -380,16 +313,12 @@ async def email_subscribe(request):
         smtp.send_message(msg)
 
     async with open_db_connection(request.app) as conn:
-        existing = await conn.fetchrow('SELECT * FROM mailing_list WHERE email = $1', email)
-        if existing:
-            return add_message(request, 'error', 'Email already subscribed.', '/schoolweek')
         await conn.execute('INSERT INTO mailing_list(email) VALUES ($1)', email)
 
-    # TODO: actually send the scheduled emails
     return add_message(request, 'success', 'Your email has been added to the mailing list.', '/schoolweek')
 
 @root.get('/schoolweek/unsubscribe/<email>')
 async def email_unsubscribe(request, email):
     async with open_db_connection(request.app) as conn:
         await conn.execute('DELETE FROM mailing_list WHERE email = $1', email)
-    return add_message(request, 'success', 'Your email has been removed from mailing list.', '/schoolweek')
+    return add_message(request, 'success', 'Your email has been removed from mailing list.', '/schoolweek')

core/utils.py

@@ -1,12 +1,14 @@
-from datetime import date, timedelta
-
-from sanic import response
-
-from jinja2 import Environment, PackageLoader
-from functools import wraps
+import asyncio
+import smtplib
 from contextlib import asynccontextmanager
+from datetime import date, datetime, time as dt_time, timedelta
+from email.message import EmailMessage
+from email.mime.text import MIMEText
+from functools import wraps
 
 import asyncpg
+from jinja2 import Environment, PackageLoader
+from sanic import response
 
 
 class Oauth2:
@@ -116,13 +118,14 @@ async def decorated_function(request, *args, **kwargs):
 
 
 def daterange(start_date: date, end_date: date) -> list:
-    '''Creates a list of dates from the start date to end date, inclusive'''
+    """Creates a list of dates from the start date to end date, inclusive"""
     day_count = (end_date - start_date).days + 1
     return [start_date + timedelta(days=i) for i in range(day_count)]
 
 
 
 def thisweek(today: date) -> list:
+    """Gets a list of dates for this week"""
     if 0 <= today.weekday() <= 4:
         # Monday to Friday
         monday = today - timedelta(days=today.weekday())  # Last Monday
@@ -131,3 +134,141 @@ def thisweek(today: date) -> list:
     friday = monday + timedelta(days=4)
 
     return daterange(monday, friday)
+
+
+async def get_school_week(requested_date: date, first_day: date, week=True):
+    no_school = [
+        date(2020, 9, 28),  # Yom Kippur
+        date(2020, 10, 12),  # Columbus day
+        date(2020, 11, 3),  # Election day
+        date(2020, 11, 11),  # Veteran's day
+        *daterange(date(2020, 11, 25), date(2020, 11, 27)),  # Thanksgiving break
+        *daterange(date(2020, 12, 24), date(2021, 1, 1))  # Holiday break
+    ]
+    special_days = [
+        date(2020, 10, 2)
+    ]
+
+    all_days = []
+    day_map = {
+        1: 'A',
+        0: 'B'
+    }
+
+    next_friday = thisweek(requested_date)[-1]
+    elapsed_dates = daterange(first_day, next_friday)
+    mondays = [d for d in elapsed_dates if d.weekday() == 0 and d not in no_school]
+    cohort_day = 'maroon'
+
+    for d in elapsed_dates:
+        if d in no_school:
+            continue
+        dow = d.weekday()
+        if dow in (5, 6):
+            continue
+        if dow in (1, 3):
+            cohort_day = 'maroon'
+        elif dow in (2, 4):
+            cohort_day = 'gray'
+        elif dow == 0:
+            # Mondays
+            if d not in mondays:
+                continue
+            if mondays.index(d) % 2 == 0:
+                cohort_day = 'maroon'
+            else:
+                cohort_day = 'gray'
+
+        try:
+            prev_day = [day for day in all_days if day['cohort'] == cohort_day][-1]['day']
+        except IndexError:
+            # We are adding 1 for the first day of each cohort so we "start" with a B (0) day
+            prev_day = 0
+
+        if d in special_days:
+            # Skip a day
+            all_days.append({
+                'cohort': cohort_day,
+                'date': d,
+                'day': prev_day + 2})
+        else:
+            all_days.append({
+                'cohort': cohort_day,
+                'date': d,
+                'day': prev_day + 1})
+
+
+    this_week = thisweek(requested_date)
+
+    if week:
+        week_fmt = []
+        for day in this_week:
+            try:
+                day_info = list(filter(lambda d: d['date'] == day, all_days))[0]
+            except IndexError:
+                week_fmt.append(f"{day.strftime('%a %m/%d')}<br>NO SCHOOL")
+            else:
+                week_fmt.append(f"{day.strftime('%a %m/%d')}<br>{day_info['cohort'].title()} {day_map[day_info['day'] % 2]} day")
+        return week_fmt
+
+    try:
+        day_info = list(filter(lambda d: d['date'] == requested_date, all_days))[0]
+    except IndexError:
+        # No school
+        return None
+    day_info['day'] = day_map[day_info['day'] % 2]
+    return day_info
+
+
+async def handle_daily_emails(app):
+    """Send out an email at a specified time every weekday"""
+    today = date.today()
+    # Saturday/Sunday
+    if today.weekday() in (5, 6):
+        today = today - timedelta(days=today.weekday() - 7)  # Not actually today, next monday
+    next_email = datetime.combine(today, dt_time(7, 0, 0))
+
+    # Past the time today, send "tomorrow"
+    if next_email < datetime.now():
+        next_email = datetime.combine(today + timedelta(days=1), dt_time(7, 0, 0))
+
+    delta = (next_email - datetime.now()).seconds
+    await asyncio.sleep(delta)
+
+    if today.weekday() in (5, 6):
+        # Could be here if the func was called after the time on Friday
+        return app.add_task(handle_daily_emails)
+
+
+    today_info = await get_school_week(date(2020, 11, 16), date(2020, 9, 8), week=False)
+    if today_info is None:
+        # No school
+        await asyncio.sleep(1)
+        return app.add_task(handle_daily_emails)
+
+    async with open_db_connection(app) as conn:
+        emails = await conn.fetch('SELECT * FROM mailing_list')
+
+    messages = []
+    for email in emails:
+        msg = EmailMessage()
+        msg['Subject'] = f"GCHS Daily Email Notification  for {date.today().strftime('%m/%d/%Y')}"
+        msg['From'] = app.config.CUSTOM_EMAIL
+        msg['To'] = email
+        body = MIMEText(
+            f"Today, {date.today().strftime('%m/%d/%Y')}, is a {today_info['cohort'].title()} {today_info['day']} day. <br><br>"
+            f"Click <a href=\"http{'s' if not app.config.DEV else ''}://{app.config.DOMAIN}"
+            f"/schoolweek/unsubscribe/{msg['To']}\">here</a> to unsubscribe.", 'html')
+
+        msg.set_content(body)
+        messages.append(msg)
+
+    with smtplib.SMTP_SSL('smtp.gmail.com', 465) as smtp:
+        smtp.login(app.config.NOREPLY_EMAIL, app.config.EMAIL_APP_PASSWORD)
+
+        for msg in messages:
+            smtp.send_message(msg)
+
+    # Prevent it from sending twice
+    await asyncio.sleep(1)
+    app.add_task(handle_daily_emails)