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)