# Imports
from flask import Flask, jsonify, request, url_for, json
from flask_cors import CORS

# Custom imports
from database.DP1Database import Database

# Start app
app = Flask(__name__)
CORS(app)

conn = Database(app=app, user='root', password='Najiseyedhosseinzadeh', db='trein')

# Custom endpoint
endpoint = '/api/v1'



# ROUTES
@app.route(endpoint + '/bestemmingen', methods=['GET'])
def besteming():
    if request.method == 'GET':
        bestemmingen = conn.get_data('SELECT * FROM bestemmingen')
        return jsonify(bestemmingen)


@app.route(endpoint + '/treinen', methods=['GET', 'POST'])
def trein():
    if request.method=='GET':
        treinen = conn.get_data('SELECT * FROM treinen')
        return jsonify(treinen)
    elif request.method=='POST':
        requestje = request.get_json()
        nieuwe_trein = conn.set_data("INSERT INTO treinen (vertrek, bestemmingID, spoor, vertraging,afgeschaft) VALUES (%s,%s,%s,%s,%s)",
        [requestje['vertrek'], requestje['bestemmingID'], requestje['spoor'],
        requestje['vertraging'],requestje['afgeschaft']])
        return jsonify(nieuwe_trein)


@app.route(endpoint + '/treinen/<idtrein>', methods=['GET','PUT', "DELETE"])
def treintje(idtrein):
    if request.method == 'GET':
        trein = conn.get_data('SELECT * FROM treinen WHERE idtrein = %s',idtrein)
        return jsonify(trein)
    elif request.method == 'PUT':
        requestje = request.get_json()
        update_trein = conn.set_data(
            "UPDATE treinen SET vertrek = %s, bestemmingID = %s, spoor = %s, vertraging = %s, afgeschaft = %s WHERE idtrein = %s",
            [requestje['vertrek'], requestje['bestemmingID'], requestje['spoor'],
            requestje['vertraging'], requestje['afgeschaft'], idtrein])
        return jsonify(update_trein)
    elif request.method == "DELETE":
        verwijderde_trein = conn.delete_data("DELETE FROM treinen WHERE idtrein = %s", idtrein)
        return jsonify(verwijderde_trein)


@app.route(endpoint+ '/treinen/bestemming/<bestemming_id>', methods=['GET'])
def get_trein_per_bestemming(bestemming_id):
    if request.method == 'GET':
        bestemming_trein = conn.get_data("SELECT * FROM treinen t INNER JOIN bestemmingen b ON t.bestemmingID = b.idbestemming WHERE b.idbestemming = %s", [bestemming_id])
        return jsonify(bestemming_trein)


@app.route(endpoint + '/')
def index():
        return 'index'


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