from flask import Blueprint, request, render_template, redirect
from definitions.models import Asset, db
from config import item_attributes
from sqlalchemy import exc  # Import exc for database exceptions

update_bp = Blueprint('editasset', __name__)

@update_bp.route('/update/<string:primary_value>/', methods=['GET', 'POST'])
def update(primary_value):
    # Identify the primary attribute
    primary_attrib = next(
        (attrib for attrib, config in item_attributes.items() if config.primary),
        None
    )

    if not primary_attrib:
        return "Primary attribute not defined in configuration."

    # Fetch the item using the primary attribute
    item = Asset.query.filter_by(**{primary_attrib: primary_value}).first()
    if not item:
        return f"{item_attributes[primary_attrib].display_name} '{primary_value}' not found."   # Move this to a template

    if request.method == 'GET':
        return render_template('update.html', item=item, item_attributes=item_attributes)

    if request.method == 'POST':
        # Dynamically get form data using item_attributes
        form_data = {attrib: request.form[attrib] for attrib in item_attributes}

        # Validate status (if it's an option field)
        if 'status' in item_attributes and item_attributes['status'].html_input_type == "select":
            if form_data['status'] not in item_attributes['status'].options:
                return render_template('update.html', item=item, exc='status', item_attributes=item_attributes)

        # Convert staffnum to int (if it's a number field)
        if 'staffnum' in item_attributes and item_attributes['staffnum'].html_input_type == "number":
            try:
                form_data['staffnum'] = int(form_data['staffnum'])
            except ValueError:
                return render_template('update.html', item=item, exc='staffnum', item_attributes=item_attributes)

        # Update the item with the new data
        for attrib, value in form_data.items():
            setattr(item, attrib, value)

        try:
            db.session.commit()
        except exc.IntegrityError:
            return render_template('update.html', item=item, exc='integrity', item_attributes=item_attributes)
        except (exc.StatementError, exc.InvalidRequestError):
            return render_template('update.html', item=item, exc='status', item_attributes=item_attributes)

        return redirect('/viewall')