def get_varieties():
with sqlite3.connect(Connection.db_path) as conn:
conn.row_factory = model_factory(Variety)
db_cursor = conn.cursor()
db_cursor.execute("""
select
v.id,
v.name,
v.country_of_origin
from icecreamapp_variety v
""")
return db_cursor.fetchall()
def get_variety_country(variety_id):
with sqlite3.connect(Connection.db_path) as conn:
conn.row_factory = model_factory(Variety)
db_cursor = conn.cursor()
db_cursor.execute(
"""
SELECT
v.id,
v.country_of_origin
FROM icecreamapp_variety v
WHERE v.id = ?
""", (variety_id, ))
return db_cursor.fetchone()
def variety_list(request):
if request.method == 'GET':
with sqlite3.connect(Connection.db_path) as conn:
conn.row_factory = model_factory(Variety)
db_cursor = conn.cursor()
db_cursor.execute("""
SELECT
v.id,
v.name,
v.country_of_origin
FROM icecreamapp_variety v
""")
all_varieties = db_cursor.fetchall()
template = 'variety/variety_list.html'
context = {'varieties': all_varieties}
return render(request, template, context)
elif request.method == 'POST':
form_data = request.POST
with sqlite3.connect(Connection.db_path) as conn:
db_cursor = conn.cursor()
db_cursor.execute(
"""
INSERT INTO icecreamapp_variety
(
'name', country_of_origin
)
VALUES (?, ?)
""",
(form_data['variety_name'], form_data['country_of_origin']))
return redirect(reverse('icecreamapp:variety_list'))
def get_variety_flavor(variety_id):
with sqlite3.connect(Connection.db_path) as conn:
conn.row_factory = model_factory(Variety)
db_cursor = conn.cursor()
db_cursor.execute(
"""
SELECT
vf.id,
vf.flavor_id,
vf.variety_id,
vf.toppings,
f.id,
f.name,
f.alcohol_percent,
v.id,
v.name,
v.country_of_origin
FROM icecreamapp_varietyflavor vf
JOIN icecreamapp_variety v ON v.id = vf.variety_id
JOIN icecreamapp_flavor f ON f.id = vf.flavor_id
WHERE v.id = ?
""", (variety_id, ))
return db_cursor.fetchall()
def flavor_list(request):
if request.method == 'GET':
with sqlite3.connect(Connection.db_path) as conn:
conn.row_factory = model_factory(Flavor)
db_cursor = conn.cursor()
db_cursor.execute("""
SELECT
f.id,
f.name,
f.alcohol_percent
FROM icecreamapp_flavor f
""")
all_flavors = db_cursor.fetchall()
template = 'flavor/flavor_list.html'
context = {'flavors': all_flavors}
return render(request, template, context)
elif request.method == 'POST':
form_data = request.POST
with sqlite3.connect(Connection.db_path) as conn:
db_cursor = conn.cursor()
db_cursor.execute(
"""
INSERT INTO icecreamapp_flavor
(
name, alcohol_percent
)
VALUES (?, ?)
""", (form_data['name'], form_data['alcohol_percent']))
return redirect(reverse('icecreamapp:flavor_list'))