def insert(table, fields=(), values=(), db=settings.get_db()):
cursor = db.cursor()
query = 'INSERT OR REPLACE INTO %s (%s) VALUES (%s)' % (
table,
', '.join(fields),
', '.join(['?'] * len(values))
)
cursor.execute(query, values)
db.commit()
row_id = cursor.lastrowid
cursor.close()
return row_id
def query(query, args=(), one=False, db=settings.get_db()):
cursor = db.execute(query, args)
results = cursor.fetchall()
cursor.close()
return (results[0] if results else None) if one else results
def app_get_db():
db = getattr(g, '_database', None)
if db is None:
db = g._database = settings.get_db()
return db
