main.py

from __future__ import print_function
from flask import Flask
from flask import request, render_template, jsonify, make_response, redirect, url_for
from flask_login import LoginManager, UserMixin, login_required, login_user, logout_user, current_user

from forms import *
from auth import User
from functions import *



'''
TODO:
- Add some dummy content on the home page
- Add search sql query
- Add "mark as read" query
- Apply/post job queries
- Error in registration: IntegrityError: 1062 (23000): Duplicate entry 'U1230' for key 'PRIMARY'
'''


app = Flask(__name__)
app.config.update(dict(
    SECRET_KEY="laks;dfjaslk;fjsal;kfjsalfja;lskdfj",
    WTF_CSRF_SECRET_KEY="ofhuio2h3r283ubrqlb32iuob4ir"
))



login_manager = LoginManager()
login_manager.init_app(app)
login_manager.login_view = 'login'

@login_manager.user_loader
def load_user_from_id(id):
    try:
        return User.from_id(id, type='student' if id.startswith('U') else 'company')
    except:
        return None


@app.context_processor
def add_global_template_vars():
    notifications = add_notification(current_user.id) if not current_user.is_anonymous else None
    return dict(
        search_form=SearchForm(query=request.args.get('query')), # add search form for every page
        notifications=notifications
    )


def student_login_required(func):
    pass


'''
Page Routes
'''


@app.route('/')
def index():
    return render_template('pages/index.j2')

@app.route('/search')
def search():
    query = request.args.get('query', '')

    if query:
        students = search_student(query)
        jobs = search_jobs(query)
        company = search_company(query)# if current_user.type != 'company' else []
    else:
        students, company, jobs = [], [], []

    print(students, company)
    return render_template('pages/search.j2', students=students, jobs=jobs, companies=company, title='Search')

##### You can remove this once the messages query is working. it's just for creating fake messages
from datetime import datetime, timedelta
#Ben
# @app.route('/student/<user>')
# def student(user):
#     student_data = stud(user) # get info for a student
#     friends = listfriends(student_data['sid'])
#     friend_ids = [f['sid'] for f in friends]
#     if student_data['privacysetting'] == 'friendly public' and (not current_user.is_authenticated or current_user.id in friend_ids):
#         return redirect(url_for('not_found_error'))
#
#     following = listfollowers_user(student_data['sid'])
#     applications = listapplications(student_data['sid'])
#
#     messages = getmessages(current_user.id, student_data['sid']) if current_user.is_authenticated else None
#     friend_request = get_friend_request(current_user.id, student_data['sid']) if current_user.is_authenticated else None
#     return render_template('pages/student.j2', user=student_data, following=following, friends=friends,
#         applications=applications, messages=messages, friend_request=friend_request, title=student_data['sname'])

@app.route('/student/<user>')
def student(user):
    student_data =  stud(user) # get info for a student
    following = listfollowers_user(student_data['sid'])
    friends = listfriends(student_data['sid'])
    applications = listapplications(student_data['sid'])
    print(applications)
    print(student_data['privacysetting'])
    print(student_data['sid'])
    print(friends)
    # messages = getmessages(current_user.id, student_data['sid'])
    messages = getmessages(current_user.id, student_data['sid']) if current_user.is_authenticated else None
    print(messages)
    return render_template('pages/student.j2', user=student_data, following=following, friends=friends, applications=applications, messages=messages, title=student_data['sname'])




# @app.route('/company/<user>')
# def company(user):
#     company_data = com(user)
#     followers = listfollowers_company(user)
#     jobs = comjobs(company_data['cid'])
#     applications_com = listapplications_com(company_data['cid'])
#     im_following = is_following(current_user.id, company_data['cid']) if current_user.is_authenticated else None
#     return render_template('pages/company.j2', company=company_data, followers=followers,
#         jobs=jobs, applications_com=applications_com, im_following=im_following, title=company_data['cname'])

@app.route('/company/<user>')
def company(user):
    company_data = com(user)
    followers = listfollowers_company(company_data['cid'])
    jobs = comjobs(company_data['cid'])
    applications_com = listapplications_com(company_data['cid'])
    print(company_data['cname'])
    print('123')
    print(user)
    print(followers)
    return render_template('pages/company.j2', company=company_data, followers=followers, jobs=jobs, applications_com=applications_com, title=company_data['cname'])



@app.route('/job/<aid>', methods=['GET', 'POST'])
def job(aid):
    job_data = selectjob(aid)
    sent_application = has_applied(aid, current_user.id) if current_user.is_authenticated else None
    print(sent_application)
    form = ApplicationForm(request.form)
    if request.method == 'POST' and form.validate() and current_user.is_authenticated:
        sendapplication(aid, current_user.id, form.email_phone.data)
        return redirect(url_for('student', user=current_user.username))
    return render_template('pages/job.j2', job=job_data, aid=aid, form=form, title=job_data['title'])



# @app.route('/notifications')
# @login_required
# def notifications():
#     Noti_data = add_notification(current_user.id)
#     return render_template('pages/notification.j2', notifs=Noti_data, user=current_user, title='Notifications')
@app.route('/notifications')
@login_required
def notifications():
    Noti_data = add_notification(current_user.id)
    Re_data = add_request(current_user.id)
    return render_template('pages/notification.j2', notifs=Noti_data, requests = Re_data, user=current_user, title='Notifications')



'''
Asynchronous Methods
'''

@app.route('/notification/<aid>/<sid>/mark_read', methods=['POST'])
@login_required
def mark_read(aid, sid):
    status = read_notification(aid, sid) # mark notification as read
    return jsonify({
        'success': status
    })

@app.route('/company/<user>/follow', methods=['POST'])
@login_required
def follow(user):
    status = follow_com(user, current_user.id)
    return jsonify({
        'success': status
    })

@app.route('/student/<user>/befriend', methods=['POST'])
@login_required
def friend_request(user):
    status = send_friend_request(current_user.id, user)
    return jsonify({
        'success': status
    })

@app.route('/student/<user>/accfri',methods = ['POST'])
@login_required
def acc_frirequest(user):
    Noti_data = add_notification(current_user.id)
    Re_data = add_request(current_user.id)
    if request.method == 'POST':
        accept_friend_request(user,current_user.id)
        print(user)
        print(current_user.id)
        print(datetime.now())
    return render_template('pages/notification.j2', notifs=Noti_data, requests = Re_data, user=current_user, title='Notifications')

#
@app.route('/student/<user>/rejfri',methods = ['POST'])
@login_required
def Rej_frirequest(user):
    Noti_data = add_notification(current_user.id)
    Re_data = add_request(current_user.id)
    if request.method == 'POST':
        reject_friend_request(user,current_user.id)
        print(user)
        print('123')
        print(current_user.id)
        print(datetime.now())
    return render_template('pages/notification.j2', notifs=Noti_data, requests = Re_data, user=current_user, title='Notifications')




@app.route('/apply/<aid>/<contact_by>', methods=['POST'])
@login_required
def apply(aid, contact_by):
    status = sendapplication(aid, current_user.id, contact_by)
    return jsonify({
        'success': status
    })


@app.route('/message/<sid>/send', methods=['POST'])
@login_required
def send_message(sid):
    status = False
    timestamp = datetime.now()
    message = request.form.get('message')

    if message:
        status = sendmessage(current_user.id, sid, message, timestamp)
    return jsonify({
        'success': status,
        'timestamp': timestamp.strftime('%Y-%m-%d %H:%M:%S')
    })

@app.route('/messages/<sid>/get')
@login_required
def get_messages(sid):
    date = request.args.get('date')
    messages = getnewmessages(sid, current_user.id, date)
    return jsonify({
        # 'success': status,
        'messages': messages
    })

@app.route('/messages/<sid>/all')
@login_required
def get_all_messages(sid):
    messages = getmessages(sid, current_user.id)
    return jsonify({
        # 'success': status,
        'messages': messages
    })


@app.route('/message/<sid>/read/<date>', methods=['POST'])
@login_required
def read_message(sid, date):
    status = markreadmessage(current_user.id, sid, date)
    return jsonify({
        'success': status
    })


'''
Form Pages
'''


@app.route('/login', methods=['GET', 'POST'])
def login():
    form = LoginForm(request.form)
    if request.method == 'POST' and form.validate():
        user = User.from_form(form)
        print(user)
        if user:
            if user.type == 'company':
                login_user(user, remember=form.remember.data)
                print(current_user)
                return redirect(url_for('company', user=user.username))
            elif user.type == 'student':
                login_user(user, remember=form.remember.data)
                return redirect(url_for('student', user=user.username))
    return render_template('form/login.j2', form=form, title='Login')



@app.route('/register', methods=['GET', 'POST'])
def register():
    form = RegistrationForm(request.form)
    if request.method == 'POST' and form.validate():
        if form.act_type.data == 'company':
            companyregister(form.name.data, form.password.data, form.username.data)
            user = User.from_form(form)
            if user:
                login_user(user)
                print('logged in company', user.username)
                return redirect(url_for('company', user=user.username))
        elif form.act_type.data == 'student':
            studentregister(form.name.data, form.password.data, form.username.data)
            user = User.from_form(form)
            if user:
                login_user(user)
                print('logged in user', user.username)
                return redirect(url_for('student', user=user.username))
    return render_template('form/register.j2', form=form, title = 'Register')



@app.route('/student/<user>/updateprofile', methods=['GET', 'POST'])
@login_required
def student_edit(user):
    form = UpdateForm(request.form)
    student_data = stud(user) #use loginname
    print(current_user.id)
    if request.method == 'POST' and form.validate():
        print('123')
        # phone,email,university,GPA,major,interests,qualification,sid
        updateprofile(form.phone.data, form.email.data, form.university.data, form.GPA.data, form.major.data, 
                        form.interest.data, form.qualifications.data, form.privacysetting.data, current_user.id)
        print(current_user.id)
        print(form.email.data)
        return redirect(url_for('student', user=current_user.username))
    return render_template('form/student_update.j2', form=form, user=student_data, title='{}|Update Profile'.format(student_data['sname']))


#func: updateprofile_com
@app.route('/company/<user>/updateprofile', methods=['GET', 'POST'])
@login_required
def company_edit(user):
    form = UpdateForm_com(request.form)
    company_data = com(user) #use loginname
    print(current_user.id)
    if request.method == 'POST' and form.validate():
        print('123')
        updateprofile_com(form.location.data, form.industry.data, current_user.id)
        print(current_user.id)
        print(form.location.data)
        return redirect(url_for('company', user=company_data['username']))
    return render_template('form/company_update.j2', form=form, user=company_data)


@app.route('/post_job/<user>', methods=['GET', 'POST'])
@login_required
def post_job(user):
    form = PostJob(request.form)
    company_data = com_from_id(user) # get info for a company
    if request.method == 'POST' and form.validate():
        postjobs(current_user.id,form.joblocation.data,form.title.data,form.salary.data, form.bk.data, form.description.data)
        return redirect(url_for('company', user=company_data['username']))
        print(company_data['location'])
        print("99")
    return render_template('form/job_posting.j2', form=form, user=company_data)

@app.route('/candidate/<user>/<job>')
def candidate(user,job):
    candidate_data =  stud(user) # get info for a student
    # print(candidate_data['sid'])
    following = listfollowers_user(candidate_data['sid'])
    friends = listfriends(candidate_data['sid'])
    job = job
    print(user)
    print(candidate_data)
    # print(candidate_data['sid'])
    # print(friends)
    return render_template('pages/candidate.j2', user=candidate_data, job = job, friends = friends, following = following)

@app.route('/Application_suc/<user>/<job>', methods=['POST'])
@login_required
def decision(user,job):
    status = Application_suc(user, job)
    print(user)
    print(job)
    return jsonify({
        'success': status
    })

@app.route('/Application_fail/<user>/<job>', methods=['POST'])
@login_required
def decision_fail(user,job):
    status = Application_fail(user, job)
    print(user)
    print(job)
    return jsonify({
        'success': status
    })



@app.route('/logout')
@login_required
def logout():
    logout_user()
    return redirect(url_for('index'))



@app.errorhandler(500)
def internal_error(error):
    return render_template('layouts/error.j2',
        error_code=500, title='Internal Server Error',
        message="It's our bad. Sorry! :|"), 500


@app.errorhandler(404)
def not_found_error(error):
    return render_template('layouts/error.j2',
        error_code=404, title='Page Not Found',
        message="I think you're lost. That sucks."), 404





if __name__ == '__main__':
    app.run(debug=True, port=5003)