Persist series on database

Author: rafaelhenrique

.gitignore

@@ -88,3 +88,6 @@ ENV/
 
 # Rope project settings
 .ropeproject
+
+# Database
+*.sqlite3

tvseries/core.py

@@ -1,12 +1,12 @@
 import os
+
 from random import choice
 from flask import Flask, render_template, request, redirect, url_for
 from flask_sqlalchemy import SQLAlchemy
 
 app = Flask(__name__)
 app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///tvseries.sqlite3'
 db = SQLAlchemy(app)
-series = []
 
 
 @app.route('/')
@@ -15,14 +15,22 @@ def home(name=None):
     image_filenames = os.listdir(image_directory)
     image = os.path.join('img', choice(image_filenames))
     img_url = url_for('static', filename=image)
+    series = TVSerie.query.all()
     return render_template('home.html', series=series, image=img_url)
 
 
 @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)
+        name = request.form.to_dict().get('serie-name')
+        description = request.form.to_dict().get('serie-description')
+        author = request.form.to_dict().get('serie-author')
+        episodies_number = request.form.to_dict().get('serie-episodes_number')
+
+        serie = TVSerie(name=name, description=description, author=author,
+                        episodies_number=episodies_number)
+        db.session.add(serie)
+        db.session.commit()
         return redirect('/')
 
     return render_template('add.html')

tvseries/templates/add.html

@@ -20,7 +20,7 @@
                     </div>
                     <div class="form-group">
                         <label>Número de episódios:</label>
-                        <input type="text" name="serie-episodes_number" id="id_serie-episodes_number">
+                        <input type="text" name="serie-episodies_number" id="id_serie-episodies_number">
                     </div>                    
                 </div>
             </div>

tvseries/templates/home.html

@@ -10,7 +10,7 @@
     <div class="panel-heading">Minhas series:</div>
     <ul class="list-group">
     {% for serie in series %}
-        <li class="list-group-item">{{ serie }}</li>
+        <li class="list-group-item">{{ serie.name }}</li>
     {% endfor %}
     </ul>
 </div>

tvseries/tests/test_core.py

@@ -11,24 +11,36 @@ def app(self):
         from tvseries.core import app
         return app
 
-    def test_get_home(self):
+    def test_get_home(self, db):
         response = self.client.get("/")
         assert '<div class="banner">' in response.data.decode('utf-8')
         assert response.status_code == 200
 
-    def test_get_add(self):
+    def test_get_add_status_code(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
+    def test_get_add_content(self):
+        response = self.client.get("/add")
+        expected = (
+            'name="serie-name" id="id_serie-name"',
+            'name="serie-description" id="id_serie-description"',
+            'name="serie-author" id="id_serie-author"',
+            'name="serie-episodies_number" id="id_serie-episodies_number"',
+        )
+        for field in expected:
+            assert field in response.data.decode('utf-8')
+
+    def test_post_add(self, db):
+        response = self.client.post("/add", data={
+            "serie-name": "Game of Thrones",
+            "serie-author": "George R.R. Martin",
+            "serie-description": "Teste",
+        })
+        result = TVSerie.query.filter(TVSerie.name == 'Game of Thrones')
+        assert response.status_code == 302 and result.count() == 1
 
-    def test_navbar(self):
+    def test_navbar(self, db):
         response = self.client.get("/")
         assert ('<nav class="navbar navbar-default"' in
                 response.data.decode('utf-8'))