84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
|
import sqlite3
|
||
|
|
|
||
|
|
garden_db_path = "sqlite/garden_db.sqlite"
|
||
|
|
|
||
|
|
def init_database():
|
||
|
|
#TODO: does this need permissions?
|
||
|
|
conn = sqlite3.connect(garden_db_path)
|
||
|
|
init_table_string = """CREATE TABLE IF NOT EXISTS garden (
|
||
|
|
plant_id tinytext PRIMARY KEY,
|
||
|
|
owner text,
|
||
|
|
description text,
|
||
|
|
age text,
|
||
|
|
score integer,
|
||
|
|
is_dead text
|
||
|
|
)"""
|
||
|
|
|
||
|
|
c = conn.cursor()
|
||
|
|
c.execute(init_table_string)
|
||
|
|
conn.close()
|
||
|
|
def update_garden_db():
|
||
|
|
# insert or update this plant id's entry in DB (should happen
|
||
|
|
# regularly)
|
||
|
|
# TODO: create a second function that is called to retrieve garden
|
||
|
|
# when called by display controller
|
||
|
|
|
||
|
|
|
||
|
|
conn = sqlite3.connect(garden_db_path)
|
||
|
|
c = conn.cursor()
|
||
|
|
##try to insert or replace
|
||
|
|
update_query = """INSERT OR REPLACE INTO garden (
|
||
|
|
plant_id, owner, description, age, score, is_dead
|
||
|
|
) VALUES (
|
||
|
|
'{pid}', '{pown}', '{pdes}', '{page}', {psco}, '{pdead}'
|
||
|
|
)
|
||
|
|
""".format(pid = "asdfaseeeedf", pown = "jaeeke", pdes = "bigger ceeooler plant", page="28dee", psco = str(244400), pdead = str(False))
|
||
|
|
# update_query = """INSERT INTO garden (
|
||
|
|
# plant_id, owner, description, age, score, is_dead
|
||
|
|
# ) VALUES (
|
||
|
|
# '{pid}', '{pown}', '{pdes}', '{page}', {psco}, '{pdead}'
|
||
|
|
# )
|
||
|
|
# """.format(pid = "asdfasdf", pown = "jake", pdes = "big cool plant", page="25d", psco = str(25), pdead = str(False))
|
||
|
|
|
||
|
|
print c.execute(update_query)
|
||
|
|
conn.commit()
|
||
|
|
conn.close()
|
||
|
|
#print "bigggg booom"
|
||
|
|
|
||
|
|
def retrieve_garden_from_db(garden_db_path):
|
||
|
|
# Builds a dict of dicts from garden sqlite db
|
||
|
|
garden_dict = {}
|
||
|
|
conn = sqlite3.connect(garden_db_path)
|
||
|
|
conn.row_factory = sqlite3.Row
|
||
|
|
c = conn.cursor()
|
||
|
|
c.execute('SELECT * FROM garden ORDER BY owner')
|
||
|
|
tuple_list = c.fetchall()
|
||
|
|
conn.close()
|
||
|
|
# Building dict from table rows
|
||
|
|
for item in tuple_list:
|
||
|
|
garden_dict[item[0]] = {
|
||
|
|
"owner":item[1],
|
||
|
|
"description":item[2],
|
||
|
|
"age":item[3],
|
||
|
|
"score":item[4],
|
||
|
|
"dead":item[5],
|
||
|
|
}
|
||
|
|
return garden_dict
|
||
|
|
|
||
|
|
#init_database()
|
||
|
|
#update_garden_db()
|
||
|
|
results = retrieve_garden_from_db(garden_db_path)
|
||
|
|
print results
|
||
|
|
|
||
|
|
|
||
|
|
# con = sqlite3.connect(garden_db_path) #
|
||
|
|
# con.row_factory = sqlite3.Row #
|
||
|
|
# cur = con.cursor() #
|
||
|
|
# cur.execute("select * from garden ORDER BY score desc") #
|
||
|
|
# blah = cur.fetchall() #
|
||
|
|
# con.close()
|
||
|
|
# print blah
|
||
|
|
|
||
|
|
|
||
|
|
|