def create(request):
if request.method == 'POST':
form = AccountForm(request.POST)
if form.is_valid():
username = form.cleaned_data['username']
password = form.cleaned_data['password']
email = form.cleaned_data['email']
#don't put email in django user object, already
#stored in the account object.
user = User.objects.create_user(username, '', password)
user.save()
acc = accounts.create_account(email)
acc.user = user
acc.save()
user = authenticate(username=user.username, password=password)
login(request, user)
return HttpResponseRedirect(reverse('webapp.lists.views.lists'))
else:
return render_to_response('account/create-user.html', {
'form' : form,
}, context_instance = RequestContext(request))
else:
return render_to_response('account/create-user.html', {
'form' : AccountForm(),
}, context_instance = RequestContext(request))
def account_new():
form = AccountForm()
if form.validate_on_submit():
if current_user.is_anonymous():
flash(current_user)
flash(
"You have no right to create new account. Please login first!",
category="error")
return redirect(url_for("main.login"))
else:
new_account = Account()
new_account.name = form.name.data
new_account.amount = float(form.amount.data)
new_account.description = form.description.data
new_account.user_id = int(current_user.get_id())
new_account.date = datetime.datetime.now()
new_found = Found()
new_found.account_id = new_account.id
new_found.user_id = int(current_user.get_id())
new_found.amount = form.amount.data
new_found.date = datetime.datetime.now()
new_found.description = u"初始"
db.session.add(new_found)
db.session.add(new_account)
db.session.commit()
return redirect(url_for(".account"))
return render_template("new.html", form=form)
def add_account():
form = AccountForm()
if form.validate_on_submit():
name = form.name.data
category = form.category.data
owner = form.owner.data
form.name.data = ''
retirement = form.retirement.data
act = Account(name=name, category=category, owner=owner, retirement=retirement)
db.session.add(act)
db.session.commit()
return redirect(url_for('.index'))
return render_template("add_account.html", form=form)
def edit(account_id):
account = Account.query.get(account_id)
form = AccountForm(obj=account)
form.validate_on_submit()
if form.validate_on_submit():
account.name = form.name.data
account.category = form.category.data
account.owner = form.owner.data
account.retirement = form.retirement.data
form.name.data = ''
db.session.add(account)
db.session.commit()
return redirect(url_for('.index'))
return render_template("edit.html", account=account, form=form)
