def fetch_restaurants():
    connection = connection_factory()
    cursor = connection.cursor()

    sql = "SELECT * FROM restaurant WHERE times = (SELECT MIN(times) FROM restaurant)"

    if not has_money():
        sql += " AND Expensive = 0"

    cursor.execute(sql)

    return cursor.fetchall()
Example #2
0
# -*- coding:utf-8 -*-
import sqlite3 as lite
from settings import APPROVED_RESTAURANTS as my_restaurants
from connection import connection_factory


try:
    connection = connection_factory()
    cursor = connection.cursor()

    cursor.executescript("""
        DROP TABLE IF EXISTS restaurant;
        CREATE TABLE restaurant(id INTEGER PRIMARY KEY, name TEXT, expensive NUMERIC, times INTEGER DEFAULT 0);
    """)

    cursor.executemany("INSERT INTO restaurant (name, expensive) VALUES(?, ?)", my_restaurants)

    connection.commit()

except lite.Error, e:
    if connection:
        connection.rollback()

        print "Error %s:" % e.args[0]

finally:
    if connection:
        connection.close()