|
|
@@ -0,0 +1,260 @@ |
|
|
|
from flask import Flask, render_template, request, redirect, url_for, jsonify, flash, session |
|
|
|
from flask_sqlalchemy import SQLAlchemy |
|
|
|
from werkzeug.security import generate_password_hash, check_password_hash |
|
|
|
from flask_jwt_extended import JWTManager, create_access_token, jwt_required, get_jwt_identity |
|
|
|
from flask_login import LoginManager, UserMixin, login_user, logout_user, login_required |
|
|
|
from datetime import datetime, timedelta |
|
|
|
from sqlalchemy import MetaData, create_engine |
|
|
|
from sqlalchemy.ext.automap import automap_base |
|
|
|
from sqlalchemy.orm import sessionmaker |
|
|
|
from sqlalchemy.ext.declarative import declarative_base |
|
|
|
|
|
|
|
|
|
|
|
app = Flask(__name__) |
|
|
|
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql+mysqlconnector://root:12345@localhost/assignment' |
|
|
|
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False |
|
|
|
app.config['JWT_SECRET_KEY'] = 'jwt_secret_key' |
|
|
|
app.config['JWT_ACCESS_TOKEN_EXPIRES'] = timedelta(days=1) |
|
|
|
app.config['SECRET_KEY'] = 'secret_key' |
|
|
|
jwt = JWTManager(app) |
|
|
|
login_manager = LoginManager(app) |
|
|
|
login_manager.login_view = 'login' |
|
|
|
|
|
|
|
db = SQLAlchemy(app) |
|
|
|
|
|
|
|
class User(db.Model): |
|
|
|
user_id = db.Column(db.Integer, primary_key=True, autoincrement=True) |
|
|
|
email = db.Column(db.String(255), nullable=False) |
|
|
|
given_name = db.Column(db.String(255), nullable=False) |
|
|
|
surname = db.Column(db.String(255), nullable=False) |
|
|
|
city = db.Column(db.String(255)) |
|
|
|
phone_number = db.Column(db.String(20)) |
|
|
|
profile_description = db.Column(db.Text) |
|
|
|
password = db.Column(db.String(255), nullable=False) |
|
|
|
role = db.Column(db.String(20), nullable = False) |
|
|
|
|
|
|
|
class Caregiver(db.Model): |
|
|
|
caregiver_user_id = db.Column(db.Integer, db.ForeignKey('user.user_id'), primary_key=True) |
|
|
|
photo = db.Column(db.LargeBinary) |
|
|
|
gender = db.Column(db.String(20)) |
|
|
|
caregiving_type = db.Column(db.String(255)) |
|
|
|
hourly_rate = db.Column(db.Integer) |
|
|
|
user = db.relationship('User', backref='caregiver', uselist=False) |
|
|
|
|
|
|
|
class Member(db.Model): |
|
|
|
member_user_id = db.Column(db.Integer, db.ForeignKey('user.user_id'), primary_key=True) |
|
|
|
house_rules = db.Column(db.Text) |
|
|
|
user = db.relationship('User', backref='member', uselist=False) |
|
|
|
|
|
|
|
class Address(db.Model): |
|
|
|
member_user_id = db.Column(db.Integer, db.ForeignKey('member.member_user_id'), primary_key=True) |
|
|
|
house_number = db.Column(db.Integer) |
|
|
|
street = db.Column(db.String(255)) |
|
|
|
town = db.Column(db.String(255)) |
|
|
|
member = db.relationship('Member', backref='address', uselist=False) |
|
|
|
|
|
|
|
class Job(db.Model): |
|
|
|
job_id = db.Column(db.Integer, primary_key=True, autoincrement=True) |
|
|
|
member_user_id = db.Column(db.Integer, db.ForeignKey('member.member_user_id')) |
|
|
|
required_caregiving_type = db.Column(db.String(255)) |
|
|
|
other_requirements = db.Column(db.Text) |
|
|
|
date_posted = db.Column(db.Date) |
|
|
|
member = db.relationship('Member', backref='jobs') |
|
|
|
|
|
|
|
class JobApplication(db.Model): |
|
|
|
caregiver_user_id = db.Column(db.Integer, db.ForeignKey('caregiver.caregiver_user_id'), primary_key=True) |
|
|
|
job_id = db.Column(db.Integer, db.ForeignKey('job.job_id'), primary_key=True) |
|
|
|
date_applied = db.Column(db.Date) |
|
|
|
|
|
|
|
class Appointment(db.Model): |
|
|
|
appointment_id = db.Column(db.Integer, primary_key=True, autoincrement=True) |
|
|
|
caregiver_user_id = db.Column(db.Integer, db.ForeignKey('caregiver.caregiver_user_id')) |
|
|
|
member_user_id = db.Column(db.Integer, db.ForeignKey('member.member_user_id')) |
|
|
|
appointment_date = db.Column(db.Date) |
|
|
|
appointment_time = db.Column(db.Time) |
|
|
|
work_hours = db.Column(db.String(50)) |
|
|
|
status = db.Column(db.String(50)) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
login_manager = LoginManager(app) |
|
|
|
login_manager.login_view = 'login' |
|
|
|
|
|
|
|
@login_manager.user_loader |
|
|
|
def load_user(user_id): |
|
|
|
return User.query.get(int(user_id)) |
|
|
|
|
|
|
|
headers = {'Content-Type': 'application/json'} |
|
|
|
|
|
|
|
@login_manager.user_loader |
|
|
|
def load_user(user_id): |
|
|
|
return User.query.get(int(user_id)) |
|
|
|
|
|
|
|
@app.route('/', methods=['GET']) |
|
|
|
def login_page(): |
|
|
|
return render_template('login.html') |
|
|
|
|
|
|
|
@app.route('/', methods=['POST']) |
|
|
|
def login_user(): |
|
|
|
if request.method == 'POST': |
|
|
|
email = request.form['email'] |
|
|
|
password = request.form['password'] |
|
|
|
|
|
|
|
user = User.query.filter_by(email=email).first() |
|
|
|
|
|
|
|
if user and user.password == password: |
|
|
|
|
|
|
|
session['user_id'] = user.user_id |
|
|
|
session['role'] = user.role |
|
|
|
access_token = create_access_token(identity=user.user_id) |
|
|
|
|
|
|
|
if user.role == 'caregiver': |
|
|
|
return redirect(url_for('caregiver_dashboard')) |
|
|
|
elif user.role == 'family': |
|
|
|
return redirect(url_for('family_dashboard')) |
|
|
|
|
|
|
|
else: |
|
|
|
return jsonify({'status': 'error', 'message': 'Invalid email or password'}), 401 |
|
|
|
|
|
|
|
@app.route('/caregiver_dashboard') |
|
|
|
def caregiver_dashboard(): |
|
|
|
if 'user_id' in session and session.get('role') == 'caregiver': |
|
|
|
user_id = session.get('user_id') |
|
|
|
user = User.query.get(user_id) |
|
|
|
return render_template('caregiver_dashboard.html', user=user) |
|
|
|
else: |
|
|
|
return redirect(url_for('login_page')) |
|
|
|
|
|
|
|
@app.route('/family_dashboard') |
|
|
|
def family_dashboard(): |
|
|
|
if 'user_id' in session and session.get('role') == 'family': |
|
|
|
user_id = session.get('user_id') |
|
|
|
user = User.query.get(user_id) |
|
|
|
jobs = Job.query.filter_by(member_user_id=user_id).all() |
|
|
|
return render_template('family_dashboard.html', user=user, jobs=jobs) |
|
|
|
else: |
|
|
|
return redirect(url_for('login_page')) |
|
|
|
|
|
|
|
@app.route('/protected', methods=['GET']) |
|
|
|
@login_required |
|
|
|
def protected(): |
|
|
|
current_user = get_jwt_identity() |
|
|
|
return jsonify(logged_in_as=current_user), 200 |
|
|
|
|
|
|
|
|
|
|
|
@app.route('/register', methods=['GET']) |
|
|
|
def registration_page(): |
|
|
|
return render_template('register.html') |
|
|
|
|
|
|
|
|
|
|
|
@app.route('/register', methods=['POST']) |
|
|
|
def register(): |
|
|
|
username = request.form.get('username', None) |
|
|
|
password = request.form.get('password', None) |
|
|
|
given_name = request.form.get('name', None) |
|
|
|
surname = request.form.get('surname', None) |
|
|
|
city = request.form.get('city', None) |
|
|
|
phone_number = request.form.get('phone_number', None) |
|
|
|
profile_description = request.form.get('profile_description', None) |
|
|
|
role = request.form.get('role', None) |
|
|
|
|
|
|
|
if not username or not password or not given_name or not surname or not city or not phone_number or not profile_description or not role: |
|
|
|
return jsonify({'message': 'All fields are required'}), 400 |
|
|
|
|
|
|
|
existing_user = User.query.filter_by(email=username).first() |
|
|
|
|
|
|
|
if existing_user: |
|
|
|
return jsonify({'message': 'User already exists'}), 409 |
|
|
|
|
|
|
|
new_user = User(email=username, password=password, given_name=given_name, surname=surname, city=city, phone_number=phone_number, profile_description=profile_description, role=role) |
|
|
|
|
|
|
|
db.session.add(new_user) |
|
|
|
db.session.commit() |
|
|
|
|
|
|
|
return redirect(url_for('login_page')) |
|
|
|
|
|
|
|
@app.route('/update/<int:user_id>', methods=['GET', 'POST']) |
|
|
|
def update_user(user_id): |
|
|
|
user = User.query.get(user_id) |
|
|
|
|
|
|
|
if request.method == 'POST': |
|
|
|
user.email = request.form['email'] |
|
|
|
user.given_name = request.form['given_name'] |
|
|
|
user.surname = request.form['surname'] |
|
|
|
user.city = request.form['city'] |
|
|
|
user.phone_number = request.form['phone_number'] |
|
|
|
user.profile_description = request.form['profile_description'] |
|
|
|
|
|
|
|
db.session.commit() |
|
|
|
|
|
|
|
if user.role == 'caregiver': |
|
|
|
return redirect(url_for('caregiver_dashboard')) |
|
|
|
elif user.role == 'family': |
|
|
|
return redirect(url_for('family_dashboard')) |
|
|
|
|
|
|
|
return render_template('update.html', user=user) |
|
|
|
|
|
|
|
|
|
|
|
@app.route('/create_job', methods=['GET', 'POST']) |
|
|
|
def create_job(): |
|
|
|
if request.method == 'POST': |
|
|
|
user_id = session.get('user_id') |
|
|
|
member = User.query.get(user_id) |
|
|
|
|
|
|
|
new_job = Job( |
|
|
|
member_user_id=user_id, |
|
|
|
required_caregiving_type=request.form['required_caregiving_type'], |
|
|
|
other_requirements=request.form['other_requirements'], |
|
|
|
date_posted=datetime.now().date() |
|
|
|
) |
|
|
|
|
|
|
|
db.session.add(new_job) |
|
|
|
db.session.commit() |
|
|
|
|
|
|
|
flash("Job created successfully", "success") |
|
|
|
return redirect(url_for('family_dashboard')) |
|
|
|
|
|
|
|
return render_template('create_job.html') |
|
|
|
|
|
|
|
@app.route('/update_job/<int:job_id>', methods=['GET', 'POST']) |
|
|
|
def update_job(job_id): |
|
|
|
job = Job.query.get(job_id) |
|
|
|
|
|
|
|
if not job: |
|
|
|
flash("Job not found", "error") |
|
|
|
return redirect(url_for('family_dashboard')) |
|
|
|
|
|
|
|
if request.method == 'POST': |
|
|
|
job.required_caregiving_type = request.form['required_caregiving_type'] |
|
|
|
job.other_requirements = request.form['other_requirements'] |
|
|
|
job.date_posted = datetime.now().date() |
|
|
|
|
|
|
|
db.session.commit() |
|
|
|
|
|
|
|
flash("Job updated successfully", "success") |
|
|
|
return redirect(url_for('family_dashboard')) |
|
|
|
|
|
|
|
return render_template('update_job.html', job=job) |
|
|
|
|
|
|
|
@app.route('/delete_job/<int:job_id>', methods=['GET']) |
|
|
|
def delete_job(job_id): |
|
|
|
job = Job.query.get(job_id) |
|
|
|
|
|
|
|
if not job: |
|
|
|
flash("Job not found", "error") |
|
|
|
return redirect(url_for('family_dashboard')) |
|
|
|
|
|
|
|
db.session.delete(job) |
|
|
|
db.session.commit() |
|
|
|
|
|
|
|
flash("Job deleted successfully", "success") |
|
|
|
return redirect(url_for('family_dashboard')) |
|
|
|
|
|
|
|
|
|
|
|
@app.route('/logout') |
|
|
|
def logout_user(): |
|
|
|
session.pop('user_id', None) |
|
|
|
return redirect(url_for('login_page')) |
|
|
|
|
|
|
|
if __name__ == '__main__': |
|
|
|
app.run(debug=True) |