Add series and list series (on home page)

Author: rafaelhenrique

tvseries/core.py

@@ -1,9 +1,20 @@
-from flask import Flask, render_template
+from flask import Flask, render_template, request, redirect
 
 app = Flask(__name__)
+series = []
 
 
 @app.route('/')
 @app.route('/<name>')
 def home(name=None):
-    return render_template('home.html', name=name)
+    return render_template('home.html', name=name, series=series)
+
+
+@app.route('/add', methods=['GET', 'POST'])
+def add():
+    if request.method == 'POST':
+        serie_name = request.form.to_dict().get('serie-name')
+        series.append(serie_name)
+        return redirect('/')
+
+    return render_template('add.html')

tvseries/templates/add.html

@@ -0,0 +1,19 @@
+{% extends 'base.html' %}
+{% block content %}
+<form action="" method="post" id="form">       
+    <div class="panel panel-default form">
+        <div class="panel-heading">Adicionar série de TV</div>
+        <div class="panel-body">
+            <div class="row">
+                <div class="col-md-6">
+                    <div class="form-group">
+                        <label>Nome:</label>
+                        <input type="text" name="serie-name" id="id_serie-name">
+                    </div>
+                </div>
+            </div>
+        </div>
+    </div>
+    <button type="submit" class="btn btn-primary">Salvar</button>
+</form>
+{% endblock %}

tvseries/templates/home.html

@@ -4,4 +4,19 @@
 <h1>Hello {% if name %}{{name}}{% else %}world{% endif %}!</h1>
 <p>This is a simple page :D</p>
 </div>
+
+<div class="panel panel-default">
+    <div class="panel-heading"><a href="{{ url_for('add') }}">Inserir séries</a></div>
+</div>
+
+{% if series %}
+<div class="panel panel-default">
+    <div class="panel-heading">Minhas series:</div>
+    <ul class="list-group">
+    {% for serie in series %}
+        <li class="list-group-item">{{ serie }}</li>
+    {% endfor %}
+    </ul>
+</div>
+{% endif %}
 {% endblock %}

tvseries/tests.py

@@ -18,3 +18,15 @@ def test_get_name(self):
         response = self.client.get("/Rafael")
         assert '<h1>Hello Rafael!</h1>' in response.data.decode('utf-8')
         assert response.status_code == 200
+
+    def test_get_add(self):
+        response = self.client.get("/add")
+        assert ('<input type="text" name="serie-name" id="id_serie-name">' in
+                response.data.decode('utf-8'))
+        assert response.status_code == 200
+
+    def test_post_add(self):
+        response = self.client.post("/add", data={"serie-name": "Teste"})
+        from tvseries.core import series
+        assert series == ['Teste']
+        assert response.status_code == 302