Example #1
0
def book():

    form = TransactionForm(current_user.username)

    if form.validate_on_submit():

        trans = Transaction(store=form.store.data,
                            date=pd.to_datetime(form.date.data),
                            flexible=form.flexible.data,
                            payment_type=form.payment_type.data,
                            payment_notes=form.payment_notes.data,
                            list=form.grocery_list.data,
                            notes=form.other_notes.data,
                            booking_date=dt.date.today())
        trans.assign_recipient(current_user)

        if current_user.buddy is not None:
            trans.claimed = True
            send_confirmation(current_user.buddy,
                              'volunteer_buddy',
                              transaction=trans)

        db.session.add(trans)
        db.session.commit()

        day_of_week = calendar.day_name[pd.to_datetime(
            form.date.data).weekday()]
        str_date = form.date.data.strftime('%m/%d')
        flash(f'Delivery booked for {day_of_week}, {str_date}!')
        send_confirmation(current_user, 'recipient_booking', transaction=trans)

        return redirect(url_for('deliveries', username=current_user.username))

    elif request.method == 'GET':

        d = dt.datetime.today()
        # default to next Friday
        while d.weekday() != 4:
            d += dt.timedelta(1)
        form.date.data = d
        # fetch most recent store, most recent payment type, most recent payment notes
        most_recent_trans = db.session.query(Transaction).join(
            Recipient).filter(
                Recipient.username == current_user.username).order_by(
                    desc(Transaction.date)).first()
        if most_recent_trans is not None:
            form.store.data = most_recent_trans.store
            form.payment_type.data = most_recent_trans.payment_type
            form.payment_notes.data = most_recent_trans.payment_notes

    return render_template('standard_form.html',
                           header='Book Delivery',
                           form=form)
Example #2
0
 def test_transaction_assign_recipient_volunteer(self):
     t = Transaction()
     r = Recipient(username='******')
     v = Volunteer(username='******')
     db.session.add_all([t, r, v])
     db.session.commit()
     t.assign_recipient(r)
     self.assertEqual(t.recipient, r)
     self.assertEqual(t.recipient_id, r.id)
     self.assertFalse(t.claimed)
     self.assertEqual(r.transactions[0], t)
     t.assign_volunteer(v)
     db.session.add(t)
     db.session.commit()
     self.assertEqual(t.volunteer, v)
     self.assertEqual(t.volunteer_id, v.id)
     self.assertTrue(t.claimed)
     self.assertEqual(v.transactions[0], t)