我是Flask的新手,我只想用它创建一个简单的市场。我的网站中有三种类型的用户[卖家,买家,管理员]。我认为每种类型的用户都应该拥有自己的注册和登录信息,因为由于他们的角色,应该提供不同类型的信息。另一方面,每个用户可能在网站中扮演多个角色,因此我认为不可能为所有用户创建一个表并为其分配角色。因此,我认为每种类型的用户应该有三个表。
由于只有用户装载程序可供用户使用,因此我如何实现我的目标。
我不确定是否正确,我想我可以创建一个具有所有必填列的完整用户,并根据用户用于注册的地址将适当的信息添加到数据库中。并根据用户用于登录的地址找到适当的角色并显示适当的视图。
答案 0 :(得分:2)
您可以对所有类型的用户使用一种模型:
class User(UserMixin, db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(64), index=True, unique=True)
email = db.Column(db.String(120), index=True, unique=True)
usertype = db.Column(db.String(120))
password_hash = db.Column(db.String(128))
User Loader仅具有ID:
@login.user_loader
def load_user(id):
return User.query.get(int(id))
确保在注册过程中应询问用户类型:
class RegistrationForm(FlaskForm):
username = StringField('Username', validators=[DataRequired()])
email = StringField('Email', validators=[DataRequired(), Email()])
usertype = SelectField('User Type', choices = [('sellers', 'sellers'),('buyers', 'buyers')], validators=[DataRequired()])
password = PasswordField('Password', validators=[DataRequired()])
password2 = PasswordField(
'Repeat Password', validators=[DataRequired(), EqualTo('password')])
submit = SubmitField('Register')
并且无需在登录时提供用户类型:
class LoginForm(FlaskForm):
username = StringField('Username', validators=[DataRequired()])
password = PasswordField('Password', validators=[DataRequired()])
remember_me = BooleanField('Remember Me')
submit = SubmitField('Sign In')
在HTMl文件中,您可以进行相应的编码。以从base.html摘录的这段代码为例。
{% if current_user.usertype == 'buyers' %}
<ul class="nav navbar-nav">
<li class="dropdown">
<a class="dropdown-toggle" data-toggle="dropdown" href="#">List
<span class="caret"></span></a>
<ul class="dropdown-menu">
<li><a href="{{ url_for('buyerslist')}}">List of buyers</a></li>
</ul>
</li>
</ul>
{% endif %}