added connection status macro and background task scheduling

app/__init__.py

@@ -1,9 +1,14 @@
-from distutils.command.config import config
-from flask import Flask
+import atexit
+
+from apscheduler.schedulers.background import BackgroundScheduler
+import requests
+
+from flask import Flask, current_app, g
 from flask_login import LoginManager
 from flask_mail import Mail
 from flask_moment import Moment
 from flask_sqlalchemy import SQLAlchemy
+
 from config import configuration, DevelopmentConfig, TestingConfig, ProductionConfig
 
 
@@ -14,6 +19,14 @@ db = SQLAlchemy()
 login_manager = LoginManager()
 login_manager.login_view = 'auth.login'
 
+def scheduled_task(func, seconds=60, args=None):
+
+    scheduler = BackgroundScheduler()
+    scheduler.add_job(func=func, trigger='interval', seconds=seconds, args=args)
+    scheduler.start()
+
+    atexit.register(lambda: scheduler.shutdown())
+
 
 def create_app(config_name):
     app = Flask(__name__)

app/auth/views.py

@@ -1,13 +1,26 @@
 from flask import render_template, redirect, request, url_for, flash
-from flask_login import login_user, logout_user, login_required
+from flask_login import login_user, logout_user, login_required, current_user
 
 from app import db
 from app.auth import auth
+from app.email import send_email
 from app.models import User
 from app.auth.forms import LoginForm, RegistrationForm
 
 
 
+@auth.before_app_request
+def before_request():
+    if current_user.is_authenticated and not current_user.confirmed and request.blueprint != 'auth' and request.endpoint != 'static':
+        return redirect(url_for('auth.unconfirmed'))
+
+
+@auth.route('/unconfirmed')
+def unconfirmed():
+    if current_user.is_anonymous or current_user.confirmed:
+        return redirect(url_for('main.index'))
+    return render_template('auth/unconfirmed.html')
+
 @auth.route('/login', methods=['get', 'post'])
 def login():
     form = LoginForm()
@@ -43,8 +56,46 @@ def register():
     if form.validate_on_submit():
         user = User(email=form.email.data, username=form.username.data, password=form.password.data)
         db.session.add(user)
+
         db.session.commit()
-        flash('You can now login', category='success')
+
+        token = user.generate_confirmation_token()
+
+        send_email(user.email, 'Confirm Your Yo!nk Account', 'auth/email/confirm', user=user, token=token)
+
+        flash(f'A confirmation email was sent to {user.email}', category='success')
         return redirect(url_for('auth.login'))
     
-    return render_template('auth/register.html', form=form)
+    return render_template('auth/register.html', form=form)
+
+
+@auth.route('/confirm/<token>')
+@login_required
+def confirm(token):
+    if current_user.confirmed:
+        return redirect(url_for('main.index'))
+
+    if current_user.confirm(token):
+        db.session.commit()
+        flash('Your account has been confirmed. Thank you!')
+    else:
+        flash('Invalid or expired confirmation link')
+
+    return redirect(url_for('main.index'))
+
+
+@auth.route('/confirm')
+@login_required
+def resend_confirm():
+    token = current_user.generate_confirmation_token()
+
+    send_email(current_user.email, 'Confirm Your Yo!nk Account', 'auth/email/confirm', user=current_user, token=token)
+
+    flash(f'A new confirmation email was sent to {current_user.email}', category='success')
+
+    return redirect(url_for('main.index'))
+
+
+# TODO: password updates
+# TODO: password reset
+# TODO: email updates

app/email.py

@@ -3,7 +3,7 @@ import threading
 from flask import current_app, render_template
 from flask_mail import Message
 
-from app import mail, config
+from app import mail
 
 
 def send_async_email(app, msg):

app/main/views.py

@@ -1,7 +1,7 @@
 import os
 from datetime import datetime
 
-from flask import render_template, session, send_from_directory, redirect, url_for, flash
+from flask import render_template, session, send_from_directory, redirect, url_for, flash, g
 
 from app.main import main
 from app.main.forms import DownloadForm
@@ -89,7 +89,7 @@ def index():
             latest = get_comic_library_meta()
             form.url.data = ''
 
-            return render_template('index.html', form=form, url=url, series=series, latest=latest), 200
+            return render_template('index.html', form=form, url=url, series=series, latest=latest, status=g.status_code), 200
 
 
         if form.series.data:

app/templates/auth/email/confirm.html

@@ -0,0 +1,12 @@
+Welcome to Yo!nk {{ user.username }},
+
+To confirm your account please click on the following link:
+
+{{ url_for('auth.confirm', token=token, _external=True) }}
+
+Sincerely,
+
+The Yo!nk Team
+
+
+Note: replies to this email address are unmonitored.

app/templates/auth/register.html

@@ -0,0 +1,14 @@
+{% extends "base.html" %}
+
+{% block meta %}
+{{ super() }}
+<!-- Add additional meta tags here -->
+{% endblock %}
+
+{% block title %}Testing{% endblock %}
+
+{% block page_content %}
+<div class="page-header">
+    <h1>Register</h1>
+</div>
+{% endblock %}

app/templates/base.html

@@ -1,3 +1,5 @@
+{% import 'macros/connection_status.html' as connection %}
+
 <!DOCTYPE html>
 <html lang="en">
 <head>
@@ -15,7 +17,7 @@
     <script src="https://cdn.jsdelivr.net/npm/bootstrap@5.1.3/dist/js/bootstrap.min.js" integrity="sha384-QJHtvGhmr9XOIpI6YVutG+2QOK9T+ZnN4kzFN1RtK3zEFEIsxhlmWl5/YESvpZ13" crossorigin="anonymous"></script>
     {{ moment.include_moment() }}
     {% endblock %}
-    <link rel="shortcut icon" href="{{ url_for('static', filename='images/comic.png') }}" type="image/x-icon">
+    <link rel="shortcut icon" href="{{ connection.toggle_connection_status(status_code) }}" type="image/x-icon">
 </head>
 <body>
     {% block navbar %}

app/templates/index.html

@@ -1,5 +1,7 @@
 {% extends "base.html" %}
 {% import 'macros/library.html' as library %}
+{% import 'macros/connection_status.html' as connection %}
+
 
 {% block meta %}
 {{ super() }}
@@ -12,7 +14,6 @@
 <div class="page-header">
     <h1>Y!oink Web App</h1>
 </div>
-
 <form method="post" onsubmit="download.disabled = true; url.readOnly = true; return true;">
     {{ form.hidden_tag() }}
     <div class="input-group mb-3">

app/templates/macros/connection_status.html

@@ -0,0 +1,7 @@
+{% macro toggle_connection_status(status_code) %}
+{% if status_code == 200 %}
+{{ url_for('static', filename='up.png') }}
+{% else %}
+{{ url_for('static', filename='down.png') }}
+{% endif %}
+{% endmacro %}