39 lines
1.4 KiB
Python
39 lines
1.4 KiB
Python
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="Sub-cont. File Import", subcontractors=subcontractors)
|
|
|
|
|
|
@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)
|