from flask import Blueprint, render_template, request, flash
from app.services.file_service import FileService
from app.models.subcontractor_model import Subcontractor

file_import_bp = Blueprint("file_import", __name__, url_prefix="/file")

@file_import_bp.route("/import", methods=["GET", "POST"])
def import_file():
    subcontractors = Subcontractor.query.all()

    if request.method == "POST":
        file = request.files.get("file")
        subcontractor_id = request.form.get("subcontractor_id")
        file_type = request.form.get("file_type")

        service = FileService()
        success, msg = service.handle_file_upload(file, subcontractor_id, file_type)

        flash(msg, "success" if success else "danger")

    return render_template("file_import.html", title="File Import", subcontractors=subcontractors)