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()
# -*- 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()