def edit_message(message_id=None):
if request.method == 'GET':
query_string = ('SELECT message_id, content, user_id '
'FROM messages '
'WHERE message_id = ?')
query_result = datamanager.query_db(query_string, [message_id],
one=True)
return render_template('edit-message.html', message=query_result)
elif request.method == 'POST':
message_content = request.form.get('message')
message_id = request.form.get('message_id')
query_string = ('UPDATE messages '
'SET content = ? '
'WHERE message_id = ?')
query_result = datamanager.query_db(query_string,
[message_content, message_id],
one=True)
print('result:', query_result)
return redirect('/')
def new_message():
if request.method == 'GET':
return render_template('new-message.html')
elif request.method == 'POST':
content = request.form.get('message')
current_time = datetime.now()
user_id = usermanager.flask_login.current_user.user_id
print(current_time)
query_string = (
'INSERT INTO messages( content, time_created, user_id ) '
'VALUES (?,?,?)'
)
query_result = datamanager.query_db(
query_string,
[content, current_time, user_id],
one=True
)
if query_result == None:
print('error')
else:
print('success')
return redirect('/')
def new_message():
if request.method == 'GET':
return render_template('new-message.html')
elif request.method == 'POST':
content = request.form.get('message')
current_time = datetime.now()
user_id = usermanager.flask_login.current_user.user_id
print(current_time)
query_string = (
'INSERT INTO messages( content, time_created, user_id ) '
'VALUES (?,?,?)')
query_result = datamanager.query_db(query_string,
[content, current_time, user_id],
one=True)
if query_result == None:
print('error')
else:
print('success')
return redirect('/')
def index():
query_string = ('SELECT message_id, content, username, time_created '
'FROM messages INNER JOIN users '
'USING (user_id) '
'ORDER BY time_created DESC')
query_results = datamanager.query_db(query_string, [], one=False)
for result in query_results:
print(result)
return render_template('index.html', messages=query_results)
def edit_message(message_id = None):
if request.method == 'GET':
query_string = (
'SELECT message_id, content, user_id '
'FROM messages '
'WHERE message_id = ?'
)
query_result = datamanager.query_db(
query_string,
[message_id],
one=True
)
return render_template('edit-message.html', message=query_result)
elif request.method == 'POST':
message_content = request.form.get('message')
message_id = request.form.get('message_id')
query_string = (
'UPDATE messages '
'SET content = ? '
'WHERE message_id = ?'
)
query_result = datamanager.query_db(
query_string,
[message_content, message_id],
one=True
)
print('result:', query_result)
return redirect('/')
def checkout(model_no=None):
if request.method == 'GET':
query_string = ('SELECT * from valves ' 'WHERE model_no = ?')
query_results = datamanager.query_db(query_string, [model_no],
one=True)
return render_template('checkout.html', item=query_results)
if request.method == 'POST':
name = request.form.get('name')
address = "{}, {} {}, {}".format(request.form.get('street'),
request.form.get('city'),
request.form.get('postcode'),
request.form.get('country'))
purchase_qty = request.form.get('purchase_qty')
item_id = request.form.get('item_id')
card_number = request.form.get('card_number')
card_name = request.form.get('card_name')
card_expiry = request.form.get('card_expiry')
card_security = request.form.get('card_security')
# if len(card_number) not 16:
# return redirect('/checkout/', model_no = item_id)
query_string = (
'INSERT INTO orders (name, address, item_id, purchase_qty, '
'card_number, card_name, card_expiry, card_security) '
'VALUES (?, ?, ?, ?, ?, ?, ?, ?)')
query_results = datamanager.query_db(query_string, [
name, address, item_id, purchase_qty, card_number, card_name,
card_expiry, card_security
],
one=False)
return redirect('/thank-you')
def index():
query_string = (
'SELECT message_id, content, username, time_created '
'FROM messages INNER JOIN users '
'USING (user_id) '
'ORDER BY time_created DESC'
)
query_results = datamanager.query_db(query_string, [], one=False)
for result in query_results:
print(result)
return render_template('index.html', messages=query_results)
def item(model_no=None):
if request.method == 'GET':
query_string = ('SELECT * from valves ' 'WHERE model_no = ?')
query_results = datamanager.query_db(query_string, [model_no],
one=True)
return render_template('item.html', item=query_results)
if request.method == 'POST':
model_no = request.form.get('model_no')
product_qty = request.form.get('product_qty')
return redirect('/checkout/{}'.format(model_no))
def load_user(username):
query_string = (
'SELECT user_id, username, password, first_name, last_name '
'FROM users '
'WHERE username = ?'
)
query_result = datamanager.query_db(query_string, [username], one=True)
if query_result == None:
user = User(
None, None
)
else:
user = User(
query_result['username'],
query_result['password']
)
user.user_id = query_result['user_id']
user.first_name = query_result['first_name']
user.last_name = query_result['last_name']
def index():
query_string = ('SELECT * FROM valves')
query_results = datamanager.query_db(query_string, [], one=False)
return render_template('index.html', items=query_results)