application.py

import os;
from time import localtime,strftime 
from flask import Flask, render_template, redirect, url_for, flash
from flask_login import LoginManager , login_user, current_user,login_required , logout_user
from flask_socketio import SocketIO, send, emit, join_room,leave_room

from wtforms_fields import *
from models import *


# Configure App
app = Flask(__name__)
app.secret_key = os.environ.get('SECRET')
app.config['WTF_CSRF_SECRET_KEY'] = "4e46dd8a391958071ee420b1a165e252f842484a53be29d98f64adbc23d9905f"

#Configure database
app.config['SQLALCHEMY_DATABASE_URI']= os.environ.get('DATABASE_URL')
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db= SQLAlchemy(app)

# Initialize Flask-SocketIO
socketio = SocketIO(app)
ROOMS = ["lounge","news","games","coding"]


# Configure flask login 
login = LoginManager(app)
login.init_app(app)


@login.user_loader
def load_user(id):

    return User.query.get(int(id))

@app.route("/", methods=['GET', 'POST'])
def index():

    # update database if validation seccess
    reg_form= RegistrationForm()
    if reg_form.validate_on_submit():
        username = reg_form.username.data
        password = reg_form.password.data
        
        hashed_pswd = pbkdf2_sha256.hash(password)

        # Add user to database
        user = User(username=username, password=hashed_pswd)
        db.session.add(user)
        db.session.commit()

        flash('Registered Succesfully. Please login.','success')

        return redirect(url_for('login'))

    return render_template("index.html", form=reg_form)

@app.route("/login", methods=['GET', 'POST'])
def login():
    
    login_form = LoginForm()

    #Allow login if validation success
    if login_form.validate_on_submit():
        user_object = User.query.filter_by(username=login_form.username.data).first()
        login_user(user_object)
        return redirect(url_for('chat'))
        
        return "Not Loggedin"    

    return render_template("login.html",form = login_form)

@app.route("/chat", methods=['GET','POST'])
# @login_required
def chat():

            if not current_user.is_authenticated:
                flash('Please login.','danger')
                return redirect(url_for('login'))
            
            return render_template('chat.html',username=current_user.username,rooms=ROOMS)
            


@app.errorhandler(404)
def page_not_found(e):
    # note that we set the 404 status explicitly
    return render_template('404.html'), 404




@app.route("/logout", methods=['GET'])
def logout():

    logout_user()
    flash('You have logout succesfully.','success')
    return redirect(url_for('login'))


@socketio.on('message')
def message(data):

    send({'msg':data['msg'], 'username':data['username'],'time_stamp': strftime('%b-%d %I:%M%p',localtime())}, room = data['room'])

@socketio.on('join')
def join(data):

    join_room(data['room'])
    send({'msg':data['username']+" has joined the " + data['room']+ " room."}, room=data['room'])

@socketio.on('leave')
def leave(data):

    leave_room(data['room'])
    send({'msg':data['username']+" has left the " + data['room']+ " room."}, room=data['room'])


if __name__ =="__main__":
    app.run()