Gagan0141 commited on
Commit
82b43cd
·
verified ·
1 Parent(s): bf5fcd5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +112 -112
app.py CHANGED
@@ -1,112 +1,112 @@
1
- from flask import Flask, render_template, request, redirect, url_for, session, flash
2
- from flask_sqlalchemy import SQLAlchemy
3
- from werkzeug.security import generate_password_hash, check_password_hash
4
-
5
- # Initialize Flask app
6
- app = Flask(__name__)
7
- app.secret_key = 'your_super_secret_key'
8
-
9
- # Database configuration
10
- app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///database.db'
11
- app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
12
-
13
- # Initialize SQLAlchemy
14
- db = SQLAlchemy(app)
15
-
16
- # ---------------------
17
- # Database Models
18
- # ---------------------
19
-
20
- class User(db.Model):
21
- id = db.Column(db.Integer, primary_key=True)
22
- username = db.Column(db.String(80), unique=True, nullable=False)
23
- password = db.Column(db.String(200), nullable=False)
24
- feedbacks = db.relationship('Feedback', backref='user', lazy=True)
25
-
26
- class Feedback(db.Model):
27
- id = db.Column(db.Integer, primary_key=True)
28
- message = db.Column(db.Text, nullable=False)
29
- user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
30
-
31
- # ---------------------
32
- # Routes
33
- # ---------------------
34
-
35
- @app.before_request
36
- def create_tables():
37
- db.create_all()
38
-
39
- @app.route('/')
40
- def home():
41
- return redirect(url_for('register'))
42
-
43
- @app.route('/register', methods=['GET', 'POST'])
44
- def register():
45
- if request.method == 'POST':
46
- username = request.form['username']
47
- password = request.form['password']
48
-
49
- existing_user = User.query.filter_by(username=username).first()
50
- if existing_user:
51
- flash('Username already exists. Please choose another.', 'danger')
52
- else:
53
- hashed_pw = generate_password_hash(password)
54
- new_user = User(username=username, password=hashed_pw)
55
- db.session.add(new_user)
56
- db.session.commit()
57
- flash('Registration successful! Please login.', 'success')
58
- return redirect(url_for('login'))
59
- return render_template('register.html')
60
-
61
- @app.route('/login', methods=['GET', 'POST'])
62
- def login():
63
- if request.method == 'POST':
64
- username = request.form['username']
65
- password = request.form['password']
66
-
67
- user = User.query.filter_by(username=username).first()
68
- if user and check_password_hash(user.password, password):
69
- session['user'] = user.username
70
- flash('Logged in successfully!', 'success')
71
- return redirect(url_for('welcome'))
72
- else:
73
- flash('Invalid credentials. Try again.', 'danger')
74
- return render_template('login.html')
75
-
76
- @app.route('/welcome')
77
- def welcome():
78
- if 'user' not in session:
79
- flash('Please log in first.', 'warning')
80
- return redirect(url_for('login'))
81
- return render_template('welcome.html', username=session['user'])
82
-
83
- @app.route('/feedback', methods=['GET', 'POST'])
84
- def feedback():
85
- if 'user' not in session:
86
- flash('Please log in first.', 'warning')
87
- return redirect(url_for('login'))
88
-
89
- user = User.query.filter_by(username=session['user']).first()
90
-
91
- if request.method == 'POST':
92
- msg = request.form['message']
93
- feedback_entry = Feedback(message=msg, user=user)
94
- db.session.add(feedback_entry)
95
- db.session.commit()
96
- flash('Feedback received! Thank you.', 'success')
97
- return render_template('feedback.html', message=msg)
98
-
99
- return render_template('feedback.html', message=None)
100
-
101
- @app.route('/logout')
102
- def logout():
103
- session.pop('user', None)
104
- flash('Logged out successfully.', 'info')
105
- return redirect(url_for('login'))
106
-
107
- # ---------------------
108
- # Run the app
109
- # ---------------------
110
-
111
- if __name__ == '__main__':
112
- app.run(debug=True)
 
1
+ from flask import Flask, render_template, request, redirect, url_for, session, flash
2
+ from flask_sqlalchemy import SQLAlchemy
3
+ from werkzeug.security import generate_password_hash, check_password_hash
4
+
5
+ # Initialize Flask app
6
+ app = Flask(__name__)
7
+ app.secret_key = 'your_super_secret_key'
8
+
9
+ # Database configuration
10
+ app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///database.db'
11
+ app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
12
+
13
+ # Initialize SQLAlchemy
14
+ db = SQLAlchemy(app)
15
+
16
+ # ---------------------
17
+ # Database Models
18
+ # ---------------------
19
+
20
+ class User(db.Model):
21
+ id = db.Column(db.Integer, primary_key=True)
22
+ username = db.Column(db.String(80), unique=True, nullable=False)
23
+ password = db.Column(db.String(200), nullable=False)
24
+ feedbacks = db.relationship('Feedback', backref='user', lazy=True)
25
+
26
+ class Feedback(db.Model):
27
+ id = db.Column(db.Integer, primary_key=True)
28
+ message = db.Column(db.Text, nullable=False)
29
+ user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
30
+
31
+ # ---------------------
32
+ # Routes
33
+ # ---------------------
34
+
35
+ @app.before_request
36
+ def create_tables():
37
+ db.create_all()
38
+
39
+ @app.route('/')
40
+ def home():
41
+ return redirect(url_for('register'))
42
+
43
+ @app.route('/register', methods=['GET', 'POST'])
44
+ def register():
45
+ if request.method == 'POST':
46
+ username = request.form['username']
47
+ password = request.form['password']
48
+
49
+ existing_user = User.query.filter_by(username=username).first()
50
+ if existing_user:
51
+ flash('Username already exists. Please choose another.', 'danger')
52
+ else:
53
+ hashed_pw = generate_password_hash(password)
54
+ new_user = User(username=username, password=hashed_pw)
55
+ db.session.add(new_user)
56
+ db.session.commit()
57
+ flash('Registration successful! Please login.', 'success')
58
+ return redirect(url_for('login'))
59
+ return render_template('register.html')
60
+
61
+ @app.route('/login', methods=['GET', 'POST'])
62
+ def login():
63
+ if request.method == 'POST':
64
+ username = request.form['username']
65
+ password = request.form['password']
66
+
67
+ user = User.query.filter_by(username=username).first()
68
+ if user and check_password_hash(user.password, password):
69
+ session['user'] = user.username
70
+ flash('Logged in successfully!', 'success')
71
+ return redirect(url_for('welcome'))
72
+ else:
73
+ flash('Invalid credentials. Try again.', 'danger')
74
+ return render_template('login.html')
75
+
76
+ @app.route('/welcome')
77
+ def welcome():
78
+ if 'user' not in session:
79
+ flash('Please log in first.', 'warning')
80
+ return redirect(url_for('login'))
81
+ return render_template('welcome.html', username=session['user'])
82
+
83
+ @app.route('/feedback', methods=['GET', 'POST'])
84
+ def feedback():
85
+ if 'user' not in session:
86
+ flash('Please log in first.', 'warning')
87
+ return redirect(url_for('login'))
88
+
89
+ user = User.query.filter_by(username=session['user']).first()
90
+
91
+ if request.method == 'POST':
92
+ msg = request.form['message']
93
+ feedback_entry = Feedback(message=msg, user=user)
94
+ db.session.add(feedback_entry)
95
+ db.session.commit()
96
+ flash('Feedback received! Thank you.', 'success')
97
+ return render_template('feedback.html', message=msg)
98
+
99
+ return render_template('feedback.html', message=None)
100
+
101
+ @app.route('/logout')
102
+ def logout():
103
+ session.pop('user', None)
104
+ flash('Logged out successfully.', 'info')
105
+ return redirect(url_for('login'))
106
+
107
+ # ---------------------
108
+ # Run the app
109
+ # ---------------------
110
+
111
+ if __name__ == '__main__':
112
+ app.run(host="0.0.0.0", port=7860, debug=True)