from flask import (
    Blueprint, flash, redirect, render_template, request, url_for
)
from db import get_db
from werkzeug.exceptions import abort
from datetime import datetime

bp = Blueprint('zamowienia', __name__, url_prefix='/zamowienia')


@bp.route('/lista')
def lista():
    dane = get_db().execute("""
        SELECT z.*, k.* FROM zamowienia z
        INNER JOIN klienci k ON z.id_klienta = k.id
        ORDER BY z.data
        """).fetchall()
    if not dane:
        flash('Brak zamówień!')
    return render_template('zamowienia_lista.html', dane=dane)


@bp.route('/dodaj', methods=['GET', 'POST'])
def dodaj():
    pass


def get_zamowienia(z_id):
    pass

@bp.route('/<int:z_id>/edytuj', methods=['GET', 'POST'])
def edytuj(z_id):
    pass


@bp.route('/usun/<int:z_id>', methods=['GET', 'POST'])
def usun(z_id):
    pass

SELECT z.*, k.* FROM zamowienia z
INNER JOIN klienci k ON z.id_klienta = k.id
ORDER BY z.data

{% extends "base.html" %}
{% block podtytul %}Zamówienia{% endblock %}

{% block tresc %}
    <ol>
      {% for rekord in dane %}
        <li>{{ rekord.imie }}, {{ rekord.nazwisko }}, {{ rekord.data }} –
          <a href="{{ url_for('zamowienia.edytuj', z_id=rekord.id) }}">Edytuj</a>,
          <a href="{{ url_for('zamowienia.usun', z_id=rekord.id) }}">Usuń</a>
        </li>
      {% endfor %}
    </ol>
    <p><a href="{{ url_for('zamowienia.dodaj') }}">Dodaj zamówienie</a></p>
{% endblock %}

{% extends "base.html" %}
{% block podtytul %}Zamówienie – dodawanie{% endblock %}

{% block tresc %}
  <form method="POST" action="{{ url_for('zamowienia.dodaj') }}">
    <label for="klient">Klient:</label>
      <select name="klient" id="klient" required>
        <option value="">Wybierz klienta...</option>
      {% for k in klienci %}
        <option value="{{ k.id }}">{{ k.imie }} {{ k.nazwisko }}</option>
      {% endfor %}
      </select>
    <br>
    <label for="data">Data, godzina:</label>
    <input type="datetime-local" id="data" name="data" value="{{ dzis }}" required>
    <br>
      <fieldset>
      <legend>Dania</legend>
      <p>Zaznacz dania:</p>
        {% for d in dania %}
          <div style="white-space:nowrap">
            <input type="checkbox" name="dania" value="{{ d.id }}"> {{ d.nazwa }}, {{ d.cena }} zł &nbsp;
            <input type="number" name="liczba{{ d.id }}" value="1" min="0" step="1">
          </div>
        {% endfor %}
      </fieldset>
    <br>
    <label for="rabat">Rabat:</label>
    <input type="number" id="rabat" name="rabat" value="0" step=".01">
    <button type="submit">Dodaj</button>
  </form>
{% endblock %}

<select name="klient" id="klient">
  <option value="k_id1">Klient 1</option>
  <option value="k_id2">Klient 2</option>
  ...
</select>

@bp.route('/dodaj', methods=['GET', 'POST'])
def dodaj():
    db = get_db().cursor()
    if request.method == 'POST':
        if not request.form.getlist('dania'):
            flash("Nie wybrano dań!")
        else:
            id_klienta = request.form['klient']
            data = datetime.strptime(request.form['data'], '%Y-%m-%dT%H:%M')
            rabat = request.form['rabat']

            db.execute('INSERT INTO zamowienia VALUES (?, ?, ?, ?)',
                       [None, id_klienta, data, rabat])
            z_id = db.lastrowid
            for d_id in request.form.getlist('dania'):
                liczba = request.form[f'liczba{d_id}']
                db.execute('INSERT INTO sklady_zamowien VALUES (?, ?, ?, ?)',
                           [None, z_id, d_id, liczba])
            db.execute('COMMIT')
            flash(f'Dodano dane zamówienia id {z_id}.')

        return redirect(url_for('zamowienia.lista'))

    klienci = db.execute('SELECT * FROM klienci').fetchall()
    dania = db.execute('SELECT * FROM dania WHERE menu IS TRUE').fetchall()
    dzis = datetime.now().strftime('%Y-%m-%dT%H:%M')
    return render_template('zamowienie_dodaj.html', klienci=klienci, dania=dania, dzis=dzis)

db.execute('INSERT INTO zamowienia VALUES (?, ?, ?, ?)',
           [None, id_klienta, data, rabat])

db.execute('INSERT INTO sklady_zamowien VALUES (?, ?, ?, ?)',
           [None, z_id, d_id, liczba])

def get_zamowienie(z_id):
    db = get_db()
    zamowienie = db.execute("""
        SELECT z.*, k.* FROM zamowienia z
        INNER JOIN klienci k ON z.id_klienta = k.id
        WHERE z.id = ?
        """, [z_id]).fetchone()

    if zamowienie is None:
        abort(404, f"Zamówienie o id {z_id} nie istnieje.")
    return zamowienie


@bp.route('/<int:z_id>/edytuj', methods=['GET', 'POST'])
def edytuj(z_id):
    dane = get_zamowienie(z_id)
    db = get_db()

    # poniżej wstawimy kod obsługujący żądania typu POST


    dania = db.execute('SELECT * FROM dania WHERE menu IS TRUE').fetchall()
    dania_zamowione = db.execute("""
        SELECT s_z.liczba, d.* FROM sklady_zamowien s_z
        INNER JOIN zamowienia z ON s_z.id_zamowienia = z.id
        INNER JOIN dania d ON s_z.id_dania = d.id
        WHERE z.id = ?
        """, [z_id]).fetchall()
    dania_id = [row[1] for row in dania_zamowione]
    dania = [row for row in dania if row['id'] not in dania_id]

    return render_template('zamowienie_edytuj.html', rekord=dane, dania=dania, dania_z=dania_zamowione)

SELECT z.*, k.* FROM zamowienia z
INNER JOIN klienci k ON z.id_klienta = k.id
WHERE z.id = ?

SELECT s_z.liczba, d.* FROM sklady_zamowien s_z
INNER JOIN zamowienia z ON s_z.id_zamowienia = z.id
INNER JOIN dania d ON s_z.id_dania = d.id
WHERE z.id = ?

{% extends "base.html" %}
{% block podtytul %}Zamówienie – edycja{% endblock %}

{% block tresc %}
  <p>{{ rekord.id }}, {{ rekord.id_klienta }}, {{ rekord.data }}</p>
  <form method="POST" action="{{ url_for('zamowienia.edytuj', z_id=rekord.id) }}">
    <label for="klient">Klient:</label>
    <input type="text" id="klient" name="klient" value="{{ rekord.id_klienta }} {{ rekord.imie }} {{ rekord.nazwisko }}" readonly>
    <br>
    <label for="data">Data, godzina:</label>
    <input type="datetime-local" id="data" name="data" value="{{ rekord.data }}" required>
    <br>
      <fieldset>
      <legend>Dania</legend>
      <p>Zamówione:</p>
        {% for d in dania_z %}
          <div style="white-space:nowrap">
            <input type="checkbox" name="dania" value="{{ d.id }}" checked> {{ d.nazwa }}, {{ d.cena }} zł &nbsp;
            <input type="number" name="liczba{{ d.id }}" value="{{ d.liczba }}" min="0" step="1">
          </div>
        {% endfor %}
      <p>Dania dostępne:</p>
        {% for d in dania %}
          <div style="white-space:nowrap">
            <input type="checkbox" name="dania" value="{{ d.id }}"> {{ d.nazwa }}, {{ d.cena }} zł &nbsp;
            <input type="number" name="liczba{{ d.id }}" value="1" min="0" step="1">
          </div>
        {% endfor %}
      </fieldset>
    <br>
    <label for="rabat">Rabat:</label>
    <input type="number" id="rabat" name="rabat" value="{{ rekord.rabat }}" step=".01">
    <button type="submit">Zapisz</button>
  </form>
{% endblock %}

    # poniżej wstawimy kod obsługujący żądania typu POST
    if request.method == 'POST':
        if not request.form.getlist('dania'):
            flash("Nie wybrano dań!")
        else:
            data = datetime.strptime(request.form['data'], '%Y-%m-%dT%H:%M')
            rabat = request.form['rabat']
            db.execute('UPDATE zamowienia SET data = ?, rabat = ? WHERE id = ?', [data, rabat, z_id])

            dania_z = [row[0] for row in db.execute("""
                SELECT id_dania FROM sklady_zamowien WHERE id_zamowienia = ?
                """, [z_id])]
            dania = [int(d_id) for d_id in request.form.getlist('dania')]

            for d_id in dania:
                liczba = request.form[f'liczba{d_id}']
                if d_id in dania_z:
                    db.execute(
                        'UPDATE sklady_zamowien SET liczba = ?'
                        'WHERE id_zamowienia = ? AND id_dania = ?', [liczba, z_id, d_id])
                else:
                    db.execute('INSERT INTO sklady_zamowien VALUES (?, ?, ?, ?)', [None, z_id, d_id, liczba])
            for d_id in set(dania_z) - set(dania):
                db.execute('DELETE FROM sklady_zamowien WHERE id_zamowienia = ? AND id_dania = ?',
                           [z_id, d_id])

            db.commit()

            flash(f'Zaktualizowano dane zamówienia id {z_id}.')

        return redirect(url_for('zamowienia.lista'))

dania_z = [row[0] for row in db.execute("""
                SELECT id_dania FROM sklady_zamowien WHERE id_zamowienia = ?
                """, [z_id])]

UPDATE sklady_zamowien SET liczba = ? WHERE id_zamowienia = ? AND id_dania = ?

INSERT INTO sklady_zamowien VALUES (?, ?, ?, ?)

DELETE FROM sklady_zamowien WHERE id_zamowienia = ? AND id_dania = ?