Chapter 5

app/templates/base.html

@@ -10,7 +10,11 @@
     <div>
       Microblog:
       <a href="{{ url_for('index') }}">Home</a>
-      <a href="{{ url_for('login') }}">Login</a>
+      {% if current_user.is_anonymous %}
+        <a href="{{ url_for('login') }}">Login</a>
+      {% else %}
+        <a href="{{ url_for('logout') }}">Logout</a>
+      {% endif %}
     </div>
     <hr />
     {% with messages = get_flashed_messages() %}

app/templates/index.html

@@ -1,8 +1,12 @@
-{% extends "base.html" %}
+{% extends 'base.html' %}
 
 {% block content %}
-    <h1>Hi, {{ user.username }}!</h1>
-    {% for post in posts %}
-    <div><p>{{ post.author.username }} says: <b>{{ post.body }}</b></p></div>
-    {% endfor %}
-{% endblock %}
+  <h1>Hi, {{ current_user.username }}!</h1>
+  {% for post in posts %}
+    <div>
+      <p>
+        {{ post.author.username }} says: <b>{{ post.body }}</b>
+      </p>
+    </div>
+  {% endfor %}
+{% endblock %}

app/templates/login.html

@@ -21,4 +21,7 @@
     <p>{{ form.remember_me() }} {{ form.remember_me.label }}</p>
     <p>{{ form.submit() }}</p>
   </form>
+  <p>
+    New User? <a href="{{ url_for('register') }}">Click to Register!</a>
+  </p>
 {% endblock %}

app/templates/register.html

@@ -0,0 +1,37 @@
+{% extends 'base.html' %}
+
+{% block content %}
+  <h1>Register</h1>
+  <form action="" method="post">
+    {{ form.hidden_tag() }}
+    <p>
+      {{ form.username.label }}<br />
+      {{ form.username(size = 32) }}<br />
+      {% for error in form.username.errors %}
+        <span style="color: red;">[{{ error }}]</span>
+      {% endfor %}
+    </p>
+    <p>
+      {{ form.email.label }}<br />
+      {{ form.email(size = 64) }}<br />
+      {% for error in form.email.errors %}
+        <span style="color: red;">[{{ error }}]</span>
+      {% endfor %}
+    </p>
+    <p>
+      {{ form.password.label }}<br />
+      {{ form.password(size = 32) }}<br />
+      {% for error in form.password.errors %}
+        <span style="color: red;">[{{ error }}]</span>
+      {% endfor %}
+    </p>
+    <p>
+      {{ form.password2.label }}<br />
+      {{ form.password2(size = 32) }}<br />
+      {% for error in form.password2.errors %}
+        <span style="color: red;">[{{ error }}]</span>
+      {% endfor %}
+    </p>
+    <p>{{ form.submit() }}</p>
+  </form>
+{% endblock %}