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='没有输入用户名或密码')
