setup login route

README.md

@@ -44,5 +44,7 @@ A website to help you keep track of all the video media that you watched, are wa
 - [X] design the database
 - [X] connect data base
 - [X] error page
-- [ ] require login wrapper
+- [X] require login wrapper
+- [ ] login page
+- [ ] logout page
 - [ ] register page design

app.py

@@ -1,4 +1,9 @@
+# ----------------------------------
+# ------: import libs and functions
+# ----------------------------------
+
 # import libs
+from crypt import methods
 import os
 from flask import Flask, render_template, session, request, redirect, jsonify
 from flask_session import Session
@@ -7,6 +12,10 @@
 from werkzeug.security import check_password_hash, generate_password_hash
 from functools import wraps
 
+# ----------------------------------
+# ------: configurations
+# ----------------------------------
+
 # init main app
 app = Flask(__name__)
 
@@ -30,6 +39,10 @@ def after_request(response):
 # connect media database
 db = SQL("sqlite:///media.db")
 
+# ----------------------------------
+# ------: Helper functions
+# ----------------------------------
+
 # error page
 def apology(message, code=400):
     return render_template("error.html", code=code, msg=message)
@@ -45,6 +58,10 @@ def decorated_function(*args, **kwargs):
         return f(*args, **kwargs)
     return decorated_function
 
+# ----------------------------------
+# ------: Routes
+# ----------------------------------
+
 # index route
 @app.route("/")
 @login_required
@@ -57,9 +74,20 @@ def register():
     return render_template("TODO.html")
 
 # login route
-@app.route("/login")
+@app.route("/login", methods=["GET", "POST"])
 def login():
-    return render_template("TODO.html")
+    """Log user in"""
+
+    # Forget any user_id
+    session.clear()
+
+    # User reached route via GET (as by clicking a link or via redirect)
+    if request.method == "GET": 
+        return render_template("login.html")
+
+    # User reached route via POST (as by submitting a form via POST)
+    else:
+        return render_template("TODO.html")
 
 # logout route
 @app.route("/logout")

templates/error.html

@@ -1,5 +1,5 @@
 {% extends "layout.html" %}
-{% block title %} TODO {% endblock %}
+{% block title %} ERROR {% endblock %}
 {% block body %}
 <div class="error">
     <div class="error__code">

templates/login.html

@@ -0,0 +1,7 @@
+{% extends "layout.html" %}
+{% block title %} Login {% endblock %}
+{% block body %}
+<div class="login">
+    <h1 class="login__heading">LOGIN</h1>
+</div>
+{% endblock %}