#!/usr/bin/python
# -*- coding: utf-8 -*-

import sqlite3
from flask import g
import ConfigParser

config = ConfigParser.SafeConfigParser()
config.read('config.ini')

def connect_db():
    """Funkcja łącząca się z bazą danych."""
    g.db = sqlite3.connect(config.get('Basic','Database'))

def disconnect_db():
    """Funkcja zamykająca połączenie z bazą danych"""
    g.db.close()

def query_db(query, args=(), one=False):
    """Funkcja ułatwiająca odpytywanie bazy danych, źródło:
       http://flask.pocoo.org/docs/patterns/sqlite3/
    """
    cur = g.db.execute(query, args)
    rv = [dict((cur.description[idx][0], value)
               for idx, value in enumerate(row)) for row in cur.fetchall()]
    return (rv[0] if rv else None) if one else rv