same changes of sub-cont dashboard
This commit is contained in:
@@ -1,10 +1,10 @@
|
||||
from app import db
|
||||
import os
|
||||
import pandas as pd
|
||||
from werkzeug.utils import secure_filename
|
||||
from app.utils.file_utils import ensure_upload_folder
|
||||
|
||||
from app.config import Config
|
||||
from app import db
|
||||
from app.utils.file_utils import get_uploads_folder
|
||||
from app.utils.file_utils import ALLOWED_EXTENSIONS
|
||||
|
||||
# Subcontractor models import
|
||||
from app.models.trench_excavation_model import TrenchExcavation
|
||||
@@ -24,7 +24,7 @@ class FileService:
|
||||
|
||||
# ---------------- COMMON HELPERS ----------------
|
||||
def allowed_file(self, filename):
|
||||
return ("." in filename and filename.rsplit(".", 1)[1].lower() in Config.ALLOWED_EXTENSIONS)
|
||||
return ("." in filename and filename.rsplit(".", 1)[1].lower() in ALLOWED_EXTENSIONS)
|
||||
|
||||
def normalize(self, val):
|
||||
if val is None or pd.isna(val):
|
||||
@@ -52,8 +52,9 @@ class FileService:
|
||||
return False, "Invalid file type! Allowed: CSV, XLSX, XLS"
|
||||
|
||||
ensure_upload_folder()
|
||||
path = get_uploads_folder()
|
||||
|
||||
folder = os.path.join(Config.UPLOAD_FOLDER, f"sub_{subcontractor_id}")
|
||||
folder = os.path.join(path, f"sub_{subcontractor_id}")
|
||||
os.makedirs(folder, exist_ok=True)
|
||||
|
||||
filename = secure_filename(file.filename)
|
||||
@@ -310,7 +311,6 @@ class FileService:
|
||||
db.session.commit()
|
||||
|
||||
|
||||
|
||||
# ---------------- CLIENT FILE UPLOAD ----------------
|
||||
def handle_client_file_upload(self, file, RA_Bill_No):
|
||||
|
||||
@@ -324,8 +324,9 @@ class FileService:
|
||||
return False, "Invalid file type! Allowed: CSV, XLSX, XLS"
|
||||
|
||||
ensure_upload_folder()
|
||||
path = get_uploads_folder()
|
||||
|
||||
folder = os.path.join(Config.UPLOAD_FOLDER, f"Client_Bill_{RA_Bill_No}")
|
||||
folder = os.path.join(path, f"Client_Bill_{RA_Bill_No}")
|
||||
os.makedirs(folder, exist_ok=True)
|
||||
|
||||
filename = secure_filename(file.filename)
|
||||
|
||||
Reference in New Issue
Block a user