烧瓶中不同类型的用户注册

时间:2020-02-19 07:52:02

标签: flask flask-sqlalchemy

我是Flask的新手,我只想用它创建一个简单的市场。我的网站中有三种类型的用户[卖家,买家,管理员]。我认为每种类型的用户都应该拥有自己的注册和登录信息,因为由于他们的角色,应该提供不同类型的信息。另一方面,每个用户可能在网站中扮演多个角色,因此我认为不可能为所有用户创建一个表并为其分配角色。因此,我认为每种类型的用户应该有三个表。

由于只有用户装载程序可供用户使用,因此我如何实现我的目标。

我不确定是否正确,我想我可以创建一个具有所有必填列的完整用户,并根据用户用于注册的地址将适当的信息添加到数据库中。并根据用户用于登录的地址找到适当的角色并显示适当的视图。

1 个答案:

答案 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 %}
相关问题