From 198ba20a47a6c78cd3a2ac334e7fdc9d028570d1 Mon Sep 17 00:00:00 2001
From: Andros Fenollosa <andros@fenollosa.email>
Date: Thu, 1 Feb 2018 20:32:28 +0100
Subject: [PATCH] Update database

---
 app.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/app.py b/app.py
index ff939b3..0ae1777 100644
--- a/app.py
+++ b/app.py
@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 # Librarys
 import os
-from flask import Flask
+from flask import Flask, jsonify
 from flask_restplus import Resource, Api
 from dotenv import load_dotenv, find_dotenv
 from models import db, User, News, Comment
@@ -14,6 +14,11 @@ app = Flask(__name__)
 # Config Flask
 app.config['SECRET_KEY'] = os.environ.get('SECRET_KEY')
 
+# Config Database
+app.config['SQLALCHEMY_DATABASE_URI'] = os.environ.get('DATABASE_URI')
+app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
+db.init_app(app)
+
 # Config API
 api = Api(app)
 PRE_URL = '/api/v1/'
@@ -41,11 +46,11 @@ class Logout(Resource):
 
 
 @api.route(PRE_URL + 'news')
-class News_all(Resource):
+class NewsList(Resource):
 
     def get(self):
         my_news = News.query.all()
-        return ([i.serialize for i in my_news])
+        return jsonify([i.serialize for i in my_news])
 
     def post(self):
         return {'hello': 'world'}