expub_compendium/web/app/create.py

# @name: create.py
# @version: 0.1
# @creation_date: 2021-10-25
# @license: The MIT License <https://opensource.org/licenses/MIT>
# @author: Simon Bowie <ad7588@coventry.ac.uk>
# @purpose: create route for creating tools, examples, and practices
# @acknowledgements:
# https://www.digitalocean.com/community/tutorials/how-to-make-a-web-application-using-flask-in-python-3

from flask import Blueprint, render_template, request, flash, redirect, url_for
from flask_login import login_required, current_user
from .models import Tool
from .models import Example
from .models import Practice
from werkzeug.exceptions import abort
from . import db

create = Blueprint('create', __name__)

# route for creating a new resource
@create.route('/create', methods=('GET', 'POST'))
@login_required
def create_resource():
    if request.method == 'POST':
        if request.form.get('type') == 'tool':
            name = request.form.get('name')
            description = request.form.get('description')

            if not name:
                flash('Name is required!')
            else:
                tool = Tool.query.filter_by(name=name).first() # if this returns a tool, then the name already exists in database

                if tool: # if a tool is found, we want to redirect back to create page
                    flash('Tool with same name already exists')
                    return redirect(url_for('create.create'))

                # create a new tool with the form data
                new_tool = Tool(name=name, description=description)

                # add the new tool to the database
                db.session.add(new_tool)
                db.session.commit()

        elif request.form.get('type') == 'example':
            name = request.form.get('name')
            description = request.form.get('description')

            if not name:
                flash('Name is required!')
            else:
                example = Example.query.filter_by(name=name).first() # if this returns an example, then the name already exists in database

                if example: # if an example is found, we want to redirect back to create page
                    flash('Example with same name already exists')
                    return redirect(url_for('create.create'))

                # create a new example with the form data
                new_example = Example(name=name, description=description)

                # add the new example to the database
                db.session.add(new_example)
                db.session.commit()

        elif request.form.get('type') == 'practice':
            name = request.form.get('name')
            description = request.form.get('description')

            if not name:
                flash('Name is required!')
            else:
                practice = Practice.query.filter_by(name=name).first() # if this returns a practice, then the name already exists in database

                if practice: # if a practice is found, we want to redirect back to create page
                    flash('Practice with same name already exists')
                    return redirect(url_for('create.create'))

                # create a new practice with the form data
                new_practice = Practice(name=name, description=description)

                # add the new practice to the database
                db.session.add(new_practice)
                db.session.commit()

    return render_template('create.html')