-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.py
More file actions
67 lines (54 loc) · 1.96 KB
/
main.py
File metadata and controls
67 lines (54 loc) · 1.96 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
from flask import Flask, request, render_template
import pyodbc
conn = pyodbc.connect('DRIVER={FreeTDS};SERVER=localhost;PORT=1433;UID=sa;PWD=Virtual1234;DATABASE'
'=Restaurant;UseNTLMv2=yes;TDS_Version=8.0;Trusted_Domain=domain.local;')
app = Flask(__name__)
user_data = {
'Name': ''
}
@app.route('/')
def home():
return render_template('index.html', user_data=user_data)
@app.route('/register', methods=['POST'])
def register():
cursor = conn.cursor()
username = request.form['Username']
cursor.execute("select dbo.CheckUsernameExists(?)", username)
res = cursor.fetchone()
if res[0]:
return "Username already exists"
email = request.form['Email']
cursor.execute("select dbo.CheckEmailExists(?)", email)
res = cursor.fetchone()
if res[0]:
return "Email already exists"
cursor.execute("exec dbo.Register ?, ?, ?, ?", (username, email, request.form['Password'], request.form['Address']))
res = cursor.fetchone()
if res[1] == username:
global user_data
user_data['Name'] = res[1]
return "OK"
else:
return "Registration failed"
@app.route('/login', methods=['GET', 'POST'])
def login():
if request.method == 'POST':
username = request.form['username']
cursor = conn.cursor()
cursor.execute("select dbo.Login(?, ?)", (username, request.form['password']))
res = cursor.fetchone()
if res[0]:
global user_data
user_data['Name'] = username
return render_template('login.html', user_data=user_data)
else:
return "Wrong credentials"
else:
return render_template('login.html', user_data=user_data)
@app.route('/items', methods=['GET'])
def items():
cursor = conn.cursor()
cursor.execute("select * from dbo.Items")
return render_template('items.html', data=cursor, user_data=user_data)
if __name__ == '__main__':
app.run(host='0.0.0.0', port='5000')