diff --git a/app/forms.py b/app/forms.py
index 8411c44..61ab915 100644
--- a/app/forms.py
+++ b/app/forms.py
@@ -1,7 +1,8 @@
from datetime import datetime, timedelta
+from flask import session
from flask.ext.login import current_user
from flask_wtf import Form
-from wtforms import SelectField, DateTimeField, validators, SubmitField, HiddenField
+from wtforms import SelectField, DateTimeField, validators, SubmitField, StringField
from models import User, Location
from utils import euro
@@ -32,4 +33,25 @@ class OrderItemForm(Form):
submit_button = SubmitField('Submit')
def populate(self, location):
- self.product_id.choices = [(i.id, (i.name + ": " + euro(i.price))) for i in location.products]
\ No newline at end of file
+ self.product_id.choices = [(i.id, (i.name + ": " + euro(i.price))) for i in location.products]
+
+
+class AnonOrderItemForm(OrderItemForm):
+ name = StringField('Name', validators=[validators.required()])
+
+ def populate(self, location):
+ OrderItemForm.populate(self, location)
+ if self.name.data is None:
+ self.name.data = session.get('anon_name', None)
+
+ def validate(self):
+ rv = OrderForm.validate(self)
+ if not rv:
+ return False
+
+ # check if we have a user with this name
+ user = User.query.filter_by(username=self.name.data).first()
+ if user is not None:
+ self.name.errors.append('Name already in use')
+ return False
+ return True
diff --git a/app/models.py b/app/models.py
index cedbce9..14ccd8c 100644
--- a/app/models.py
+++ b/app/models.py
@@ -80,6 +80,7 @@ class Order(db.Model):
location_id = db.Column(db.Integer, db.ForeignKey('location.id'))
starttime = db.Column(db.DateTime)
stoptime = db.Column(db.DateTime)
+ public = db.Column(db.Boolean, default=True)
items = db.relationship('OrderItem', backref='order', lazy='dynamic')
def configure(self, courrier, location, starttime, stoptime):
@@ -94,13 +95,13 @@ class Order(db.Model):
def group_by_user(self):
group = defaultdict(list)
for item in self.items:
- group[item.user_id] += [item.product]
+ group[item.get_name()] += [item.product]
return group
def group_by_user_pay(self):
group = defaultdict(int)
for item in self.items:
- group[item.user] += item.product.price
+ group[item.get_name()] += item.product.price
return group
class OrderItem(db.Model):
@@ -108,20 +109,26 @@ class OrderItem(db.Model):
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
order_id = db.Column(db.Integer, db.ForeignKey('order.id'))
product_id = db.Column(db.Integer, db.ForeignKey('product.id'))
+ name = db.Column(db.String(120))
def configure(self, user, order, product):
self.user = user
self.order = order
self.product = product
- def __repr__(self):
- return 'OrderItem'
+ def get_name(self):
+ if self.user_id is not None and self.user_id > 0:
+ return self.user.username
+ return self.name
- def can_delete(self, order_id, user_id):
- if self.user_id != user_id:
- return False
+ def __repr__(self):
+ return 'Order %d: %s wants %s' % (self.order_id, self.get_name(), self.product.name)
+
+ def can_delete(self, order_id, user_id, name):
if int(self.order_id) != int(order_id):
return False
if self.order.stoptime and self.order.stoptime < datetime.now():
return False
- return True
+ if self.user_id == user_id or self.name == name:
+ return True
+ return False
diff --git a/app/templates/home.html b/app/templates/home.html
index f2bd54d..2d42adc 100644
--- a/app/templates/home.html
+++ b/app/templates/home.html
@@ -5,4 +5,14 @@
Welcome to FoodBot
This is the home page for FoodBot
+
{% endblock %}
\ No newline at end of file
diff --git a/app/templates/home_loggedin.html b/app/templates/home_loggedin.html
deleted file mode 100644
index 51b3f32..0000000
--- a/app/templates/home_loggedin.html
+++ /dev/null
@@ -1,14 +0,0 @@
-{% extends "home.html" %}
-{% block container %}
-{{ super() }}
-
-{% endblock %}
\ No newline at end of file
diff --git a/app/templates/order.html b/app/templates/order.html
index 5aba8f0..136a12f 100644
--- a/app/templates/order.html
+++ b/app/templates/order.html
@@ -10,7 +10,7 @@
Close
{%- endif %}
Courrier: {{ order.courrier.username }}
- {% if order.courrier == None %}
+ {% if order.courrier == None and not current_user.is_anonymous() %}
Volunteer
{% endif %}
@@ -20,12 +20,12 @@
Total price: {{ total_price|euro }}
Orders
{% for item in order.items %}
- {{ item.user.username }} - {{ item.product.name }} - {{ item.product.price|euro }}
- {% if item.can_delete(order.id, current_user.id) -%}{%- endif %}
+ {{ item.get_name() }} - {{ item.product.name }} - {{ item.product.price|euro }}
+ {% if item.can_delete(order.id, current_user.id, session.get('anon_name', '')) -%}{%- endif %}
{% endfor %}
Debts
{% for key, value in total_payments.items() %}
- {{ key.username }} - {{ value|euro }}
+ {{ key }} - {{ value|euro }}
{% endfor %}
diff --git a/app/templates/orders.html b/app/templates/orders.html
index abae873..6cab482 100644
--- a/app/templates/orders.html
+++ b/app/templates/orders.html
@@ -12,9 +12,11 @@
{% endfor %}
+ {% if not current_user.is_anonymous() %}
{{ wtf.quick_form(form, action=url_for('.order_create'), button_map={'submit_button': 'primary'}, form_type='horizontal') }}
+ {% endif %}
{% endblock %}
\ No newline at end of file
diff --git a/app/utils.py b/app/utils.py
index 60150bf..0aff1fe 100644
--- a/app/utils.py
+++ b/app/utils.py
@@ -1,6 +1,8 @@
from flask import render_template
from app import app
+from login import login_manager
+
__author__ = 'feliciaan'
@app.template_filter('euro')
@@ -14,4 +16,22 @@ def handle404(e):
@app.errorhandler(401)
def handle401(e):
- return render_template('errors/401.html'), 401
\ No newline at end of file
+ return render_template('errors/401.html'), 401
+
+class AnonymouseUser:
+ def is_active(self):
+ return False
+
+ def is_authenticated(self):
+ return False
+
+ def is_anonymous(self):
+ return True
+
+ def is_admin(self):
+ return False
+
+ def get_id(self):
+ return None
+
+login_manager.anonymous_user = AnonymouseUser
\ No newline at end of file
diff --git a/app/views/__init__.py b/app/views/__init__.py
index a44037b..06aa7e2 100644
--- a/app/views/__init__.py
+++ b/app/views/__init__.py
@@ -12,10 +12,7 @@ import views.order
@app.route('/')
def home():
- if not current_user.is_anonymous():
- orders = Order.query.filter((Order.stoptime > datetime.now()) | (Order.stoptime == None)).all()
- return render_template('home_loggedin.html', orders=orders)
- return render_template('home.html')
+ return render_template('home.html', orders=views.order.get_orders())
@app.route('/about/')
diff --git a/app/views/order.py b/app/views/order.py
index 8e7d0e9..a172edb 100644
--- a/app/views/order.py
+++ b/app/views/order.py
@@ -1,22 +1,22 @@
__author__ = 'feliciaan'
-from flask import url_for, render_template, abort, redirect, Blueprint, flash
+from flask import url_for, render_template, abort, redirect, Blueprint, flash, session
from flask.ext.login import current_user, login_required
import random
from datetime import datetime
from app import app, db
from models import Order, OrderItem
-from forms import OrderItemForm, OrderForm
+from forms import OrderItemForm, OrderForm, AnonOrderItemForm
order_bp = Blueprint('order_bp', 'order')
@order_bp.route('/')
-@login_required
def orders():
- orders = Order.query.filter((Order.stoptime > datetime.now()) | (Order.stoptime == None)).all()
- orderForm = OrderForm()
- orderForm.populate()
- return render_template('orders.html', orders=orders, form=orderForm)
+ orderForm = None
+ if not current_user.is_anonymous():
+ orderForm = OrderForm()
+ orderForm.populate()
+ return render_template('orders.html', orders=get_orders(), form=orderForm)
@order_bp.route('/create', methods=['GET', 'POST'])
@@ -35,41 +35,52 @@ def order_create():
@order_bp.route('/')
-@login_required
def order(id):
order = Order.query.filter(Order.id == id).first()
- if order is not None:
+ if order is None:
+ abort(404)
+ form = None
+ if not current_user.is_anonymous():
form = OrderItemForm()
- form.populate(order.location)
- total_price = sum([o.product.price for o in order.items])
- total_payments = order.group_by_user_pay()
- return render_template('order.html', order=order, form=form, total_price=total_price, total_payments=total_payments)
- return abort(404)
+ else:
+ form = AnonOrderItemForm()
+ form.populate(order.location)
+ total_price = sum([o.product.price for o in order.items])
+ total_payments = order.group_by_user_pay()
+ return render_template('order.html', order=order, form=form, total_price=total_price, total_payments=total_payments)
@order_bp.route('//create', methods=['GET', 'POST'])
-@login_required
def order_item_create(id):
order = Order.query.filter(Order.id == id).first()
- if order is not None:
+ if order is None:
+ abort(404)
+ form = None
+ if not current_user.is_anonymous():
form = OrderItemForm()
- form.populate(order.location)
- if form.validate_on_submit():
- item = OrderItem()
- form.populate_obj(item)
- item.order_id = id
+ else:
+ form = AnonOrderItemForm()
+ form.populate(order.location)
+ if form.validate_on_submit():
+ item = OrderItem()
+ form.populate_obj(item)
+ item.order_id = id
+ if not current_user.is_anonymous():
item.user_id = current_user.id
- db.session.add(item)
- db.session.commit()
- return redirect(url_for('.order', id=id))
- return render_template('order_form.html', form=form, url=url_for(".order_item_create", id=id))
- return abort(404)
+ else:
+ session['anon_name'] = item.name
+ db.session.add(item)
+ db.session.commit()
+ return redirect(url_for('.order', id=id))
+ return render_template('order_form.html', form=form, url=url_for(".order_item_create", id=id))
@order_bp.route('///delete')
-@login_required
def delete_item(order_id, item_id):
item = OrderItem.query.filter(OrderItem.id == item_id).first()
- if item.can_delete(order_id, current_user.id):
+ id = None
+ if not current_user.is_anonymous():
+ id = current_user.id
+ if item.can_delete(order_id, id, session.get('anon_name', '')):
db.session.delete(item)
db.session.commit()
return redirect(url_for('.order', id=order_id))
@@ -80,33 +91,32 @@ def delete_item(order_id, item_id):
@login_required
def volunteer(id):
order = Order.query.filter(Order.id == id).first()
- if order is not None:
- print(order.courrier_id)
- if order.courrier_id == 0:
- order.courrier_id = current_user.id
- db.session.commit()
- flash("Thank you for volunteering!")
- else:
- flash("Volunteering not possible!")
- return redirect(url_for('.order', id=id))
- abort(404)
+ if order is None:
+ abort(404)
+ if order.courrier_id is None or order.courrier_id == 0:
+ order.courrier_id = current_user.id
+ db.session.commit()
+ flash("Thank you for volunteering!")
+ else:
+ flash("Volunteering not possible!")
+ return redirect(url_for('.order', id=id))
@order_bp.route('//close')
@login_required
def close_order(id):
order = Order.query.filter(Order.id == id).first()
- if order is not None:
- if (current_user.id == order.courrier_id or current_user.is_admin()) \
- and order.stoptime is None or (order.stoptime > datetime.now()):
- order.stoptime = datetime.now()
- if order.courrier_id == 0 or order.courrier_id is None:
- courrier = select_user(order.items)
- if courrier is not None:
- order.courrier_id = courrier.id
- db.session.commit()
- return redirect(url_for('.order', id=id))
- abort(401)
+ if order is None:
+ abort(401)
+ if (current_user.id == order.courrier_id or current_user.is_admin()) \
+ and order.stoptime is None or (order.stoptime > datetime.now()):
+ order.stoptime = datetime.now()
+ if order.courrier_id == 0 or order.courrier_id is None:
+ courrier = select_user(order.items)
+ if courrier is not None:
+ order.courrier_id = courrier.id
+ db.session.commit()
+ return redirect(url_for('.order', id=id))
app.register_blueprint(order_bp, url_prefix='/order')
@@ -120,7 +130,16 @@ def select_user(items):
while user is None:
item = random.choice(items)
user = item.user
- if random.randint(user.bias, 100) < 80:
- user = None
+ if user:
+ if random.randint(user.bias, 100) < 80:
+ user = None
return user
+
+def get_orders():
+ orders = []
+ if not current_user.is_anonymous():
+ orders = Order.query.filter((Order.stoptime > datetime.now()) | (Order.stoptime == None)).all()
+ else:
+ orders = Order.query.filter(((Order.stoptime > datetime.now()) | (Order.stoptime == None) & (Order.public == True))).all()
+ return orders