def change():
id = request.args.get('id', type=int)
card = Card.query.get(int(id))
form = CardForm()
if form.validate_on_submit():
category = form.category_expenses.data if form.kind.data == 0 else form.category_incomes.data
card.kind = form.kind.data
card.price = form.price.data
card.category = category
card.day = form.day.data
card.month = form.month.data
card.year = form.year.data
card.note = form.note.data
db.session.commit()
return redirect(url_for('main.index'))
elif request.method == 'GET':
form.kind.data = card.kind
form.price.data = card.price
form.category_expenses.data = card.category
form.category_incomes.data = card.category
form.day.data = card.day
form.month.data = card.month
form.year.data = card.year
form.note.data = card.note
return render_template('card/change_card.html', form=form)
def create():
form = CardForm()
if form.validate_on_submit():
category = form.category_expenses.data if form.kind.data == 0 else form.category_incomes.data
card = Card(price=form.price.data,
category=category,
note=form.note.data,
kind=bool(form.kind.data),
payer=current_user,
year=form.year.data,
month=form.month.data,
day=form.day.data)
db.session.add(card)
db.session.commit()
return redirect(url_for('main.index'))
elif request.method == 'GET':
form.day.data = datetime.utcnow().date().day
form.month.data = datetime.utcnow().date().month
form.year.data = datetime.utcnow().date().year
return render_template('card/create_card.html', form=form)
