from flask import Flask, jsonify, request, redirect, flash, url_for

from viva1 import *
from werkzeug.utils import secure_filename
import os

app = Flask(__name__)
connect(db="time_frame")
UPLOAD_FOLDER = 'uploads'
ALLOWED_EXTENSIONS = {'txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif'}
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER


def allowed_file(filename):
    return '.' in filename and \
           filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS


# Per klasen user
@app.route('/User', methods=['POST'])
def addOne():
    user = User(
        first_name=request.json['first_name'],
        last_name=request.json['last_name'],
        email=request.json['email'],
        password=request.json['password'],
        role=request.json['role'],
        identifier=request.json['identifier'],
        picture=request.json['picture'],
        activate=request.json['activate']
    )
    user.save()
    return jsonify(success=True)


# Per klasen leaflet
@app.route('/leaflets', methods=['POST'])
def add_leaflets():
    filename = ""
    print(request.files)
    if 'image' not in request.files:
        flash('No File part')
        return redirect(request.url)
    file = request.files['image']
    if file.filename == '':
        flash('No selected file')
        return redirect(request.url)
    if file and allowed_file(file.filename):
        filename = secure_filename(file.filename)
        file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
    products = []
    product = Product(
        name=request.form['name'],
        price=request.form['price'],
        salePrice=request.form['salePrice'],
        image=filename,
        category=request.form['category']
    )
    products.append(product)
    leaflets = Leaflets(
        slug=request.form['slug'],
        title=request.form['title'],
        header="TESt",
        products=products,
        end_date=request.form['end_date']
    )
    leaflets.save()
    product.save()
    return jsonify(success=True)


# Per klasen Order
@app.route('/order', methods=['POST'])
def add_order():
    products = []
    orderProduct = OrderProduct(
        id=request.form['id'],
        name=request.form['name'],
        price=request.form['price'],
        salePrice=request.form['salePrice'],
        quantity=request.form['quantity']
    )
    products.append(orderProduct)
    order = Order(
        title=request.form['title'],
        header="TEST",
        leaflet_id=request.form['leaflet_id'],
        products=products
    )
    order.save()
    return jsonify(success=True)


# Per klasen News
@app.route('/news', methods=['POST'])
def add_news():
    news = News(
        title=request.json['title'],
        description=request.json['description'],
        photos=request.json['photos']
    )
    news.save()
    return jsonify(success=True)


# Per klasen activity
@app.route('/activity', methods=['POST'])
def add_activity():
    filename = ""
    print(request.files)
    if 'image' not in request.files:
        flash('No File part')
        return redirect(request.url)
    file = request.files['image']
    if file.filename == '':
        flash('No selected file')
        return redirect(request.url)
    if file and allowed_file(file.filename):
        filename = secure_filename(file.filename)
        file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
    activity = Activity(
        title=request.form['title'],
        url=request.form['url'],
        image=filename
    )
    activity.save()
    return jsonify(success=True)


# Per klasen Subscriber
@app.route('/subscriber', methods=['POST'])
def add_subscriber():
    subscriber = Subscriber(
        frist_name=request.json['first_name']
    )
    subscriber.save()
    return jsonify(success=True)