Comparison_Project/app/routes/file_import.py

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")

# this route import Subcontractor files
@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")
        RA_Bill_No = request.form.get("RA_Bill_No")
        # file_type = request.form.get("file_type")

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

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

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


# this route import client files
@file_import_bp.route("/import_client", methods=["GET", "POST"])
def client_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_client.html", title="Client File Import", subcontractors=subcontractors)