app.py

import os
from flask import (Flask, render_template, jsonify, abort,
                   redirect, url_for, request, session)
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy import or_, func, and_
from flask_wtf import FlaskForm
from wtforms import StringField, TextField, PasswordField, StringField
from wtforms.validators import DataRequired, Email, EqualTo
from wtforms.fields.html5 import EmailField

app = Flask(__name__)
app.secret_key = os.environ['SECRET_KEY']
app.config.from_object(os.environ['APP_SETTINGS'])
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db = SQLAlchemy(app)


from controllers.controller import *
from models import Tasks


class LoginForm(FlaskForm):
    email = EmailField('Email', [DataRequired(), Email()])
    password = PasswordField('Password', validators=[DataRequired()])


class RegisterForm(FlaskForm):
    email = EmailField('Email', [DataRequired(), Email()])
    name = StringField('Name', [DataRequired()])
    password = PasswordField('Password', validators=[DataRequired()])
    confirm_password = PasswordField('Confirmar Password', validators=[
        DataRequired(),
        EqualTo('password', message='Passwords devem ser idênticas')
    ])
    photo = StringField('URL da foto')
    address = StringField('Localização')
    contact = StringField('Contacto')
    description = StringField('Descrição')



class NewProposalForm(FlaskForm):
    user = StringField('Utilizador', [DataRequired()])
    offer = StringField('Preço', [DataRequired()])
    description = TextField('Descrição', [DataRequired()])
    task_id = StringField('task_id', [DataRequired()])
    type = StringField('type', [DataRequired()])


class TaskForm(FlaskForm):
    title = StringField('Nome do Trabalho')
    location = StringField('Localização')
    price = StringField('Preço Desejado')
    description = StringField('Descrição do Trabalho')


@app.route("/")
def home():
    types = get_task_types()
    tasks = get_all_tasks()
    for task in tasks:
        task.user_info = get_user_by_id(task.user)
    return render_template("homepage.html", tasks=tasks, types=types)


@app.route("/login", methods=["GET", "POST"])
def login():
    login_form = LoginForm()
    if login_form.validate_on_submit():
        email = request.form['email']
        password = request.form['password']
        user_id = validate_login(email, password)

        if user_id:
            session['id'] = user_id
            session['name'] = get_user_name(user_id)
            return redirect(url_for('home'))
        else:
            login_form.password.errors = ['Login error']
            return render_template('login.html', login_form=login_form)

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


@app.route("/logout", methods=["GET"])
def logout():
    session['id'] = None
    session['name'] = None
    return redirect(url_for('home'))


@app.route("/register", methods=["GET", "POST"])
def register():
    register_form = RegisterForm()
    if register_form.validate_on_submit():
        email = request.form['email']
        name = request.form['name']
        password = request.form['password']
        confirm_password = request.form['confirm_password']
        photo = request.form['photo']
        address = request.form['address']
        contact = request.form['contact']
        description = request.form['description']
        print(photo)

        user = get_user_by_email(email)
        if user:
            register_form.password.errors = ['Utilizador já existe.']
            return render_template('register.html',
                                   register_form=register_form)
        else:
            user_id = register_user(email, name, password, photo, address, contact, description)
            if user_id:
                session['id'] = user_id
                session['name'] = get_user_name(user_id)
                return redirect(url_for('home'))
            else:
                register_form.confirm_password.errors = [
                                'Erro a registar utilizador.'
                                ]
                return render_template('register.html',
                                       register_form=register_form)

    return render_template("register.html", register_form=register_form)


@app.route("/task/<task_id>")
def task(task_id):
    newproposal_form = NewProposalForm()
    id = int(task_id)
    task = get_task(id)
    task.user_info = get_user_by_id(task.user)
    proposals = get_proposals(task_id)
    accepted = False
    for proposal in proposals:
        proposal.user_info = get_user_by_id(proposal.user)
        if proposal.accepted == True:
            accepted =  True

    rating = get_task_rating(id)
    return render_template("task.html", task=task,
                           newproposal_form=newproposal_form,
                           proposals=proposals,
                           session=session,
                           rating=rating,
                           accepted=accepted
                           )


@app.route("/profile/<user_id>")
def profile(user_id):
    id = int(user_id)

    show_info_div = True

    user = get_user_by_id(id)
    tasks = get_user_tasks(id)
    average = get_user_average(id)

    if average:
        average = float("{0:.2f}".format(average))

    for task in tasks:
        task.rating = get_task_rating(task.id)
    return render_template("profile.html",
                           user=user,
                           tasks=tasks,
                           average=average)


@app.route("/proposal", methods=['POST'])
def proposal():
    type = request.form.get('type')

    if 'create' == type:
        taskID = request.form.get('task_id')
        user = request.form.get('user')
        offer = request.form.get('offer')
        description = request.form.get('description')
        msg = insertProposal(taskID, user, offer, description)
        return redirect("/task/" + taskID + "?status=aceite")
    else:
        proposalID = request.form.get('proposalID')
        msg = update_proposal(proposalID, type)
        return msg


@app.route("/new", methods=["GET"])
def new_task():
    task_form = TaskForm()
    types = get_task_types()
    return render_template("new.html", task_form=task_form, task_types=types)


@app.route("/rate", methods=["POST"])
def rate_task():
    print(request.form)
    rate = request.form.get('rate')
    task_id = request.form.get('task_id')
    from_user = request.form.get('from_user')
    to_user = request.form.get('to_user')
    print(rate)
    insert_rating(rate, task_id, from_user, to_user)

    return task(task_id)


@app.route("/new", methods=["POST"])
def new_task_post():
    task_form = TaskForm(request.form)
    title = request.form
    types = get_task_types()
    if task_form.validate():
        title = request.form["title"]
        location = request.form["location"]
        price = float(request.form["price"])
        description = request.form["description"]
        type = request.form["type"]
        user = session["id"]

        try:
            task = Tasks(title=title,
                         location=location,
                         price=price,
                         description=description,
                         user=user,
                         type=type)
            db.session.add(task)
            db.session.commit()
            link = "/task/{}".format(task.id)
            return redirect(link)
        except Exception as e:
            db.session.rollback()
            return render_template("new.html",
                                   task_form=task_form,
                                   task_types=types)
    else:
        return render_template("new.html",
                               task_form=task_form,
                               task_types=types)


@app.route("/search", methods=["GET"])
def search():
    types = get_task_types()
    search_word = request.args.get('search')

    if (search_word in types):
        result = Tasks.query.filter(
                    and_(or_(Tasks.title.like("%" + search_word + "%"),
                        Tasks.description.like("%" + search_word + "%"),
                        Tasks.type == search_word), Tasks.approved == False )).all()
    else:
        result = Tasks.query.filter(
                    and_(or_(Tasks.title.like("%" + search_word + "%"),
                        Tasks.description.like("%" + search_word + "%")),  Tasks.approved == False)).all()

    for task in result:
        task.user_info = get_user_by_id(task.user)

    return render_template("search.html",
                           tasks=result,
                           search_word=search_word,
                           types=types
                           )


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