def create(): form = CompanyForm(request.form) if request.method=='POST' and g.user.id and form.validate(): company = Company(user_id=g.user.id, name=form.name.data, scale=form.scale.data, contact_name=form.contact_name.data, telephone=form.telephone.data, website=form.website.data, address=form.address.data, postcode=form.postcode.data, ) db_session.add(company) db_session.commit() if company.id: g.db.execute(users.update().values({users.c.company_id: company.id}).where(users.c.id==g.user.id)) return redirect(url_for('home.index')) return render_template('company/create.html',form=form)
def register(): username = request.form['username'] password = request.form['password'] if username and password: user = User(username, password, realname = request.form['realname'] if request.form.has_key('realname') else '', company_id = request.form['company_id'] if request.form.has_key('company_id') else 0, ) result = User.query.filter_by(username = user.username).first() if result: return jsonify(error=1, code='username_exists', message='用户名已经存在') else: db_session.add(user) db_session.commit() if request.form.has_key('company_name'): row = g.db.execute(select([companies.c.id], and_(companies.c.name==request.form['company_name']))).fetchone() company_id = g.db.execute(companies.insert(), name=request.form['company_name'], user_id=user.id).inserted_primary_key[0] if row is None else row['id'] g.db.execute(users.update().values({users.c.company_id: company_id, users.c.is_active: 1}).where(users.c.id==user.id)) return jsonify(error=0, code='success', message='成功注册', user_id = user.id, company_id=user.company_id) return jsonify(error=1, code = 'no_username_or_password', message='没有输入用户名或密码')