diff --git a/app/models/order.py b/app/models/order.py index b738c45..31f5620 100644 --- a/app/models/order.py +++ b/app/models/order.py @@ -114,3 +114,17 @@ class Order(db.Model): if self.courier_id == user_id or (user and user.is_admin()): return True return False + + def can_modify_prices(self, user_id: int) -> bool: + if not self.is_closed(): + return False + if user_id is None: + return False + if self.courier_id == user_id: + return True + user = User.query.filter_by(id=user_id).first() + return user and user.is_admin() + + def can_modify_payment(self, user_id: int) -> bool: + user = User.query.filter_by(id=user_id).first() + return user and (user.is_admin() or user == self.order.courier) diff --git a/app/templates/order.html b/app/templates/order.html index 7285ce8..313e164 100644 --- a/app/templates/order.html +++ b/app/templates/order.html @@ -290,9 +290,9 @@ {{ order_items | map(attribute="price") | sum | euro }} - {% if paid %}{% endif %} + {% if paid %}{% endif %} - {{ user_name }} + {{ user_name }}