137 lines
4.3 KiB
Python
137 lines
4.3 KiB
Python
from flask import Blueprint, render_template, redirect, url_for, session, flash
|
|
from .models import DeliveryOrder
|
|
|
|
main = Blueprint('main', __name__)
|
|
|
|
|
|
@main.route('/')
|
|
def home():
|
|
return """
|
|
<h1>DO Tracker Online</h1>
|
|
<p><a href='/login'>Login as Store</a></p>
|
|
<p><a href='/store'>Store Dashboard</a></p>
|
|
"""
|
|
|
|
|
|
@main.route('/store')
|
|
def store_dashboard():
|
|
if 'store_id' not in session:
|
|
flash("You must be logged in to access the store dashboard.", "warning")
|
|
return redirect(url_for('auth.login'))
|
|
|
|
store_name = session.get('store_name')
|
|
|
|
return f"""
|
|
<h1>Welcome, {store_name}</h1>
|
|
<p><a href='/logout'>Logout</a></p>
|
|
<ul>
|
|
<li><a href='/store/do-entry'>Enter New DO</a> (coming next)</li>
|
|
<li><a href='/store/track'>Track DOs</a> (coming soon)</li>
|
|
</ul>
|
|
"""
|
|
from flask import request
|
|
from .models import DeliveryOrder, Store
|
|
from . import db
|
|
|
|
@main.route('/store/do-entry', methods=['GET', 'POST'])
|
|
def do_entry():
|
|
if 'store_id' not in session:
|
|
return redirect(url_for('auth.login'))
|
|
|
|
stores = Store.query.order_by(Store.name).all()
|
|
|
|
if request.method == 'POST':
|
|
do_number = request.form.get('do_number')
|
|
delivery_number = request.form.get('delivery_number')
|
|
final_location = int(request.form.get('final_location'))
|
|
created_by = session.get('store_name')
|
|
|
|
# Prevent duplicate DO numbers
|
|
existing = DeliveryOrder.query.filter_by(do_number=do_number).first()
|
|
if existing:
|
|
flash("DO already exists!", "danger")
|
|
return redirect(url_for('main.do_entry'))
|
|
|
|
new_do = DeliveryOrder(
|
|
do_number=do_number,
|
|
delivery_number=delivery_number,
|
|
final_location=final_location,
|
|
created_by=created_by,
|
|
status="Ready for Collection"
|
|
)
|
|
|
|
db.session.add(new_do)
|
|
db.session.commit()
|
|
flash("DO created successfully.", "success")
|
|
return redirect(url_for('main.store_dashboard'))
|
|
|
|
return render_template("do_entry.html", stores=stores)
|
|
@main.route('/store/track', methods=['GET', 'POST'])
|
|
def track_do():
|
|
if 'store_id' not in session:
|
|
return redirect(url_for('auth.login'))
|
|
|
|
do = None
|
|
movements = []
|
|
|
|
if request.method == 'POST':
|
|
search = request.form.get('search')
|
|
do = DeliveryOrder.query.filter(
|
|
(DeliveryOrder.do_number == search) |
|
|
(DeliveryOrder.delivery_number == search)
|
|
).first()
|
|
|
|
if do:
|
|
movements = do.movements
|
|
|
|
return render_template('track_do.html', do=do, movements=movements)
|
|
from datetime import datetime
|
|
from .models import Movement
|
|
|
|
@main.route('/store/move', methods=['GET', 'POST'])
|
|
def move_do():
|
|
if 'store_id' not in session:
|
|
return redirect(url_for('auth.login'))
|
|
|
|
message = None
|
|
if request.method == 'POST':
|
|
search = request.form.get('do_search')
|
|
do = DeliveryOrder.query.filter(
|
|
(DeliveryOrder.do_number == search) |
|
|
(DeliveryOrder.delivery_number == search)
|
|
).first()
|
|
|
|
if not do:
|
|
flash("DO not found", "danger")
|
|
return redirect(url_for('main.move_do'))
|
|
|
|
handled_by = request.form.get('handled_by')
|
|
comment = request.form.get('comment')
|
|
mark_departed = request.form.get('departed')
|
|
|
|
branch_id = session.get('store_id')
|
|
|
|
# Check if this store has already logged a movement
|
|
existing = Movement.query.filter_by(do_id=do.id, branch_id=branch_id).first()
|
|
|
|
if not existing:
|
|
# First time it arrived at this store
|
|
move = Movement(
|
|
do_id=do.id,
|
|
branch_id=branch_id,
|
|
handled_by=handled_by,
|
|
comment=comment
|
|
)
|
|
db.session.add(move)
|
|
else:
|
|
# Already exists → optionally mark departure
|
|
if mark_departed == "on" and not existing.departed_at:
|
|
existing.departed_at = datetime.utcnow()
|
|
existing.comment = (existing.comment or '') + f" | {comment}"
|
|
|
|
db.session.commit()
|
|
flash("Movement updated successfully.", "success")
|
|
return redirect(url_for('main.track_do'))
|
|
|
|
return render_template("move_do.html")
|