191 lines
6.6 KiB
Python
191 lines
6.6 KiB
Python
from flask import Blueprint, render_template, request, redirect, flash
|
|
from app import db
|
|
from app.models.subcontractor_model import Subcontractor
|
|
from app.utils.helpers import login_required
|
|
|
|
subcontractor_bp = Blueprint("subcontractor", __name__, url_prefix="/subcontractor")
|
|
|
|
# ---------------- ADD -----------------
|
|
@subcontractor_bp.route("/add")
|
|
@login_required
|
|
def add_subcontractor():
|
|
return render_template("subcontractor/add.html")
|
|
@subcontractor_bp.route("/save", methods=["POST"])
|
|
@login_required
|
|
def save_subcontractor():
|
|
# 1. Get and clean the name from the form
|
|
name = request.form.get("subcontractor_name", "").strip()
|
|
|
|
# 2. Basic validation: Ensure the name isn't empty
|
|
if not name:
|
|
flash("Subcontractor name cannot be empty.", "danger")
|
|
return redirect("/subcontractor/add")
|
|
|
|
# 3. Check if a subcontractor with this name already exists
|
|
existing_sub = Subcontractor.query.filter_by(subcontractor_name=name).first()
|
|
|
|
if existing_sub:
|
|
flash(f"Subcontractor with name '{name}' already exists!", "danger")
|
|
return redirect("/subcontractor/add")
|
|
|
|
# 4. If no duplicate is found, proceed to save
|
|
try:
|
|
subcontractor = Subcontractor(
|
|
subcontractor_name=name,
|
|
contact_person=request.form.get("contact_person"),
|
|
mobile_no=request.form.get("mobile_no"),
|
|
email_id=request.form.get("email_id"),
|
|
gst_no=request.form.get("gst_no")
|
|
)
|
|
|
|
db.session.add(subcontractor)
|
|
db.session.commit()
|
|
flash("Subcontractor added successfully!", "success")
|
|
|
|
except Exception as e:
|
|
db.session.rollback()
|
|
flash("An error occurred while saving. Please try again.", "danger")
|
|
|
|
return redirect("/subcontractor/list")
|
|
|
|
# ---------------- LIST -----------------
|
|
@subcontractor_bp.route("/list")
|
|
@login_required
|
|
def subcontractor_list():
|
|
subcontractors = Subcontractor.query.all()
|
|
return render_template("subcontractor/list.html", subcontractors=subcontractors)
|
|
|
|
# ---------------- EDIT -----------------
|
|
@subcontractor_bp.route("/edit/<int:id>")
|
|
@login_required
|
|
def edit_subcontractor(id):
|
|
subcontractor = Subcontractor.query.get_or_404(id)
|
|
return render_template("subcontractor/edit.html", subcontractor=subcontractor)
|
|
|
|
# ---------------- UPDATE -----------------
|
|
@subcontractor_bp.route("/update/<int:id>", methods=["POST"])
|
|
@login_required
|
|
def update_subcontractor(id):
|
|
subcontractor = Subcontractor.query.get_or_404(id)
|
|
new_name = request.form.get("subcontractor_name")
|
|
|
|
# Check if the new name is taken by someone ELSE (not this current ID)
|
|
duplicate = Subcontractor.query.filter(
|
|
Subcontractor.subcontractor_name == new_name,
|
|
Subcontractor.id != id
|
|
).first()
|
|
|
|
if duplicate:
|
|
flash("Another subcontractor already uses this name.", "danger")
|
|
return redirect(f"/subcontractor/edit/{id}")
|
|
|
|
subcontractor.subcontractor_name = new_name
|
|
|
|
db.session.commit()
|
|
|
|
flash("Subcontractor updated successfully!", "success")
|
|
return redirect("/subcontractor/list")
|
|
|
|
# ---------------- DELETE -----------------
|
|
@subcontractor_bp.route("/delete/<int:id>")
|
|
@login_required
|
|
def delete_subcontractor(id):
|
|
subcontractor = Subcontractor.query.get_or_404(id)
|
|
|
|
db.session.delete(subcontractor)
|
|
db.session.commit()
|
|
|
|
flash("Subcontractor deleted successfully!", "success")
|
|
return redirect("/subcontractor/list")
|
|
from flask import Blueprint, render_template, request, redirect, flash
|
|
from app import db
|
|
from app.models.subcontractor_model import Subcontractor
|
|
from app.utils.helpers import login_required
|
|
|
|
subcontractor_bp = Blueprint("subcontractor", __name__, url_prefix="/subcontractor")
|
|
|
|
# ---------------- ADD -----------------
|
|
@subcontractor_bp.route("/add")
|
|
@login_required
|
|
def add_subcontractor():
|
|
return render_template("subcontractor/add.html")
|
|
|
|
@subcontractor_bp.route("/save", methods=["POST"])
|
|
@login_required
|
|
def save_subcontractor():
|
|
name = request.form.get("subcontractor_name", "").strip()
|
|
if not name:
|
|
flash("Subcontractor name cannot be empty.", "danger")
|
|
return redirect("/subcontractor/add")
|
|
existing_sub = Subcontractor.query.filter_by(subcontractor_name=name).first()
|
|
|
|
if existing_sub:
|
|
flash(f"Subcontractor with name '{name}' already exists!", "danger")
|
|
return redirect("/subcontractor/add")
|
|
try:
|
|
subcontractor = Subcontractor(
|
|
subcontractor_name=name,
|
|
contact_person=request.form.get("contact_person"),
|
|
mobile_no=request.form.get("mobile_no"),
|
|
email_id=request.form.get("email_id"),
|
|
gst_no=request.form.get("gst_no")
|
|
)
|
|
|
|
db.session.add(subcontractor)
|
|
db.session.commit()
|
|
flash("Subcontractor added successfully!", "success")
|
|
|
|
except Exception as e:
|
|
db.session.rollback()
|
|
flash("An error occurred while saving. Please try again.", "danger")
|
|
|
|
return redirect("/subcontractor/list")
|
|
|
|
# ---------------- LIST -----------------
|
|
@subcontractor_bp.route("/list")
|
|
@login_required
|
|
def subcontractor_list():
|
|
subcontractors = Subcontractor.query.all()
|
|
return render_template("subcontractor/list.html", subcontractors=subcontractors)
|
|
|
|
# ---------------- EDIT -----------------
|
|
@subcontractor_bp.route("/edit/<int:id>")
|
|
@login_required
|
|
def edit_subcontractor(id):
|
|
subcontractor = Subcontractor.query.get_or_404(id)
|
|
return render_template("subcontractor/edit.html", subcontractor=subcontractor)
|
|
|
|
# ---------------- UPDATE -----------------
|
|
@subcontractor_bp.route("/update/<int:id>", methods=["POST"])
|
|
@login_required
|
|
def update_subcontractor(id):
|
|
subcontractor = Subcontractor.query.get_or_404(id)
|
|
new_name = request.form.get("subcontractor_name")
|
|
|
|
# Check if the new name is taken by someone ELSE (not this current ID)
|
|
duplicate = Subcontractor.query.filter(
|
|
Subcontractor.subcontractor_name == new_name,
|
|
Subcontractor.id != id
|
|
).first()
|
|
|
|
if duplicate:
|
|
flash("Another subcontractor already uses this name.", "danger")
|
|
return redirect(f"/subcontractor/edit/{id}")
|
|
|
|
subcontractor.subcontractor_name = new_name
|
|
db.session.commit()
|
|
|
|
flash("Subcontractor updated successfully!", "success")
|
|
return redirect("/subcontractor/list")
|
|
|
|
# ---------------- DELETE -----------------
|
|
@subcontractor_bp.route("/delete/<int:id>")
|
|
@login_required
|
|
def delete_subcontractor(id):
|
|
subcontractor = Subcontractor.query.get_or_404(id)
|
|
|
|
db.session.delete(subcontractor)
|
|
db.session.commit()
|
|
|
|
flash("Subcontractor deleted successfully!", "success")
|
|
return redirect("/subcontractor/list") |