MyCity/server_side/voting-system/db.py

import sqlite3

import click
from flask import current_app, g
from flask.cli import with_appcontext

def get_db():
	if 'db' not in g:
		g.db = sqlite3.connect(current_app.config['DATABASE'])
		g.db.row_factory = sqlite3.Row
		
	return g.db
	
def close_db()
	db = g.pop('db', None)

	if db is not None:
		db.close()