Update database.py

database.py

@@ -3,7 +3,7 @@ from flask import g
 
 def init_db(app):
     with app.app_context():
-        db = get_db()
+        db = get_db(app)
         db.execute('''
             CREATE TABLE IF NOT EXISTS news (
                 id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -24,6 +24,11 @@ def get_db(app):
         g.db.row_factory = sqlite3.Row
     return g.db
 
+def close_db(e=None):
+    db = g.pop('db', None)
+    if db is not None:
+        db.close()
+
 def cache_news(app, articles):
     db = get_db(app)
     for article in articles: