oddjob.py

#Import Flask Library
from flask import Flask, render_template, request, session, url_for, redirect
import pymysql.cursors
from account import Account
from job import Job
from transaction import Transaction
from skill import Skill
from message import Message
from address import Address
from support import Support

app = Flask(__name__)


conn = pymysql.connect(host='localhost',
                       port = 3306,
                       user='root',
                       password='',
                       db='OddJob',
                       charset='utf8mb4',
                       cursorclass=pymysql.cursors.DictCursor)


@app.route('/admin')
def admin():
    applications = Skill.getPendingApplications(conn)
    messages = Support.getNewSupport(conn)
    return render_template('admin.html', applications=applications,messages=messages)

# CUSTOMER Index
@app.route('/')
def index():
    session['message'] = None
    return render_template('index.html', type='CUSTOMER')


#WORKER Index
@app.route('/worker')
def indexWorker():
    session['message'] = None
    return render_template('worker.html', type='WORKER')

#CUSTOMER Home
@app.route('/home')
def home():
    session['message'] = None
    #Only CUSTOMER can access this page
    if (session.get("account") is None or session['account']['accounttype'] != "CUSTOMER"):
        return render_template('error.html')

    email = session['account']['email']

    # Get all related jobs for this email acount
    postedJobs = Job.getPostedJobs(conn, email);
    confirmedJobs = Job.getConfirmedJobs(conn, email);
    completedJobs = Job.getCompletedJobs(conn, email);
    return render_template('home.html', postedJobs=postedJobs, confirmedJobs=confirmedJobs, completedJobs=completedJobs )


#WORKER Home
@app.route('/homeworker')
def homeWorker():
    session['message'] = None
    #Only WORKER can access this page
    if (session.get("account") is None or session['account']['accounttype'] != "WORKER"):
        return render_template('error.html')

    email = session['account']['email']
    acceptedJobs = Job.getWorkerJobs(conn, email, "PENDING");
    confirmedJobs = Job.getWorkerJobs(conn, email, "CONFIRMED");
    completedJobs = Job.getWorkerJobs(conn, email, "COMPLETED");
    print(acceptedJobs)
    return render_template('homeworker.html', acceptedJobs=acceptedJobs, confirmedJobs=confirmedJobs, completedJobs=completedJobs )

#Define a route to hello function
@app.route('/profile')
def profile():
     #Only WORKER can access this page
    if (session.get("account") is None):
        return render_template('error.html')

    email = session['account']['email']
    addresses = Address.getAllAddress(conn, email)
    return render_template('profile.html', addresses = addresses)


#Define a route to jobBoard function
@app.route('/jobboard')
def jobBoard():
    session['message'] = None
     #Only WORKER can access the job board
    if (session.get("account") is None or session['account']['accounttype'] != "WORKER"):
        return render_template('error.html')

    email = session['account']['email']
    approvedSkills = Skill.getApprovedSkills(conn, email)
    jobs = None
    return render_template('jobboard.html', jobs=jobs,approvedSkills=approvedSkills)


#Define route for register
@app.route('/register/<type>')
def register(type):
	#redirect to user's home
    if (session.get('account')):
        return redirect(url_for('home'))

    type2='OddJob Customer'

    if type=='WORKER':
        type2='OddJob Worker'

    return render_template('register.html', type=type, type2=type2)


#Define route for login
@app.route('/login/<type>')
def login(type):
	#redirect to user's home
    if (session.get('account')):
        return redirect(url_for('home'))
    
    type2='OddJob Customer'

    if type=='WORKER':
        type2='OddJob Worker'
    return render_template('login.html', type=type, type2=type2)


#Authenticates the register
@app.route('/registerAuth', methods=['GET', 'POST'])
def registerAuth():

    #grabs information from the forms
    firstName = str(request.form['firstName'])
    lastName = str(request.form['lastName'])
    email = str(request.form['email'])
    password = str(request.form['password1'])
    accountType = str(request.form['accountType'])

    password2 = str(request.form['password2'])
    if (password != password2) :
        session['message'] = "Password does not match"
        return redirect(url_for('register', type=accountType))
    
    acc = Account(email, password, firstName, lastName, accountType);

    try:
        session['account'] = acc.register(conn)
        if (accountType == 'CUSTOMER'):
            return redirect(url_for('home'))
        elif (accountType == 'WORKER'):
            return redirect(url_for('homeWorker'))
    except Exception as e:
        session['message'] = str(e)
        return redirect(url_for('register',type=accountType))

#Authenticates the login
@app.route('/loginAuth', methods=['GET', 'POST'])
def loginAuth():
    session['message'] = None
    #grabs information from the forms
    email = request.form['email']
    password = request.form['password']
    accountType = request.form['accountType']
    data = Account.checkLogIn(conn, email, password, accountType)
    
    if(data):
        #creates a session for the the user
        session['account'] = data
        if (accountType == 'CUSTOMER'):
            return redirect(url_for('home'))
        elif (accountType == 'WORKER'):
            return redirect(url_for('homeWorker'))
    else:
        #returns an error message to the html page
        session['message'] = 'Invalid login'
        return redirect(url_for('login', type=accountType))

@app.route('/logout')
def logout():
    session.pop('account')
    session.pop('message')
    return redirect('/')


@app.route('/jobpost')
def jobPost():
    session['message'] = None
     #Only CUSTOMER can post a job
    if (session.get("account") is None or session['account']['accounttype'] != "CUSTOMER"):
        return render_template('error.html')

    allSkills = Skill.validSkills
    return render_template('jobpost.html', allSkills=allSkills)

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

    job = Job()
    job.jobName = str(request.form['jobName'])
    job.jobType = str(request.form['jobType'])
    job.jobCity = str(request.form['jobCity'])
    job.jobState = str(request.form['jobState'])
    job.jobDescription = str(request.form['jobDescription'])
    job.requestorEmail = session.get('account').get('email')

    job.postJob(conn)
    
    return redirect(url_for('home'))

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

    if not session.get('account'):
        return redirect(url_for('index'))

    trans = Transaction()
    trans.jobId = str(request.form['jobId'])
    trans.amount = str(request.form['jobAmount'])
    trans.note = str(request.form['note'])
    trans.acceptorEmail = session['account']['email']

    trans.postTransaction(conn)
    
    return redirect(url_for('jobBoard'))


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

    if not session.get('account'):
        return redirect(url_for('index'))

    acceptorEmail = str(request.form['acceptorEmail'])
    jobId = str(request.form['jobId'])
    Transaction.confirmTransaction(conn, jobId, acceptorEmail)
    
    return redirect(url_for('home'))


@app.route('/completeJobAuth', methods=['GET', 'POST'])
def completeJobAuth():
    
    if not session.get('account'):
        return redirect(url_for('index'))

    jobId = str(request.form['jobId'])
    Transaction.completeTransaction(conn, jobId, session['account']['email'])
    
    return redirect(url_for('viewJob', id=jobId))


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

    jobId = str(request.form['jobId'])
    tran = Transaction.getConfirmedTransaction(conn, jobId)

    return render_template("payment.html", line=tran)


@app.route('/rateAuth', methods=['GET', 'POST'])
def rateAuth():
    if not session.get('account'):
        return redirect(url_for('index'))

    jobId = str(request.form['jobId'])
    rating = str(request.form['rating'])
    comment = str(request.form['comment'])
    Transaction.rateTransaction(conn, jobId, rating, comment)
    
    return redirect(url_for('viewJob', id=jobId))


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

    if not session.get('account'):
        return redirect(url_for('index'))

    fname = str(request.form['firstName'])
    lname = str(request.form['lastName'])
    email = session['account']['email']
    accType = session['account']['accounttype']
    motto = str(request.form['motto'])

    session['account'] = Account.update(conn, fname, lname, email, motto)
    
    if accType == 'CUSTOMER':
        return redirect(url_for('home'))
    else:
        return redirect(url_for('homeWorker'))


@app.route('/job/<id>')
def viewJob(id):
    session['message'] = None
    if (session.get("account") is None):
        return render_template('error.html')

    job = Job.getJobInfo(conn, id)
    acceptor = Job.getAcceptor(conn, id)
    # Only the owner of this Job and the acceptor of this job can view the detail of this job
    if (job['requestoremail'] == session['account']['email'] or (acceptor is not None and acceptor['acceptoremail'] == session['account']['email'])):
        trans = Job.getAllTransactions(conn, id)
        rate = Job.getRating(conn, id)
        messages = Message.getAllMessages(conn, id)
        validAddress = Address.getAllAddressInCityState(conn, job['requestoremail'], job['jobcity'], job['jobstate'] )
        acc = Job.getAcceptedTransaction(conn, id)
        return render_template('job.html', job=job, trans=trans, messages=messages, rate=rate,validAddress=validAddress, acc=acc)

    return render_template('error.html')


@app.route('/viewworker/<email>')
def viewWorker(email):
    session['message'] = None

    if (session.get("account") is None or session['account']['accounttype'] != "CUSTOMER"):
        return render_template('error.html')

    worker = Account.getAccountInfo(conn, email)
    rating = Account.getAvgRating(conn, email)
    trans = Account.getAllCompletedTransactions(conn, email)

    total = len(trans)
    rating = rating[0].get('avgRate')

    return render_template('viewworker.html', worker=worker, rating=rating,total=total,trans=trans)


@app.route('/backgroundcheck')
def backgroundCheck():
    session['message'] = None

    if (session.get("account") is None or session['account']['accounttype'] != "WORKER"):
        return render_template('error.html')
    remainingSkills = Skill.getRemainingSkills(conn, session['account']['email'])
    pendingSkills = Skill.getPendingSkills(conn, session['account']['email'])
    approvedSkills = Skill.getApprovedSkills(conn, session['account']['email'])
    return render_template('backgroundcheck.html', remainingSkills=remainingSkills, pendingSkills=pendingSkills, approvedSkills=approvedSkills)


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

    if not session.get('account'):
        return redirect(url_for('index'))


    email = session['account']['email']
    skills = request.form.getlist('skill')

    for skill in skills:
        Skill.applySkill(conn, email, skill)

    return redirect(url_for('backgroundCheck'))


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

    if not session.get('account'):
        return redirect(url_for('index'))

    jobid = request.form['jobid']

    Job.cancelJob(conn, jobid)

    return redirect(url_for('home'))

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

    if not session.get('account'):
        return redirect(url_for('index'))

    email = session['account']['email']
    approvedSkills = Skill.getApprovedSkills(conn, email)
    if ('jobType' not in request.form):
        return render_template('jobboard.html', jobs=None,approvedSkills=approvedSkills)
    if ('jobState' not in request.form):
        return render_template('jobboard.html', jobs=None,approvedSkills=approvedSkills)
   
    jobType = request.form['jobType']
    jobState = request.form['jobState']

    jobs = Job.getJobSearchResult(conn, jobType, jobState, email)
    return render_template('jobboard.html', jobs=jobs,approvedSkills=approvedSkills)

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

    if not session.get('account'):
        return redirect(url_for('index'))

    jobid = request.form['jobid']
    email = request.form['email']
    quote = request.form['quote']

    Message.addMessage(conn, jobid, email, quote)

    return redirect(url_for('viewJob', id=jobid))

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

    email = request.form['email']
    skillname = request.form['skillname']

    Skill.approveSkill(conn, email, skillname)

    return redirect(url_for('admin'))


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

    email = session['account']['email']
    oldPassword = request.form['oldpassword']
    newPassword1 = request.form['newpassword1']
    newPassword2 = request.form['newpassword2']
    same = Account.comparePassword(conn, email, oldPassword)
    if (newPassword1 != newPassword2):
        session['message'] = 'Password does not match'
    elif not same:
        session['message'] = 'Old password is wrong'
    else:
        Account.updatePassword(conn, email, newPassword1)
        session['message'] = 'Password changed'

    return redirect(url_for('profile'))


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

    email = session['account']['email']
    alias = request.form['alias']
    address = request.form['address']
    city = request.form['city']
    state = request.form['state']
    
    Address.addAddress(conn, email, alias, address, city, state)

    return redirect(url_for('profile'))

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

    email = session['account']['email']
    alias = request.form['alias']
    address = request.form['address']
    
    Address.editAddress(conn, email, alias, address)

    return redirect(url_for('profile'))

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

    email = session['account']['email']
    alias = request.form['alias']
    jobid = request.form['jobid']
    
    Job.updateJobAddress(conn, alias, jobid)

    return redirect(url_for('viewJob', id=jobid))

@app.route('/support')
def support():
    email = session['account']['email']
    oldMessages = Support.getMyMessages(conn, email)
    return render_template('support.html', oldMessages=oldMessages)

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

    email = request.form['email']
    title = request.form['title']
    message = request.form['message']
    
    Support.addSupport(conn, email, title, message)

    return redirect(url_for('support'))


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

    id = request.form['id']
    reply = request.form['reply']
    
    Support.replySupport(conn, id, reply)

    return redirect(url_for('admin'))

app.secret_key = 'ODDJOB'

if __name__ == "__main__":
    app.run('127.0.0.1', 5000, debug = True)